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	[fx] Add activation checkpoint solver rotor (#1496)	2022-08-26 10:34:21 +08:00
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