Boyuan Yao f3687e4ee2 [fx] Add nested checkpoint in activation checkpoint codegen (#1585) ...

* [fx] add nested activation_checkpoint codegen

* undo algorithms commits

* solver

* undo some commits

* [fx] torch11 add nested activation checkpoint codegen

* remove some imports

* [fx] add some comments in activation codegen

* [fx] codegen instance error fix

2022-09-12 20:00:48 +08:00

..

codegen

[fx] Add nested checkpoint in activation checkpoint codegen (#1585)

2022-09-12 20:00:48 +08:00

passes

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21:04 +08:00

profiler

[fx] provide a stable but not accurate enough version of profiler. (#1547)

2022-09-07 11:21: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] Fix ckpt functions' definitions in forward (#1476)

2022-08-22 16:59:54 +08:00

proxy.py

[hotfix] fix coloproxy typos. (#1519)

2022-08-30 11:39:03 +08:00