ColossalAI/colossalai/auto_parallel/solver/strategy
YuliangLiu0306 af718e83f2
[autoparallel] add reshape handler v2 and fix some previous bug (#1683)
2022-10-11 18:12:59 +08:00
..
__init__.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
batch_norm_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
conv_strategy_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
getitem_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
layer_norm_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
matmul_strategy_generator.py [autoparallel] added bias comm spec to matmul strategy (#1664) 2022-09-29 11:08:05 +08:00
reshape_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
strategy_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00
unary_elementwise_generator.py [autoparallel] add reshape handler v2 and fix some previous bug (#1683) 2022-10-11 18:12:59 +08:00