Baizhou Zhang 36e546b2cc [pipeline] add pipeline support for T5Stack/T5EncoderModel (#4300) ...

* modify t5 policy & add test

* pipeline stage distribution for t5

* complete t5 base policy

* t5 stack: halfway

* modify gpt2 pipeline test

* complete pipeline forward for T5Stack/T5EncoderModel

* fix docstring

* move t5 util tests to test_pipeline

2023-08-15 23:25:14 +08:00

..

test_policy

[pipeline] add pipeline support for T5Stack/T5EncoderModel (#4300)

2023-08-15 23:25:14 +08:00

test_schedule

[pipeline] refactor 1f1b schedule (#4115)

2023-08-15 23:25:14 +08:00

rpc_test_utils.py

[test] refactor tests with spawn (#3452)

2023-04-06 14:51:35 +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

[test] refactor tests with spawn (#3452)

2023-04-06 14:51:35 +08:00

test_p2p_communication.py

[pipeline] implement p2p communication (#4100)

2023-08-15 23:25:14 +08:00

test_pipelinable.py

[test] refactor tests with spawn (#3452)

2023-04-06 14:51:35 +08:00

test_pipeline_process_group.py

[test] refactor tests with spawn (#3452)

2023-04-06 14:51:35 +08:00

test_stage_manager.py

[pipeline]add pipeline policy and bert forward (#4130)

2023-08-15 23:25:14 +08:00