github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-25 01:03:35 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
df63db7e63d951017fd1fa797fbcaec259fba644
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