[AMD][MORI] Fix MTP crash with FP4/FP8 dispatch and add NEXTN dispatch env vars. (#20453)

2026-03-13 14:03:17 -07:00
parent c37ef7f18b
commit 0eea80bc00
6 changed files with 46 additions and 24 deletions
@@ -403,6 +403,7 @@ class DeepseekV2MoE(nn.Module):
                config, "routing_method_type", RoutingMethodType.DeepSeekV3
            ),
            prefix=add_prefix("experts", prefix),
+            is_nextn=is_nextn,
        )

        self.topk = TopK(