ColossalAI/colossalai/fx/graph_module.py at 07f5a4e054e183f4604734b174fda5a84c5cd652

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-05-02 08:37:36 +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

6.7 KiB

Raw Blame History

View Raw

6.7 KiB Raw Blame History

6.7 KiB

Raw Blame History