github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-15 23:23:11 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
2b859502d5c0fa4e03aaeefca2b3808a27aeea1f
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