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 auto checkpointing tests (#3029)

Browse Source 
* [hotfix] skip auto checkpointing tests

* fix test name issue
This commit is contained in:
YuliangLiu0306
2023-03-07 15:50:00 +08:00
committed by GitHub
parent 8fedc8766a
commit 4269196c79
4 changed files with 15 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
tests/test_tensor/test_dtensor/test_sharding_spec.py → tests/test_tensor/test_dtensor/test_dtensor_sharding_spec.py
View File

@@ -4,7 +4,7 @@ from functools import reduce
from colossalai.tensor.d_tensor.sharding_spec import ALLGATHER_COST, SHARD_COST, STEP_PENALTY, ShardingSpec
def test_sharding_spec():
def test_dtensor_sharding_spec():
dims = 4
dim_partition_dict_0 = {0: [0, 1]}
# DistSpec:
@@ -31,4 +31,4 @@ def test_sharding_spec():
if __name__ == '__main__':
test_sharding_spec()
test_dtensor_sharding_spec()