ColossalAI/colossalai/tensor/d_tensor
YuliangLiu0306 258b43317c
[hotfix] layout converting issue (#3188)
2023-03-21 13:24:18 +08:00
..
__init__.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
comm_spec.py [DTensor] refactor CommSpec (#3034) 2023-03-08 10:45:31 +08:00
d_tensor.py [DTensor] refactor dtensor with new components (#3089) 2023-03-14 16:25:47 +08:00
layout.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
layout_converter.py [hotfix] layout converting issue (#3188) 2023-03-21 13:24:18 +08:00
misc.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
sharding_spec.py [DTensor] refactor CommSpec (#3034) 2023-03-08 10:45:31 +08:00
utils.py [DTensor] implement layout converter (#3055) 2023-03-10 09:53:52 +08:00