Fix: PersistentTileSchedulerParams cluster_shape must be Python ints not MLIR values

2026-06-01 09:38:08 +00:00
parent 28d0cb4f41
commit ec79f30709
1 changed files with 1 additions and 1 deletions
--- a/dsv4/kernels/router/nvfp4_fused_router_kernel.py
+++ b/dsv4/kernels/router/nvfp4_fused_router_kernel.py
@@ -308,7 +308,7 @@ class Nvfp4FusedRouterKernel:

            tile_sched_params = utils.PersistentTileSchedulerParams(
                (cutlass.Int32(num_M_tiles), cutlass.Int32(num_N_tiles), cutlass.Int32(1)),
-                (cutlass.Int32(self.cluster_shape_mn[0]), cutlass.Int32(self.cluster_shape_mn[1]), cutlass.Int32(1)))
+                (1, 1, 1))

            self._kernel(
                tiled_mma, tiled_mma_sfb,