diff --git a/src/graph/backend/graph_compiler/core/src/runtime/microkernel/cpu/brgemm_onednn.cpp b/src/graph/backend/graph_compiler/core/src/runtime/microkernel/cpu/brgemm_onednn.cpp index 40683e09ea8..9e9596d8425 100644 --- a/src/graph/backend/graph_compiler/core/src/runtime/microkernel/cpu/brgemm_onednn.cpp +++ b/src/graph/backend/graph_compiler/core/src/runtime/microkernel/cpu/brgemm_onednn.cpp @@ -125,7 +125,7 @@ static brgemm_attr_t get_dnnl_brgemm_attrs(const attrs_setting_t &attrs) { = static_cast(it.second); break; case attr_key::wary_tail_read: - dnnl_attrs.wary_tail_read = static_cast(it.second); + dnnl_attrs.wary_A_k_tail_read = static_cast(it.second); break; case attr_key::generate_skip_accumulation: dnnl_attrs.generate_skip_accumulation