From 183c6d7cd58015e80d513c04985801df4007f582 Mon Sep 17 00:00:00 2001 From: Daniel Youssif Date: Wed, 18 Dec 2024 00:34:53 +0000 Subject: [PATCH 1/2] xe: jit: conv: cast mask to s16 before mad --- src/gpu/intel/jit/conv/zp_plan.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/gpu/intel/jit/conv/zp_plan.cpp b/src/gpu/intel/jit/conv/zp_plan.cpp index f65eefcb28f..64c277ae7c2 100644 --- a/src/gpu/intel/jit/conv/zp_plan.cpp +++ b/src/gpu/intel/jit/conv/zp_plan.cpp @@ -1268,7 +1268,7 @@ class zp_comp_apply_plan_t : public base_plan_t { const expr_t &c_buf, const split_dispatcher_t &sd, int subtile_idx) const { const auto comp_type = comp_layout_.type(); - const auto mask_type = mask_layout_.type(); + const auto mask_type = type_t::s16(); const dim_t kw_dim = comp_layout_.dim(comp_kw_idx_); std::vector comp_off; std::vector mask_off; From 39d01dcc6d499352ef8579c7d413eb7c38403cdb Mon Sep 17 00:00:00 2001 From: Daniel Youssif Date: Wed, 18 Dec 2024 19:31:27 +0000 Subject: [PATCH 2/2] xe: jit: gemm: align kxq lookahead with kxq_load --- src/gpu/intel/jit/gemm/generator/pieces/k_loop.cxx | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/gpu/intel/jit/gemm/generator/pieces/k_loop.cxx b/src/gpu/intel/jit/gemm/generator/pieces/k_loop.cxx index 99a107bcea2..6db0fbdfba3 100644 --- a/src/gpu/intel/jit/gemm/generator/pieces/k_loop.cxx +++ b/src/gpu/intel/jit/gemm/generator/pieces/k_loop.cxx @@ -960,8 +960,8 @@ void BLASKernelGenerator::kLoop(KLoop type, const GEMMProblem &problem, GEMM ls.swapLast2(); // A/B 2D quantization parameter loads. - auto reqLoadAq = every(kaq_load) | lookahead(ka_repackMain); - auto reqLoadBq = every(kbq_load) | lookahead(kb_loadMain); + auto reqLoadAq = every(kaq_load) | lookahead(kaq_load); + auto reqLoadBq = every(kbq_load) | lookahead(kbq_load); auto reqLoadAqLate = every(kaq_loadLate) | lookahead(kaq_loadLate); auto reqLoadBqLate = every(kbq_loadLate) | lookahead(kbq_loadLate);