2 Commits

Author	SHA1	Message	Date
Baizhou Zhang	660eed9124	[pipeline] set optimizer to optional in execute_pipeline (#4630) ... * set optimizer to optional in execute_pipeline * arrange device and mixed precision in booster init * fix execute_pipeline in booster.py	2023-09-07 10:42:59 +08:00
LuGY	a78daf6180	[shardformer] support interleaved pipeline (#4448) ... * support interleaved pipeline * fix unit test * remove virtual stage test in stage mgr * add droped type hint and updated bwd	2023-08-16 19:29:03 +08:00