ColossalAI

Author	SHA1	Message	Date
ver217	369a288bf3	polish utils docstring (#620 )	2022-04-01 16:36:47 +08:00
Jiarui Fang	e956d93ac2	[refactor] memory utils (#577 )	2022-04-01 09:22:33 +08:00
Liang Bowen	2c45efc398	html refactor (#555 )	2022-03-31 11:36:56 +08:00
Jie Zhu	73d36618a6	[profiler] add MemProfiler (#356 ) * add memory trainer hook * fix bug * add memory trainer hook * fix import bug * fix import bug * add trainer hook * fix #370 git log bug * modify `to_tensorboard` function to support better output * remove useless output * change the name of `MemProfiler` * complete memory profiler * replace error with warning * finish trainer hook * modify interface of MemProfiler * modify `__init__.py` in profiler * remove unnecessary pass statement * add usage to doc string * add usage to trainer hook * new location to store temp data file	2022-03-29 12:48:34 +08:00
Jiarui Fang	0bebda6ea5	[zero] fix init device bug in zero init context unittest (#516 )	2022-03-25 12:24:18 +08:00
Jiarui Fang	9330be0f3c	[memory] set cuda mem frac (#506 )	2022-03-24 16:57:13 +08:00
Frank Lee	b03b3ae99c	fixed mem monitor device (#433 ) fixed mem monitor device	2022-03-16 15:25:02 +08:00
Jiarui Fang	21dc54e019	[zero] memtracer to record cuda memory usage of model data and overall system (#395 )	2022-03-14 22:05:30 +08:00
Jiarui Fang	10e2826426	move async memory to an individual directory (#345 )	2022-03-11 15:50:28 +08:00