fix: store sf_vec_size as instance variable

2026-06-01 06:56:33 +00:00
parent c225d195ea
commit b97f30e289
1 changed files with 1 additions and 0 deletions
--- 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