github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-24 03:03:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] ignore 8 gpu test (#1080)

Browse Source 
* [test] ignore 8 gpu test

* polish code

* polish workflow

* polish workflow
This commit is contained in:
Frank Lee
2022-06-08 23:14:18 +08:00
committed by GitHub
parent 0653c63eaa
commit 65ee6dcc20
11 changed files with 15 additions and 8 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
tests/test_utils/test_checkpoint/test_checkpoint_1d.py
View File

@@ -67,6 +67,7 @@ def check_checkpoint_1d(rank, world_size, port):
@pytest.mark.dist
@pytest.mark.skip("This test should be invoked with 8 GPUs")
@rerun_on_exception(exception_type=mp.ProcessRaisedException, pattern=".*Address already in use.*")
def test_checkpoint_1d():
world_size = 8

1
tests/test_utils/test_checkpoint/test_checkpoint_2d.py
View File

@@ -67,6 +67,7 @@ def check_checkpoint_2d(rank, world_size, port):
@pytest.mark.dist
@pytest.mark.skip("This test should be invoked with 8 GPUs")
@rerun_on_exception(exception_type=mp.ProcessRaisedException, pattern=".*Address already in use.*")
def test_checkpoint_2d():
world_size = 8

1
tests/test_utils/test_checkpoint/test_checkpoint_2p5d.py
View File

@@ -67,6 +67,7 @@ def check_checkpoint_2p5d(rank, world_size, port):
@pytest.mark.dist
@pytest.mark.skip("This test should be invoked with 8 GPUs")
@rerun_on_exception(exception_type=mp.ProcessRaisedException, pattern=".*Address already in use.*")
def test_checkpoint_2p5d():
world_size = 8

1
tests/test_utils/test_checkpoint/test_checkpoint_3d.py
View File

@@ -67,6 +67,7 @@ def check_checkpoint_3d(rank, world_size, port):
@pytest.mark.dist
@pytest.mark.skip("This test requires 8 GPUs to execute")
@rerun_on_exception(exception_type=mp.ProcessRaisedException, pattern=".*Address already in use.*")
def test_checkpoint_3d():
world_size = 8