ColossalAI/colossalai/auto_parallel/solver/strategy
YuliangLiu0306 451cd72dea
[autoparallel] adapt runtime passes (#1703)
* [autoparallel] adapt runtime passes v2

* polish code
2022-10-14 10:14:07 +08:00
..
__init__.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
batch_norm_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
conv_strategy_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
getitem_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
layer_norm_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
matmul_strategy_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
normal_pooling_generator.py [autoparallel] adapt runtime passes (#1703) 2022-10-14 10:14:07 +08:00
output_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
placeholder_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
reshape_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
strategy_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
unary_elementwise_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00
where_generator.py [autoparallel] collated all deprecated files (#1700) 2022-10-13 18:24:11 +08:00