biondizzle/nvfp4-megamoe-kernel
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Fix: self.mma_tiler_mnk not mma_tiler_mnk

Browse Source
This commit is contained in:
biondizzle
2026-06-01 10:49:05 +00:00
parent c082843ecc
commit be2bb2fe84
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
dsv4/kernels/router/nvfp4_fused_router_kernel.py
View File

@@ -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],