mirror of https://github.com/hpcaitech/ColossalAI
![]() * add DAG to split_module * add comment * add test case for DAG * remove print * add DAG middleware in scheduler * add test case for scheduler * remove break * recover old lifecycle Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com> |
||
---|---|---|
.. | ||
__init__.py | ||
_pipeline_base.py | ||
_pipeline_schedule.py | ||
utils.py |