Add vector-matrix product case to wasm GEMM benchmark

robertknight · Mar 25, 2024 · 7be2109 · 7be2109
1 parent 8047a52
commit 7be2109
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/tools/benchmarks/wasm-gemm.js b/tools/benchmarks/wasm-gemm.js
@@ -26,6 +26,7 @@ const cases = [
   { m: 512, n: 512, k: 512 }, // Square
   { m: 128, n: 2048, k: 512 }, // Wide
   { m: 2048, n: 128, k: 512 }, // Tall
+  { m: 1, n: 4096, k: 512 }, // Vector
 ];
 
 function logResult(engine, elapsedMs, m, n, k, iters) {