From c339fe7ad9cf57c1afa99f3d26e3a08f7900c769 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 00:54:19 +0000 Subject: [PATCH] =?UTF-8?q?fix:=20router=20A=20operand=20major=20mode=20MN?= =?UTF-8?q?=20(not=20K)=20=E2=80=94=20fixes=20CuTeDSL=20local=5Ftile=20coo?= =?UTF-8?q?rd=20error?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- dsv4/kernels/router/dense_router_decode_kernel.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/dsv4/kernels/router/dense_router_decode_kernel.py b/dsv4/kernels/router/dense_router_decode_kernel.py index 43e9d651..e537406e 100644 --- a/dsv4/kernels/router/dense_router_decode_kernel.py +++ b/dsv4/kernels/router/dense_router_decode_kernel.py @@ -101,7 +101,9 @@ class DenseRouterDecodeKernel: self.num_tmem_alloc_cols = utils.get_num_tmem_alloc_cols(tCtAcc_fake) def run(self, X, W_gate, e_bias, out_w, out_ids, M, E, K, scaling, top_k, stream=None): - self.a_major_mode = OperandMajorMode.K + # A operand (hidden_states) is (M, K) — MN-major layout + # B operand (W_gate) is (K, E) — K-major layout + self.a_major_mode = OperandMajorMode.MN self.b_major_mode = OperandMajorMode.K if stream is None: