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
90cd5227a348dfe506e95b2e49f2a8dcd34fdbca
ColossalAI/colossalai/tensor/param_op_hook.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

5.0 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink