From 7967e854da4868872b4b13d2b7f039061fee50fe Mon Sep 17 00:00:00 2001 From: Richard Zou Date: Tue, 17 Feb 2026 12:07:56 -0500 Subject: [PATCH] [BugFix] Fix sp tests (#34716) Signed-off-by: Richard Zou --- tests/compile/correctness_e2e/test_sequence_parallel.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/compile/correctness_e2e/test_sequence_parallel.py b/tests/compile/correctness_e2e/test_sequence_parallel.py index 6c084f603..281ffbfd2 100644 --- a/tests/compile/correctness_e2e/test_sequence_parallel.py +++ b/tests/compile/correctness_e2e/test_sequence_parallel.py @@ -229,7 +229,7 @@ def _compare_sp( if chunked_prefill: common_args.append("--enable-chunked-prefill") if eager_mode: - common_args.append("--enforce-eager") + common_args.append("-cc.cudagraph_mode=none") if runner != "auto": common_args.extend(["--runner", runner]) if trust_remote_code: