ColossalAI/colossalai/auto_parallel/solver
Frank Lee 9dae9bb2bc
[autoparallel] introduced baseclass for op handler and reduced code redundancy (#1471)
* [autoparallel] introduced baseclass for op handler and reduced code redundancy

* polish code
2022-08-19 16:51:38 +08:00
..
__init__.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00
conv_handler.py [autoparallel] introduced baseclass for op handler and reduced code redundancy (#1471) 2022-08-19 16:51:38 +08:00
dot_handler.py [autoparallel] introduced baseclass for op handler and reduced code redundancy (#1471) 2022-08-19 16:51:38 +08:00
operator_handler.py [autoparallel] introduced baseclass for op handler and reduced code redundancy (#1471) 2022-08-19 16:51:38 +08:00
sharding_strategy.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00