ColossalAI/tests/test_pipeline
github-actions[bot] d20dceb9a3
[format] applied code formatting on changed files in pull request 4441 (#4445)
Co-authored-by: github-actions <github-actions@github.com>
2023-08-16 10:47:23 +08:00
..
test_schedule [pipeline] add unit test for 1f1b (#4303) 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 [format] applied code formatting on changed files in pull request 4441 (#4445) 2023-08-16 10:47:23 +08:00