benchdnn: add regression tests for cases when M == 1

oneapi-src · Feb 1, 2025 · 30c17b9 · 30c17b9
1 parent d891dd6
commit 30c17b9
Show file tree

Hide file tree

Showing 3 changed files with 12 additions and 0 deletions.
diff --git a/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16 b/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16
@@ -25,3 +25,7 @@
 --wtag=ab
 --dtag=ab
 8x2664:2664x256_n"k_tail"
+
+# Test that cases when M == 1 are handled correctly.
+--reset
+--stag=ba,ab --wtag=ab --dtag=ab --dt=bf16 1x2:2x256
diff --git a/tests/benchdnn/inputs/matmul/harness_matmul_regression_float16 b/tests/benchdnn/inputs/matmul/harness_matmul_regression_float16
@@ -2,3 +2,7 @@
 # test shapes with b_buffer
 --reset
 --dt=f16:f16:f16 --stag=ab --wtag=ba --dtag=ab --bia-dt=f16 327x256:256x256
+
+# Test that cases when M == 1 are handled correctly.
+--reset
+--stag=ba,ab --wtag=ab --dtag=ab --dt=f16 1x2:2x256
diff --git a/tests/benchdnn/inputs/matmul/harness_matmul_regression_int8 b/tests/benchdnn/inputs/matmul/harness_matmul_regression_int8
@@ -26,3 +26,7 @@
 --dt=s8:s8:f32
 --stag=ab --wtag=ab --dtag=ba,ab
 1x1:1x200
+
+# Test that cases when M == 1 are handled correctly.
+--reset
+--stag=ba,ab --wtag=ab --dtag=ab --dt=s8:s8:f32 1x2:2x256