ColossalAI/colossalai/auto_parallel/solver
Frank Lee d39e11dffb
[autoparallel] added namespace constraints (#1490)
2022-08-24 15:44:07 +08:00
..
__init__.py [autoparallel] integrate auto parallel with torch fx (#1479) 2022-08-23 14:23:08 +08:00
conv_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
dot_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
operator_handler.py [autoparallel] added namespace constraints (#1490) 2022-08-24 15:44:07 +08:00
sharding_strategy.py [autoparallel] integrate auto parallel with torch fx (#1479) 2022-08-23 14:23:08 +08:00