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

2026-03-17 01:13:42 -07:00
parent ee106757df
commit 385a35bd11
6 changed files with 49 additions and 25 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(