Boyuan Yao a7cda6f57d [fx] Add offload codegen (#1598) ... * [fx] add input activation offload to codegen * [fx] modify unit test * [fx] remove two skips in torch11 * [fx] use all_input_nodes instead of _input_nodes		2022-09-14 15:49:06 +08:00
..
codegen	[fx] Add offload codegen (#1598)	2022-09-14 15:49:06 +08:00
passes	[hotfix/rotor] fix variable names (#1597)	2022-09-14 14:27:04 +08:00
profiler	[hotfix/rotor] fix variable names (#1597)	2022-09-14 14:27:04 +08:00
tracer	[NFC] polish colossalai/fx/tracer/meta_patch/patched_module/convolution.py code style (#1563)	2022-09-08 22:11:04 +08:00
__init__.py	[fx] support meta tracing for aten level computation graphs like functorch. (#1536)	2022-09-05 12:10:09 +08:00
graph_module.py	[fx] Add offload codegen (#1598)	2022-09-14 15:49:06 +08:00
proxy.py	[hotfix] fix coloproxy typos. (#1519)	2022-08-30 11:39:03 +08:00