cpu: x64: binary injector: add missing entry for per_mb_spatial

tests: benchdnn: inputs: matmul: add regression test
oneapi-src · Jun 23, 2023 · 04b180b · 04b180b
1 parent 6beb64a
commit 04b180b
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 0 deletions.
diff --git a/src/cpu/x64/injectors/jit_uni_binary_injector.cpp b/src/cpu/x64/injectors/jit_uni_binary_injector.cpp
@@ -32,6 +32,7 @@ static bcast_set_t get_all_strategies_supported_by_injector() {
     return bcast_set_t {broadcasting_strategy_t::scalar,
             broadcasting_strategy_t::per_oc,
             broadcasting_strategy_t::per_oc_spatial,
+            broadcasting_strategy_t::per_mb_spatial,
             broadcasting_strategy_t::per_mb_w, broadcasting_strategy_t::per_w,
             broadcasting_strategy_t::no_broadcast};
 }

diff --git a/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16 b/tests/benchdnn/inputs/matmul/harness_matmul_regression_bf16
@@ -5,3 +5,10 @@
 --runtime_dims_masks=1:0
 --attr-post-ops=add:bf16:per_oc,add:f32:per_oc
 10x1:1x20_n"runtime_m_and_binary_po"
+
+# small shape with binary po per_dim_023 broadcast
+--reset
+--dt=bf16
+--stag=abcd --wtag=abcd --dtag=abcd
+--attr-post-ops=add:bf16:per_dim_023:abcd
+2x2x32x16:2x2x16x64"small_shape_with_per_dim_023_binary_po"