github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-12 19:06:17 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
36ab2cb783fdee9899f734f99633fc0d63d4e980
ColossalAI/colossalai/fx/passes/adding_split_node_pass.py
Ziyue Jiang 59e343328d [Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156) 
* add splitter

* polish code

* remove comment

* fix async nan by moving to cpu first

Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>
2022-12-23 11:38:43 +08:00

6.1 KiB
Raw Blame History

View Raw
Reference in New Issue View Git Blame Copy Permalink