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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-11 14:43:10 +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