10 Commits

Author	SHA1	Message	Date
HELSON	e5ea3fdeef	[gemini] add GeminiMemoryManger (#832) ... * refactor StatefulTensor, tensor utilities * add unitest for GeminiMemoryManager	2022-04-24 13:08:48 +08:00
Frank Lee	5a1a095b92	[test] refactored with the new rerun decorator (#763) ... * [test] refactored with the new rerun decorator * polish test case	2022-04-15 00:33:04 +08:00
Jiarui Fang	4d90a7b513	[refactor] zero directory (#724)	2022-04-11 23:13:02 +08:00
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