From 30c17b96b25e0a9f8f7918249c9f4646c06de3d2 Mon Sep 17 00:00:00 2001 From: Denis Samoilov Date: Wed, 29 Jan 2025 22:31:34 -0800 Subject: [PATCH] benchdnn: add regression tests for cases when M == 1 --- tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16 | 4 ++++ .../benchdnn/inputs/matmul/harness_matmul_regression_float16 | 4 ++++ tests/benchdnn/inputs/matmul/harness_matmul_regression_int8 | 4 ++++ 3 files changed, 12 insertions(+) diff --git a/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16 b/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16 index bab5bdf03e7..be37bc934db 100644 --- 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 index 56badf32744..5145a18340f 100644 --- 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 index d31d3e82bc2..c696837a2f4 100644 --- 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