ColossalAI/tests/test_pipeline
Ziyue Jiang e4705ba4e2
[Pipeline Middleware] fix data race in Pipeline Scheduler for DAG (#2087)
* 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>
2022-12-08 13:32:27 +08:00
..
rpc_test_utils.py [Pipeline Middleware] fix data race in Pipeline Scheduler for DAG (#2087) 2022-12-08 13:32:27 +08:00
test_cuda_rpc_chimera.py [pipeline/pytree] add pytree to process args and kwargs | provide `data_process_func` to process args and kwargs after forward (#1642) 2022-09-29 10:58:58 +08:00
test_cuda_rpc_optimizer.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_performance.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_pipeline.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_cuda_rpc_value_correctness.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00
test_middleware_1f1b.py [Pipeline Middleware] fix data race in Pipeline Scheduler for DAG (#2087) 2022-12-08 13:32:27 +08:00
test_pipelinable.py [pipeline] refactor the pipeline module (#1087) 2022-06-10 11:27:38 +08:00
test_pipeline_process_group.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2022-09-20 18:00:39 +08:00