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 [hotfix] add shard dim to aviod backward communication error (#2954) 2023-03-01 11:41:53 +08:00
solver [autoparallel] apply repeat block to reduce solving time (#2912) 2023-02-28 11:03:30 +08:00
utils [autoparallel] find repeat blocks (#2854) 2023-02-23 17:28:19 +08:00
__init__.py
constants.py
initialize.py [autoparallel] apply repeat block to reduce solving time (#2912) 2023-02-28 11:03:30 +08:00
options.py [autoparallel] add shard option (#2696) 2023-02-15 13:48:28 +08:00
sharding_strategy.py