You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/colossalai/tensor/d_tensor
YuliangLiu0306 258b43317c
[hotfix] layout converting issue (#3188)
2 years ago
..
__init__.py [DTensor] refactor sharding spec (#2987) 2 years ago
comm_spec.py [DTensor] refactor CommSpec (#3034) 2 years ago
d_tensor.py [DTensor] refactor dtensor with new components (#3089) 2 years ago
layout.py [DTensor] refactor sharding spec (#2987) 2 years ago
layout_converter.py [hotfix] layout converting issue (#3188) 2 years ago
misc.py [DTensor] refactor sharding spec (#2987) 2 years ago
sharding_spec.py [DTensor] refactor CommSpec (#3034) 2 years ago
utils.py [DTensor] implement layout converter (#3055) 2 years ago