1
0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-05-04 14:38:10 +00:00
ColossalAI/tests/test_pipeline
Frank Lee 80eba05b0a
[test] refactor tests with spawn ()
* [test] added spawn decorator

* polish code

* polish code

* polish code

* polish code

* polish code

* polish code
2023-04-06 14:51:35 +08:00
..
rpc_test_utils.py [test] refactor tests with spawn () 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 () 2022-09-29 10:58:58 +08:00
test_cuda_rpc_optimizer.py [pipeline/chimera] test chimera | fix bug of initializing () 2022-09-20 18:00:39 +08:00
test_cuda_rpc_performance.py [pipeline/chimera] test chimera | fix bug of initializing () 2022-09-20 18:00:39 +08:00
test_cuda_rpc_pipeline.py [pipeline/chimera] test chimera | fix bug of initializing () 2022-09-20 18:00:39 +08:00
test_cuda_rpc_value_correctness.py [pipeline/chimera] test chimera | fix bug of initializing () 2022-09-20 18:00:39 +08:00
test_middleware_1f1b.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
test_pipelinable.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00
test_pipeline_process_group.py [test] refactor tests with spawn () 2023-04-06 14:51:35 +08:00