github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 07:28:45 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
b38efe4e8a39876a1dc5c15469cc62ce09de0450
ColossalAI/colossalai/fx/codegen
History
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
..
__init__.py
[fx] added activation checkpoint codegen support for torch < 1.12 (#1359)
2022-07-25 23:35:31 +08:00
activation_checkpoint_codegen.py
[fx] fix offload codegen test (#1648)
2022-09-27 10:25:27 +08:00