Ziyue Jiang b0936e4a44 [rpc] split with dag (#2028) ...

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

2022-11-29 11:36:28 +08:00

..

rpc_test_utils.py

[rpc] split with dag (#2028)

2022-11-29 11:36:28 +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

[rpc] split with dag (#2028)

2022-11-29 11:36:28 +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