Skip to content

GitLab

Unverified Commit 7967e854 authored by Richard Zou's avatar Richard Zou Committed by GitHub
Browse files

[BugFix] Fix sp tests (#34716) 


Signed-off-by: default avatarRichard Zou <zou3519@gmail.com>
parent 6bd6d0c3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/compile/correctness_e2e/test_sequence_parallel.py
View file @ 7967e854
...@@ -229,7 +229,7 @@ def _compare_sp( ...@@ -229,7 +229,7 @@ def _compare_sp(
if chunked_prefill: if chunked_prefill:
common_args.append("--enable-chunked-prefill") common_args.append("--enable-chunked-prefill")
if eager_mode: if eager_mode:
common_args.append("--enforce-eager") common_args.append("-cc.cudagraph_mode=none")
if runner != "auto": if runner != "auto":
common_args.extend(["--runner", runner]) common_args.extend(["--runner", runner])
if trust_remote_code: if trust_remote_code:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment