Simplify eagle tests and TP sync in grammar backend (#4066)

2025-03-04 13:40:40 -08:00
parent 03b0364f76
commit 77a3954bf7
14 changed files with 122 additions and 126 deletions
--- a/test/srt/test_triton_attention_backend.py
+++ b/test/srt/test_triton_attention_backend.py
@@ -26,6 +26,8 @@ class TestTritonAttnBackend(unittest.TestCase):
                "--attention-backend",
                "triton",
                "--enable-torch-compile",
+                "--cuda-graph-max-bs",
+                16,
            ],
        )