Skip to content

Commit f0aed5e

Browse files
committed
xe: jit: gemm: restrict adding new int4 strategies if not aligned
1 parent 2e06896 commit f0aed5e

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/gpu/intel/jit/gemm/gen_gemm_kernel.cpp

+2
Original file line numberDiff line numberDiff line change
@@ -538,6 +538,8 @@ status_t gen_gemm_nocopy_kernel_desc_t::select_kernel(compute::gpu_arch_t arch,
538538
auto add_mode_matches = [&](bool has_mode, const char *(*match)(Type)) {
539539
if (!has_mode) return;
540540
auto &def = base.selector.precisions;
541+
if ((problem_.Ta.isInt4() || problem_.Tb.isInt4()) && k % 2 != 0)
542+
return;
541543
if (match(problem_.Ta)) {
542544
match_params.push_back(base);
543545
match_params.back().selector.precisions[0] = match(problem_.Ta);

0 commit comments

Comments
 (0)