cpu: x64: binary injector: add rhs_addr_cache_reg to forked/diverged …

…kernels
openvinotoolkit · Nov 16, 2022 · 93655de · 93655de
1 parent 4a49ca7
commit 93655de
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/src/cpu/x64/jit_avx512_core_fork_bf16_dw_conv_kernel.cpp b/src/cpu/x64/jit_avx512_core_fork_bf16_dw_conv_kernel.cpp
@@ -591,7 +591,7 @@ void jit_avx512_fork_dw_conv_fwd_kernel_bf16::generate() {
                 % (cpu_isa_traits<avx512_core>::vlen / sizeof(float));
         static constexpr bool use_exact_tail_scalar_bcast = false;
         const binary_injector::rhs_arg_static_params_t rhs_sp {
-            helper_vmm_idx, r10, r11, preserve_gpr,
+            helper_vmm_idx, r10, r11, r12, preserve_gpr,
             preserve_vmm, GET_OFF(post_ops_binary_rhs_arg_vec),
             GET_OFF(dst_orig), memory_desc_wrapper(&dst_md_),
             tail_size, k_oc_tail_mask, use_exact_tail_scalar_bcast};

diff --git a/src/cpu/x64/jit_gemm_convolution_utils.cpp b/src/cpu/x64/jit_gemm_convolution_utils.cpp
@@ -66,7 +66,7 @@ struct jit_pp_kernel_t : pp_kernel_t, public jit_generator {
             static constexpr size_t tail_size = 0;
             static constexpr bool use_exact_tail_scalar_bcast = false;
             const binary_injector::rhs_arg_static_params_t rhs_sp {
-                helper_vmm_idx, r13, r14, preserve_gpr,
+                helper_vmm_idx, r13, r14, r15, preserve_gpr,
                 preserve_vmm, PARAM_OFF(post_ops_binary_rhs_arg_vec),
                 PARAM_OFF(dst_orig), memory_desc_wrapper(pd->dst_md()),
                 tail_size, kreg_rem_mask, use_exact_tail_scalar_bcast};

diff --git a/src/cpu/x64/jit_uni_fork_dw_conv_kernel_f32.cpp b/src/cpu/x64/jit_uni_fork_dw_conv_kernel_f32.cpp
@@ -772,7 +772,7 @@ void jit_uni_fork_dw_conv_fwd_kernel_f32<isa>::generate() {
                 % (cpu_isa_traits<isa>::vlen / sizeof(float));
         static constexpr bool use_exact_tail_scalar_bcast = false;
         const binary_injector::rhs_arg_static_params_t rhs_sp {
-            helper_vmm_idx, r10, r11, preserve_gpr,
+            helper_vmm_idx, r10, r11, r12, preserve_gpr,
             preserve_vmm, GET_OFF(post_ops_binary_rhs_arg_vec),
             GET_OFF(dst_orig), memory_desc_wrapper(&dst_md_),
             tail_size, k_oc_tail_mask, use_exact_tail_scalar_bcast};