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,