github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity
ColossalAI/colossalai/utils/memory_utils
History
Jiarui Fang 9330be0f3c
[memory] set cuda mem frac (#506) 		2022-03-24 16:57:13 +08:00
..
__init__.py [memory] add model data tensor moving api (#503) 2022-03-24 14:29:41 +08:00
bucket_tensor_copy.py [memory] set cuda mem frac (#506) 2022-03-24 16:57:13 +08:00
memory_monitor.py [memory] add model data tensor moving api (#503) 2022-03-24 14:29:41 +08:00
utils.py [memory] set cuda mem frac (#506) 2022-03-24 16:57:13 +08:00