opencl: generalize Adreno MoE kernels on M (#23449)

2026-05-22 17:08:41 -07:00
parent 1acee6bf89
commit 0f3cb3fc8b
18 changed files with 145 additions and 17 deletions
@@ -163,7 +163,7 @@ kernel void kernel_gemm_moe_mxfp4_f32_ns(
    uint block_id_n = get_global_id(2); // n_tile

    // Boundary check
-    if (((get_global_id(0) + block_id_m * TILESIZE_M) >= ne01) || (block_id_n >= total_tiles[0])) {
+    if (block_id_n >= total_tiles[0]) {
        return;
    }

@@ -248,6 +248,10 @@ kernel void kernel_gemm_moe_mxfp4_f32_ns(
        dotx16_reduce8(reg_a, shared_b, reg_c.hi, 16);
    }

+    if ((get_global_id(0) + block_id_m * TILESIZE_M) >= ne01) {
+        return;
+    }
+
    // Load poster router and share in LM
    __local uint out_idx[TILESIZE_N];