github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-01-16 15:37:57 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,217 Commits 60 Branches 51 Tags
f8aeecef46461ff574f51982d03310fa8c57888e
Commit Graph

3 Commits

Author SHA1 Message Date
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
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
Boyuan Yao
a7cda6f57d [fx] Add offload codegen (#1598) 
* [fx] add input activation offload to codegen

* [fx] modify unit test

* [fx] remove two skips in torch11

* [fx] use all_input_nodes instead of _input_nodes
 2022-09-14 15:49:06 +08:00