ColossalAI/colossalai/auto_parallel/tensor_shard
YuliangLiu0306 47fb214b3b
[hotfix] add shard dim to aviod backward communication error (#2954)
2023-03-01 11:41:53 +08:00
..
node_handler
solver
utils
__init__.py
constants.py
initialize.py
options.py
sharding_strategy.py