fix benchmark error

luoyu-intel · Nov 3, 2023 · bd36779 · bd36779
1 parent 178ec77
commit bd36779
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/onnxruntime/test/mlas/bench/bench_q4gemm.cpp b/onnxruntime/test/mlas/bench/bench_q4gemm.cpp
@@ -86,11 +86,11 @@ void Q4GEMM_Jblas(benchmark::State& state, int block_size, bool is_asym, MLAS_CO
   params1.ldc = N;
   params1.B = B1_packed.data();
   params1.OutputProcessor = nullptr;
-
-  MlasJblasQ4GemmBatch(M, N, K, 1, &params1, tp.get());
+  std::vector<int8_t> workspace(size_t(M) * K * 4);
+  MlasJblasQ4GemmBatch(M, N, K, 1, &params1, workspace.data(), tp.get());
 
   for (auto _ : state) {
-    MlasJblasQ4GemmBatch(M, N, K, 1, &params1, tp.get());
+    MlasJblasQ4GemmBatch(M, N, K, 1, &params1, workspace.data(), tp.get());
   }
 }