github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-15 06:54:52 +00:00
Code Issues Packages Projects Releases Wiki Activity
1,417 Commits 62 Branches 51 Tags
44ea461890cb194b89ccfbdd00a0f36ccb555740
Commit Graph

3 Commits

Author SHA1 Message Date
Zihao
38ea4ba1bd [Gemini] fix grad unreleased issue and param recovery issue (#2052) 2022-12-02 16:04:19 +08:00
Zihao
6a9158f1fa [Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040) 2022-11-30 15:57:45 +08:00
Zihao
95c4532fff [Gemini] paramWrapper paramTracerHook unitest (#2030) 2022-11-26 13:30:24 +08:00