ColossalAI/colossalai/fx/codegen/activation_checkpoint_codegen.py at 8fb09a950a2bf207d8165acd81660d08deed60bf

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-30 20:04:42 +00:00

Files

Boyuan Yao 1f2e547f7a [fx] Fix ckpt functions' definitions in forward (#1476 )

* [fx] fix defining ckpt functions inside forward

* [fx] Modify activation checkpoint codegen and add ColoGraphModule

* [fx] some modification

* some modifications

* some modifications

* some modifications

* some modifications

* some code modifications

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

28 KiB

Raw Blame History

View Raw

28 KiB Raw Blame History

28 KiB

Raw Blame History