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(PackedQuantBData.get()); + params.PackedQuantBData = PackedQuantBData.get(); else params.QuantBDataWorkspace = static_cast(QuantBData.data()); + params.QuantBScale = QuantBScale.data(); params.QuantBZeroPoint = Symmetric ? nullptr : QuantBZeroPoint.data(); params.Bias = HasBias ? Bias.data() : nullptr;