github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-08-06 18:43:58 +00:00
Code Issues Packages Projects Releases Wiki Activity
feature/stable-diffusion
ColossalAI/colossalai/tensor/d_tensor
History
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_converter.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
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