github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-17 02:00:25 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
73494de57773cfc804f729234bf3611b65f13447
ColossalAI/colossalai/tensor/colo_parameter.py
Hongxin Liu 2dd01e3a14 [gemini] fix param op hook when output is tuple (#5355) 
* [gemini] fix param op hook when output is tuple

* [gemini] fix param op hook
2024-02-04 11:58:26 +08:00

3.4 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink