github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-10 15:55:36 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
95c35f73bd4ab65e89ee3f3e3dc2936cff01bfc7
ColossalAI/tests/test_fx/test_codegen
History
Boyuan Yao d6b01feb66 [fx] Modify offload codegen (#1618) 
* [fx] modify offload codegen

* [fx] remove repeated hook definitions

* [fx] modify offload test
2022-09-23 11:04:52 +08:00
..
test_activation_checkpoint_codegen.py
[fx] Add offload codegen (#1598)
2022-09-14 15:49:06 +08:00
test_nested_activation_checkpoint_codegen.py
[fx] Add nested checkpoint in activation checkpoint codegen (#1585)
2022-09-12 20:00:48 +08:00
test_offload_codegen.py
[fx] Modify offload codegen (#1618)
2022-09-23 11:04:52 +08:00