Fix B200 Nightly tests and move one manual test back to unit test to prevent the same issue (#13746)

This commit is contained in:
Kangyan-Zhou
2025-11-21 17:41:12 -08:00
committed by GitHub
parent a56f770277
commit 59b4d7f8d6
3 changed files with 5 additions and 2 deletions

View File

@@ -1037,7 +1037,9 @@ class FlashInferFusedMoE(FusedMoE):
final_hidden_states = self.quant_method.apply_with_router_logits(
layer=self,
dispatch_output=StandardDispatchOutput(
hidden_states=hidden_states, topk_output=topk_output
hidden_states=hidden_states,
hidden_states_scale=None,
topk_output=topk_output,
),
)