fix: use mma_tiler_mn in _create_tiled_mma (attribute exists at init time)

2026-05-31 23:36:01 +00:00
parent d05dd50bf5
commit 1dbc57e2cd
1 changed files with 1 additions and 1 deletions
--- a/dsv4/kernels/router/dense_router_decode_kernel.py
+++ b/dsv4/kernels/router/dense_router_decode_kernel.py
@@ -60,7 +60,7 @@ class DenseRouterDecodeKernel:
    def _create_tiled_mma(self):
        return utils.sm100.make_trivial_tiled_mma(
            self.a_dtype, self.a_major_mode, self.b_major_mode,
-            self.acc_dtype, self.cta_group, self.mma_tiler[:2],
+            self.acc_dtype, self.cta_group, self.mma_tiler_mn,
        )

    def _setup_attributes(self):