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
..
middleware	[Pipeline Middleware] Adapt scheduler for Topo (#2066)	2022-12-05 20:23:41 +08:00
rpc	[Pipeline Middleware ] Fix deadlock when num_microbatch=num_stage (#2156)	2022-12-23 11:38:43 +08:00
__init__.py	…
layer_spec.py	…
pipelinable.py	…
pipeline_process_group.py	…
utils.py	…