github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-07 20:10:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

[devops] fix compatibility (#5444)

Browse Source 
* [devops] fix compatibility

* [hotfix] update compatibility test on pr

* [devops] fix compatibility

* [devops] record duration during comp test

* [test] decrease test duration

* fix falcon
This commit is contained in:
Hongxin Liu
2024-03-13 15:24:13 +08:00
committed by GitHub
parent 385e85afd4
commit f2e8b9ef9f
9 changed files with 15 additions and 40 deletions
Download Patch File Download Diff File Expand all files Collapse all files

10
tests/test_checkpoint_io/test_gemini_checkpoint_io.py
View File

@@ -16,7 +16,6 @@ from colossalai.testing import (
clear_cache_before_run,
parameterize,
rerun_if_address_is_in_use,
skip_if_not_enough_gpus,
spawn,
)
from tests.kit.model_zoo import model_zoo
@@ -178,12 +177,5 @@ def test_gemini_ckpIO():
spawn(run_dist, 4)
@pytest.mark.largedist
@skip_if_not_enough_gpus(min_gpus=8)
@rerun_if_address_is_in_use()
def test_gemini_ckpIO_3d():
spawn(run_dist, 8)
if __name__ == "__main__":
test_gemini_ckpIO()
test_gemini_ckpIO()