cpu: aarch64: fix scratchpad memory initialization

aditew01 · vpirogov · commit 218b41ddb3e9 · 2025-08-27T17:26:50.000-07:00
diff --git a/src/cpu/aarch64/jit_uni_dw_conv_kernel_utils.hpp b/src/cpu/aarch64/jit_uni_dw_conv_kernel_utils.hpp
@@ -281,8 +281,11 @@ template <cpu_isa_t isa, data_type_t kernel_dt>
 void jit_uni_dw_conv_fwd_kernel_t<isa, kernel_dt>::init_scratchpad(
         memory_tracking::registrar_t &scratchpad, const jit_conv_conf_t &jcp) {
     using namespace dnnl::impl::memory_tracking::names;
-    if (jcp.with_bias && jcp.oc_without_padding != jcp.oc)
+    if (jcp.with_bias && jcp.bia_dt == data_type::bf16) {
+        scratchpad.book<float>(key_conv_bias_bf16_convert_wsp, jcp.oc);
+    } else if (jcp.with_bias && jcp.oc_without_padding != jcp.oc) {
         scratchpad.book<float>(key_conv_padded_bias, jcp.oc);
+    }
 }
 
 template struct jit_uni_dw_conv_fwd_kernel_t<sve_512, data_type::f32>;
diff --git a/tests/benchdnn/inputs/conv/harness_conv_regression_general b/tests/benchdnn/inputs/conv/harness_conv_regression_general
@@ -412,3 +412,7 @@ mb1_g50ic50oc50_ih28oh26_kh3sh1ph0_n"dw_post-op_ngroups-tail"
 # test peform_outwork for brgemm kernel
 --reset
 --dir=FWD_I --dt=f32:f32:f32 --stag=acdb --wtag=any --dtag=acdb ic48oc134_ih23oh12kh2sh2dh0ph1_iw12ow4kw1sw13dw0pw14_n"brgemm_perform_outwork"
+
+# tests JIT bf16 depthwise convolution with bf16 bias
+--reset
+--dt=bf16 --bia-dt=bf16 --dir=FWD_I g4mb2_ic4oc4_ih6oh4kh3sh1dh0ph0_iw6ow4kw3sw1dw0pw0_n"aarch64_jit_dw_bia_dt_bf16"
