xe: jit: gemm: fixup post-ops k-parallel skip

kealan-barbieri · kealan-barbieri · commit 04c22c96de07 · 2025-09-02T20:37:35.000-07:00
diff --git a/src/gpu/intel/jit/gemm/jit_gemm_pd.cpp b/src/gpu/intel/jit/gemm/jit_gemm_pd.cpp
@@ -91,6 +91,7 @@ status_t jit_gemm_pd_t::init_post_ops() {
                                 &e.binary.src1_desc);
                 binary_srcs_.push_back(
                         binary_src_t {binary_src_t::binary, int(i)});
+                non_scale_po_ = true;
                 break;
             case sum:
                 ok &= !with_sum_;
@@ -102,6 +103,7 @@ status_t jit_gemm_pd_t::init_post_ops() {
             case eltwise:
                 ok &= eltwise_injector_f32_is_supported(e.eltwise.alg);
                 binary_srcs_.push_back(binary_src_t {binary_src_t::none, 0});
+                non_scale_po_ = true;
                 break;
             case prelu:
                 binary_srcs_.push_back(
@@ -111,10 +113,10 @@ status_t jit_gemm_pd_t::init_post_ops() {
                         == status::success;
                 prelu_count++;
                 ok &= prelu_count <= 1;
+                non_scale_po_ = true;
                 break;
             default: return status::unimplemented;
         }
-        non_scale_po_ = true;
     }
 
     if (!ok) return status::unimplemented;
