Skip to content

GitLab

Unverified Commit a98e16ed authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

Merge pull request #3926 from hpcaitech/feature/dtensor 

[feature] updated device mesh and dtensor
parents 12c90db3 eb39154d
Show whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
tests/test_tensor/test_sharding_spec.py
View file @ a98e16ed
......@@ -5,7 +5,7 @@ from colossalai.tensor.sharding_spec import ShardingSpec, _DimSpec
def test_sharding_spec():
physical_mesh_id = torch.arange(0, 16).reshape(2, 8)
physical_mesh_id = torch.arange(0, 16)
mesh_shape = (4, 4)
# [[0, 1, 2, 3],
# [4, 5, 6, 7],
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment