| .. |
|
test_pipeline_utils
|
[shardformer] Pipeline/whisper (#4456)
|
2023-08-18 21:29:25 +08:00 |
|
test_schedule
|
[pipeline] 1f1b schedule receive microbatch size (#4589)
|
2023-09-01 21:45: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
|
[legacy] move trainer to legacy (#4545)
|
2023-09-05 21:53:10 +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
|
[gemini] improve compatibility and add static placement policy (#4479)
|
2023-08-24 09:29:25 +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
|
[shardformer] support interleaved pipeline (#4448)
|
2023-08-16 19:29:03 +08:00 |
fae6c92ead