Fix B200 Nightly tests and move one manual test back to unit test to prevent the same issue (#13746)
This commit is contained in:
@@ -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,
|
||||
),
|
||||
)
|
||||
|
||||
|
||||
Reference in New Issue
Block a user