Skip to content

Commit e28a35a

Browse files
committed
cpu: x64: matmul: set wei_tag any layout into plain layout for int4 weights
1 parent 0cf9a8a commit e28a35a

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/cpu/x64/matmul/brgemm_matmul_utils.cpp

+2
Original file line numberDiff line numberDiff line change
@@ -296,6 +296,7 @@ status_t brgemm_matmul_conf_utils_t::set_or_check_B_tag(memory_desc_t &B_md,
296296
? get_default_n_block(format_tag::undef)
297297
: bgmmc.N_blk;
298298
bgmmc.wei_tag = blocked_B_layouts_allowed && !bgmmc.is_runtime_N
299+
&& !bgmmc.is_int4_weights
299300
? this->pick_blocked_B_layout(default_n_block)
300301
: plain_tensor_layout_tag;
301302
VCONDCHECK_BG(
@@ -311,6 +312,7 @@ status_t brgemm_matmul_conf_utils_t::set_or_check_B_tag(memory_desc_t &B_md,
311312
}
312313
} else {
313314
bgmmc.wei_tag = blocked_B_layouts_allowed && !bgmmc.is_runtime_N
315+
&& !bgmmc.is_int4_weights
314316
? memory_desc_matches_one_of_tag(B_md, plain_tensor_layout_tag,
315317
transposed_tensor_layout_tag, blocked_64n_B_layout_tag,
316318
blocked_48n_B_layout_tag, blocked_32n_B_layout_tag,

0 commit comments

Comments
 (0)