github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-05 19:13:01 +00:00
Code Issues Packages Projects Releases Wiki Activity

[test] fix gemini checkpoint and gpt test (#4620)

Browse Source
This commit is contained in:
Hongxin Liu
2023-09-05 16:02:23 +08:00
committed by GitHub
parent e71d245293
commit bd18678478
2 changed files with 2 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
tests/test_shardformer/test_model/test_shard_gpt2.py
View File

@@ -102,7 +102,6 @@ def check_forward_backward(model_fn, data_gen_fn, output_transform_fn, loss_fn,
torch.cuda.empty_cache()
@pytest.mark.skip(reason="This test will hang in CI")
@parameterize('test_config', [{
'tp_size': 2,
'pp_size': 2,
@@ -220,7 +219,7 @@ def check_gpt2_3d(rank, world_size, port):
run_gpt2_3d_test()
@pytest.mark.skip(reason="This test will hang in CI")
@pytest.mark.dist
@rerun_if_address_is_in_use()
@clear_cache_before_run()