From 1dbc57e2cd4b11a32edb02046ef89949ea080bd9 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Sun, 31 May 2026 23:36:01 +0000 Subject: [PATCH] fix: use mma_tiler_mn in _create_tiled_mma (attribute exists at init time) --- dsv4/kernels/router/dense_router_decode_kernel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/dsv4/kernels/router/dense_router_decode_kernel.py b/dsv4/kernels/router/dense_router_decode_kernel.py index 31a5b326..42216f50 100644 --- 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):