From be2bb2fe8489186791422e8fed30022ddf7c6b25 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 10:49:05 +0000 Subject: [PATCH] Fix: self.mma_tiler_mnk not mma_tiler_mnk --- dsv4/kernels/router/nvfp4_fused_router_kernel.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/dsv4/kernels/router/nvfp4_fused_router_kernel.py b/dsv4/kernels/router/nvfp4_fused_router_kernel.py index 53436c6d..cb22ab06 100644 --- a/dsv4/kernels/router/nvfp4_fused_router_kernel.py +++ b/dsv4/kernels/router/nvfp4_fused_router_kernel.py @@ -110,8 +110,8 @@ class Nvfp4FusedRouterKernel: # ── MMA tiler — K is refined in _setup_attributes ── # ── MMA tiler — K is refined in _setup_attributes ── - self.mma_tiler = (mma_tiler_mnk[0], mma_tiler_mnk[1], 1) - self.mma_tiler_sfb = (mma_tiler_mnk[0] // (2 if self.use_2cta_instrs else 1), cute.round_up(mma_tiler_mnk[1], 128), 1) + self.mma_tiler = (self.mma_tiler_mnk[0], self.mma_tiler_mnk[1], 1) + self.mma_tiler_sfb = (self.mma_tiler_mnk[0] // (2 if self.use_2cta_instrs else 1), cute.round_up(self.mma_tiler_mnk[1], 128), 1) self.cta_tile_shape_mnk = ( self.mma_tiler[0] // cute.size(tiled_mma.thr_id.shape), self.mma_tiler[1],