ColossalAI/colossalai/device
Genghan Zhang 6630d45546
[autoparallel] Add alpha beta (#1973)
* Add alpha beta

* Fix test

* Fix test
2022-11-17 16:01:14 +08:00
..
__init__.py [autoparallel] Add alpha beta (#1973) 2022-11-17 16:01:14 +08:00
calc_pipeline_strategy.py [autoparallel] Add alpha beta (#1973) 2022-11-17 16:01:14 +08:00
device_mesh.py [autoparallel] add numerical test for node strategies (#1760) 2022-10-27 10:42:54 +08:00
profile_alpha_beta.py [autoparallel] Add alpha beta (#1973) 2022-11-17 16:01:14 +08:00