From b97f30e2899dd8088d911342355c40a8a91d9248 Mon Sep 17 00:00:00 2001 From: biondizzle Date: Mon, 1 Jun 2026 06:56:33 +0000 Subject: [PATCH] fix: store sf_vec_size as instance variable --- dsv4/kernels/router/nvfp4_fused_router_kernel.py | 1 + 1 file changed, 1 insertion(+) diff --git a/dsv4/kernels/router/nvfp4_fused_router_kernel.py b/dsv4/kernels/router/nvfp4_fused_router_kernel.py index 920600ac..398422f6 100644 --- a/dsv4/kernels/router/nvfp4_fused_router_kernel.py +++ b/dsv4/kernels/router/nvfp4_fused_router_kernel.py @@ -70,6 +70,7 @@ class Nvfp4FusedRouterKernel: self.mma_tiler_mn = mma_tiler_mn self.cluster_shape_mn = cluster_shape_mn self.top_k = top_k + self.sf_vec_size = sf_vec_size self.use_2cta_instrs = mma_tiler_mn[0] == 256 self.cta_group = tcgen05.CtaGroup.TWO if self.use_2cta_instrs else tcgen05.CtaGroup.ONE