github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-04-26 17:53:08 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
8795bb2e8089b250376dde2cf8dd67cba41b7787
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