ColossalAI/tests/test_fx/test_ckpt_solvers
Boyuan Yao 1df98d5b66
[autoparallel] add rotor C version (#1658)
* [autoparallel] add rotor c version

* [fx] remove metainfoprop in rotor solver

* [autoparallel] modify C
 code format

* [autoparallel] remove build.py

* [autoparallel] fix C extension build

* [autoparallel] add C solver consistency test

* [autoparallel] remove some unused imports

* [autoparallel] refactor rotor solver code

* [autoparallel] replace print with colossalai logger

* [autoparallel] ranks fixed
2022-10-03 17:13:30 +08:00
..
test_C_solver_consistency.py [autoparallel] add rotor C version (#1658) 2022-10-03 17:13:30 +08:00
test_ckpt_torchvision.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
test_linearize.py [fx] Add pofo solver (#1608) 2022-09-20 11:20:48 +08:00