ColossalAI/colossalai/auto_parallel/tensor_shard
YuliangLiu0306 cd0af9f7f6
[autoparallel] gpt2lp runtimee test (#2113)
2022-12-12 18:06:40 +08:00
..
deprecated
node_handler
solver [autoparallel] gpt2lp runtimee test (#2113) 2022-12-12 18:06:40 +08:00
utils
__init__.py
constants.py
sharding_strategy.py