github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2026-07-13 20:39:59 +00:00
Code Issues Packages Projects Releases Wiki Activity
Files
de8a65babcf3bdf50fd1a60ff0baabe3e4f7803e
ColossalAI/tests/test_pipeline/test_schedule
History
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
..
test_interleaved.py
[shardformer] support interleaved pipeline (#4448)
2023-08-16 19:29:03 +08:00
test_oneF_oneB.py
[pipeline] add unit test for 1f1b (#4303)
2023-08-15 23:25:14 +08:00
test_pipeline_schedule_utils.py
[pipeline] refactor 1f1b schedule (#4115)
2023-08-15 23:25:14 +08:00