xe: jit: gemm: fix SLM remask condition for zp

oneapi-src · Jul 12, 2024 · 604f27f · 604f27f
1 parent 5897aa0
commit 604f27f
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/gpu/intel/jit/gemm/gen_gemm_kernel_generator.cpp b/src/gpu/intel/jit/gemm/gen_gemm_kernel_generator.cpp
@@ -14756,7 +14756,8 @@ void gemm_kernel_generator_t<hw>::kLoopActivateSLMRemainder(bool active,
         loadMasks(kMasksBi, remsBi, offsetsBi, strategy, state);
     }
 
-    bool mayAccessAllK = (minOPCount > 1) || problem.sumA || problem.sumB;
+    bool mayAccessAllK
+            = (minOPCount > 1) || problem.needsASums() || problem.needsBSums();
     bool asIfMaskedAi = Ai_lateKRem && state.Ai_strategy.padded;
     bool asIfMaskedBi = Bi_lateKRem && state.Bi_strategy.padded;
     slmRemaskA = slmA && mayAccessAllK && !Ai_remIncrCopy