github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-16 14:41:53 +00:00
Code Issues Packages Projects Releases Wiki Activity

[hotfix] skip torchaudio tracing test (#3211)

Browse Source 
* [hotfix] skip torchaudio tracing test

* fix lazy init test issue
This commit is contained in:
YuliangLiu0306
2023-03-24 12:15:33 +08:00
committed by GitHub
parent 78fd31f9c1
commit 045afa3ea2
2 changed files with 8 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
tests/test_utils/test_lazy_init/test_distribute.py
View File

@@ -13,7 +13,11 @@ from colossalai.tensor.d_tensor.sharding_spec import ShardingSpec
from colossalai.testing import parameterize, rerun_if_address_is_in_use
from colossalai.utils import free_port
from colossalai.utils.common import print_rank_0
from colossalai.utils.model.experimental import LazyInitContext, LazyTensor, _MyTensor
try:
from colossalai.utils.model.experimental import LazyInitContext, LazyTensor, _MyTensor
except:
pass
from tests.kit.model_zoo import model_zoo
# from utils import assert_dist_model_equal, set_seed