From ec79f3070993566545ad18ea89853e7fafc45cc8 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 09:38:08 +0000 Subject: [PATCH] Fix: PersistentTileSchedulerParams cluster_shape must be Python ints not MLIR values --- dsv4/kernels/router/nvfp4_fused_router_kernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsv4/kernels/router/nvfp4_fused_router_kernel.py b/dsv4/kernels/router/nvfp4_fused_router_kernel.py index 9f88dbec..fed39c77 100644 --- 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,