[Bugfix] Fix Nemotron ForwardFlags across custom op boundary (#30968)

2026-07-12 22:10:23 -07:00
parent 4cec9ef9d7
commit cbcbef6811
2 changed files with 11 additions and 8 deletions
@@ -27,11 +27,6 @@ class TestNvidiaNemotron3Nano30BFP8(LMEvalMixin, DefaultServerBase):
    other_args = [
        "--tp-size",
        "2",
-        # FlashInfer trtllm allreduce fusion on flashinfer 0.6.14 degrades
-        # gsm8k for this model (strict-match 0.84 -> 0.78); the ground truth
-        # was calibrated with fusion disabled. Keep it off until the fusion
-        # numerics regression is resolved.
-        "--enforce-disable-flashinfer-allreduce-fusion",
    ] + NEMOTRON_3_NANO_THINKING_ARGS