github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-10-22 23:32:37 +00:00
Code Issues Packages Projects Releases Wiki Activity

added CI for unit testing (#69)

Browse Source
This commit is contained in:
Frank Lee
2021-12-16 10:32:08 +08:00
committed by GitHub
parent 45355a62f7
commit cd9c28e055
68 changed files with 1089 additions and 766 deletions
Download Patch File Download Diff File Expand all files Collapse all files

7
tests/test_utils/test_gradient_accumluation.py
View File

@@ -104,13 +104,14 @@ def run_no_pipeline(rank, world_size):
'param should be the same in the first few iterations and only changed in the last iteration'
gpc.destroy()
torch.cuda.empty_cache()
@pytest.mark.skip("This test should be invoked using the test.sh provided")
@pytest.mark.dist
def test_engine():
func = partial(run_no_pipeline, world_size=4)
mp.spawn(func, nprocs=4)
world_size = 4
func = partial(run_no_pipeline, world_size=world_size)
mp.spawn(func, nprocs=world_size)
if __name__ == '__main__':