You can not select more than 25 topics Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
ColossalAI/tests/test_pipeline
Jianghai a88e92251d
[pipeline] add chatglm (#4363)
1 year ago
..
test_schedule [pipeline] add unit test for 1f1b (#4303) 1 year ago
rpc_test_utils.py [test] refactor tests with spawn (#3452) 2 years ago
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) 2 years ago
test_cuda_rpc_optimizer.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_performance.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_pipeline.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_cuda_rpc_value_correctness.py [pipeline/chimera] test chimera | fix bug of initializing (#1615) 2 years ago
test_middleware_1f1b.py [test] refactor tests with spawn (#3452) 2 years ago
test_p2p_communication.py [pipeline] implement p2p communication (#4100) 1 year ago
test_pipelinable.py [test] refactor tests with spawn (#3452) 2 years ago
test_pipeline_process_group.py [test] refactor tests with spawn (#3452) 2 years ago
test_stage_manager.py [pipeline]add pipeline policy and bert forward (#4130) 1 year ago