gpu: jit: gemm: omit unwanted m/n blocking for single wave sizes

oneapi-src · May 19, 2023 · 276cc21 · 276cc21
1 parent cf73e4e
commit 276cc21
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/gpu/jit/gemm/gen_gemm_kernel.cpp b/src/gpu/jit/gemm/gen_gemm_kernel.cpp
@@ -100,6 +100,8 @@ status_t gen_gemm_kernel_desc_t::finalize() {
             if (thread_count <= thread_gpu) {
                 strategy_.persistent = false;
                 strategy_.cWalkOrder = WalkOrder::HW2D;
+                strategy_.blocking[LoopM] = 16777216;
+                strategy_.blocking[LoopN] = 16777216;
             }
         }
     }