Xuanlei Zhao 6b3bb2c249 [NFC] polish code style (#3273)		2023-03-29 15:22:21 +08:00
..
codegen	[autoparallel] move ckpt solvers to autoparallel folder / refactor code (#1764)	2022-11-01 10:43:15 +08:00
passes	[NFC] polish code style (#3273)	2023-03-29 15:22:21 +08:00
profiler	[NFC] polish colossalai/fx/profiler/experimental/profiler_module/embedding.py code style (#3256)	2023-03-29 15:22:21 +08:00
tracer	[fx] allow native ckpt trace and codegen. (#2438)	2023-01-11 13:49:59 +08:00
__init__.py	[fx] metainfo_trace as an API. (#1873)	2022-11-10 20:58:37 +08:00
_compatibility.py	[fx] meta registration compatibility (#3253)	2023-03-27 15:22:17 +08:00
_meta_regist_12.py	[fx] meta registration compatibility (#3253)	2023-03-27 15:22:17 +08:00
_meta_regist_13.py	[fx] meta registration compatibility (#3253)	2023-03-27 15:22:17 +08:00
graph_module.py	[fx] allow control of ckpt_codegen init (#2498)	2023-01-18 17:02:46 +08:00
proxy.py	[NFC] policy colossalai/fx/proxy.py code style (#3269)	2023-03-29 15:22:21 +08:00