github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-30 20:04:42 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
ded1005667402ee9458afa53852ce2018b1ccb10
ColossalAI/colossalai/fx/codegen/activation_checkpoint_codegen.py
Boyuan Yao 5d0fdb9cb4 [fx] fix offload codegen test (#1648) 
* [fx] fix offload codegen test

* [fx] modify typing
2022-09-27 10:25:27 +08:00

44 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink