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

github/ColossalAI

mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 03:38:19 +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