Commit Graph

7 Commits

Author SHA1 Message Date
Jiarui Fang
e956d93ac2 [refactor] memory utils (#577) 2022-04-01 09:22:33 +08:00
Jiarui Fang
705f56107c [zero] refactor model data tracing (#537) 2022-03-28 16:38:18 +08:00
Frank Lee
3601b2bad0 [test] fixed rerun_on_exception and adapted test cases (#487) 2022-03-25 17:25:12 +08:00
Jiarui Fang
4d322b79da [refactor] remove old zero code (#517) 2022-03-25 14:54:39 +08:00
Jiarui Fang
920c5889a7 [zero] add colo move inline (#521) 2022-03-25 14:02:55 +08:00
Jiarui Fang
7ef3507ace [zero] show model data cuda memory usage after zero context init. (#515) 2022-03-25 11:23:35 +08:00
Jiarui Fang
0035b7be07 [memory] add model data tensor moving api (#503) 2022-03-24 14:29:41 +08:00