github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

[fix] update

pull/6034/head
duanjunwen 2024-08-28 02:33:42 +00:00
parent fe209164f1
commit 29383b2de0
1 changed files with 0 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_pipeline/test_schedule/test_zerobubble_pp.py
View File

@ -616,10 +616,6 @@ def test_run_fwd_bwd_with_vschedule(
# @pytest.mark.parametrize("num_model_chunk", [2]) # @pytest.mark.parametrize("num_model_chunk", [2])
@rerun_if_address_is_in_use() @rerun_if_address_is_in_use()
def test_pp(): def test_pp():
# spawn(
# test_run_fwd_bwd_iter_input,
# nprocs=4,
# )
spawn( spawn(
test_run_fwd_bwd_with_vschedule, test_run_fwd_bwd_with_vschedule,
nprocs=4, nprocs=4,