ColossalAI/colossalai/tensor/d_tensor
Frank Lee ddcf58cacf
Revert "[sync] sync feature/shardformer with develop"
2023-06-09 09:41:27 +08:00
..
__init__.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
comm_spec.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
d_tensor.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
layout.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
layout_converter.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
misc.py [DTensor] refactor sharding spec (#2987) 2023-03-07 11:08:11 +08:00
sharding_spec.py Revert "[sync] sync feature/shardformer with develop" 2023-06-09 09:41:27 +08:00
utils.py [DTensor] implement layout converter (#3055) 2023-03-10 09:53:52 +08:00