ColossalAI/colossalai/utils/memory_utils at df7e6506d4500af6a9220ef7fe4d3c7b1daebd4c - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-16 00:14:39 +00:00

Files

History

Jiarui Fang e956d93ac2 [refactor] memory utils (#577 )

2022-04-01 09:22:33 +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

[refactor] memory utils (#577 )

2022-04-01 09:22:33 +08:00

utils.py

[refactor] memory utils (#577 )

2022-04-01 09:22:33 +08:00