ColossalAI/colossalai/fx/passes/algorithms
Super Daniel 4f59693207
[fx] provide a stable but not accurate enough version of profiler. (#1547)
* [fx] compute memory stat and flop count for MetaInfoProp.

* [fx] modify node attribute.

* [fx] modify ckpt_chen.

* [fx] fix compatibility.

* [fx] fix import error.

* [fx] skip test for MetaInfoProp.

* [fx] skip test for MetaInfoProp.

* [fx] skip test for MetaInfoProp.

* [fx] skip test for MetaInfoProp.

* [fx] skip if torch 1.11.0.

* [fx] recover MetaInfoProp support for PyTorch 1.11.

* [fx] provide a stable but not accurate enough version of profiler.

* [fx] provide a stable but not accurate enough version of profiler.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix compatibility in tests.

* [fx] fix import error.
2022-09-07 11:21:04 +08:00
..
__init__.py
ckpt_solver_chen.py [fx] provide a stable but not accurate enough version of profiler. (#1547) 2022-09-07 11:21:04 +08:00
ckpt_solver_rotor.py [fx] Add common node in model linearize (#1542) 2022-09-05 18:35:05 +08:00
linearize.py [fx] Add common node in model linearize (#1542) 2022-09-05 18:35:05 +08:00
utils.py [fx] fix wrong variable name in solver rotor (#1502) 2022-08-26 15:47:08 +08:00