ColossalAI/colossalai/auto_parallel
Frank Lee 3a4d6f63a8
[autoparallel] added node handler for bmm (#1655)
2022-09-28 11:32:16 +08:00
..
solver [autoparallel] added node handler for bmm (#1655) 2022-09-28 11:32:16 +08:00
__init__.py [autoparallel] standardize the code structure (#1469) 2022-08-19 15:51:54 +08:00