From 72b0979e8a0b112a86986de9b14845e043a82363 Mon Sep 17 00:00:00 2001
From: liqun Fu <liqfu@microsoft.com>
Date: Wed, 25 Sep 2024 22:53:28 +0000
Subject: [PATCH] Fix a wrong assignment that causing mlas benchmark to crash
 (#22221)

### Description
<!-- Describe your changes. -->



### Motivation and Context
<!-- - Why is this change required? What problem does it solve?
- If it fixes an open issue, please link to the issue here. -->

Signed-off-by: Liqun Fu <liqfu@microsoft.com>
---
 onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp b/onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp
index 73c78b8cc3..cb4a389ceb 100644
--- a/onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp
+++ b/onnxruntime/test/mlas/bench/bench_sqnbitgemm.cpp
@@ -80,9 +80,10 @@ void RunSQNBitGemmBenchmark(size_t BlkLen,
   params.A = A.data();
   params.lda = K;
   if (PackedQuantBData != nullptr)
-    params.QuantBDataWorkspace = static_cast<const void*>(PackedQuantBData.get());
+    params.PackedQuantBData = PackedQuantBData.get();
   else
     params.QuantBDataWorkspace = static_cast<const void*>(QuantBData.data());
+
   params.QuantBScale = QuantBScale.data();
   params.QuantBZeroPoint = Symmetric ? nullptr : QuantBZeroPoint.data();
   params.Bias = HasBias ? Bias.data() : nullptr;