move eagle draft post process to cuda graph (#11434)

Co-authored-by: Lianmin Zheng <lianminzheng@gmail.com>
2025-10-14 16:50:53 +02:00
parent dc965db0e0
commit 56222658ec
8 changed files with 469 additions and 549 deletions
--- a/test/srt/run_suite.py
+++ b/test/srt/run_suite.py
@@ -69,6 +69,7 @@ suites = {
        TestFile("test_chunked_prefill.py", 313),
        TestFile("test_create_kvindices.py", 2),
        TestFile("test_deterministic.py", 300),
+        TestFile("test_build_eagle_tree.py", 8),
        TestFile("test_eagle_infer_a.py", 370),
        TestFile("test_eagle_infer_b.py", 700),
        TestFile("test_eagle_infer_beta.py", 300),