Skip to content

Commit 882685a

Browse files
committed
update test
Signed-off-by: cascade812 <[email protected]>
1 parent 3b29755 commit 882685a

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

tests/compile/test_async_tp.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -199,9 +199,8 @@ def test_async_tp_pass_correctness(
199199
pytest.skip(f"Need at least {tp_size} x {pp_size} GPUs")
200200

201201
common_args = [
202-
# use half precision for speed and memory savings in CI environment
203202
"--dtype",
204-
"float16",
203+
"bfloat16",
205204
"--max-model-len",
206205
"2048",
207206
"--max-num-seqs",

0 commit comments

Comments
 (0)