Sync the changes in #23402 (#29063)

Co-authored-by: Brayden Zhong <brayden@radixark.ai>
2026-06-24 17:01:02 -07:00
parent f82addd4a8
commit 5d4e63d49e
2 changed files with 4 additions and 2 deletions
@@ -80,11 +80,11 @@ class TestFlashInferCommFusion(unittest.TestCase):
            flashinfer_allreduce_fusion_backend="auto", nnodes=2
        )

-        # Blackwell: trtllm on single-node, mnnvl on multi-node.
+        # Blackwell: mnnvl on both single-node and multi-node.
        with patch.object(fusion, "is_sm100_supported", return_value=True):
            self.assertEqual(
                fusion.resolve_flashinfer_allreduce_fusion_backend(single_node),
-                "trtllm",
+                "mnnvl",
            )
            self.assertEqual(
                fusion.resolve_flashinfer_allreduce_fusion_backend(multi_node), "mnnvl"