ColossalAI/colossalai/auto_parallel/solver/strategy
YuliangLiu0306 517b63939a
[autoparallel] add unary element wise handler v2 (#1674)
2022-10-09 17:30:42 +08:00
..
__init__.py [autoparallel] add unary element wise handler v2 (#1674) 2022-10-09 17:30:42 +08:00
batch_norm_generator.py [autoparallel] add following node generator (#1673) 2022-10-09 14:49:18 +08:00
conv_strategy_generator.py [autoparallel] add following node generator (#1673) 2022-10-09 14:49:18 +08:00
getitem_generator.py [autoparallel] add following node generator (#1673) 2022-10-09 14:49:18 +08:00
layer_norm_generator.py [autoparallel] add layer norm handler v2 (#1671) 2022-10-09 14:23:22 +08:00
matmul_strategy_generator.py [autoparallel] added bias comm spec to matmul strategy (#1664) 2022-09-29 11:08:05 +08:00
strategy_generator.py [autoparallel] add following node generator (#1673) 2022-10-09 14:49:18 +08:00
unary_elementwise_generator.py [autoparallel] add unary element wise handler v2 (#1674) 2022-10-09 17:30:42 +08:00