ColossalAI/colossalai/gemini/tensor_utils.py at ecd643f1e4e0100c08bd0765337fe5d2287f07dd - ColossalAI - Gitea: Git with a cup of tea at home

github/ColossalAI

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

Files

Zihao 6a9158f1fa [Gemini] free and allocate cuda memory by tensor.storage, add grad hook (#2040 )

2022-11-30 15:57:45 +08:00

3.9 KiB

Raw Blame History

View Raw