Commits · 80eba05b0abc0ce24f02254cbe2c7b8f9ff5d688 · OpenDAS / ColossalAI

06 Apr, 2023 1 commit

[test] refactor tests with spawn (#3452) · 80eba05b

Frank Lee authored Apr 06, 2023

* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code

80eba05b

12 Dec, 2022 1 commit

[PP Middleware] Add bwd and step for PP middleware (#2111) · 09d69e1c

Ziyue Jiang authored Dec 12, 2022



* add bwd and step for PP middleware

* pre-commit
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

09d69e1c

08 Dec, 2022 1 commit

[Pipeline Middleware] fix data race in Pipeline Scheduler for DAG (#2087) · e4705ba4

Ziyue Jiang authored Dec 08, 2022



* add DAG test case

* fix datarace by adjusting theposition of lock

* polish code

* fix pytest for middleware

* remove test
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

e4705ba4

05 Dec, 2022 1 commit

[Pipeline Middleware] Adapt scheduler for Topo (#2066) · 597cdd30

Ziyue Jiang authored Dec 05, 2022



* adapt scheduler for Topo

* remoove comment

* fix set input
Co-authored-by: Ziyue Jiang <ziyue.jiang@gmail.com>

597cdd30

29 Nov, 2022 1 commit

[rpc] split with dag (#2028) · b0936e4a

Ziyue Jiang authored Nov 29, 2022



* 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>

b0936e4a