mirror of https://github.com/hpcaitech/ColossalAI
a78daf6180
* support interleaved pipeline * fix unit test * remove virtual stage test in stage mgr * add droped type hint and updated bwd |
||
---|---|---|
.. | ||
test_interleaved.py | ||
test_oneF_oneB.py | ||
test_pipeline_schedule_utils.py |