ColossalAI/colossalai/auto_parallel/tensor_shard
YuliangLiu0306 1dc003c169
[autoparallel] distinguish different parallel strategies (#2699)
2023-02-15 22:28:28 +08:00
..
node_handler [autoparallel] distinguish different parallel strategies (#2699) 2023-02-15 22:28:28 +08:00
solver [autoparallel] add shard option (#2696) 2023-02-15 13:48:28 +08:00
utils
__init__.py
constants.py
initialize.py
options.py
sharding_strategy.py