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.

Baizhou Zhang 660eed9124 [pipeline] set optimizer to optional in execute_pipeline (#4630) ... * set optimizer to optional in execute_pipeline * arrange device and mixed precision in booster init * fix execute_pipeline in booster.py		1 year ago
..
test_pipeline_utils	[shardformer] Pipeline/whisper (#4456)	1 year ago
test_schedule	[pipeline] set optimizer to optional in execute_pipeline (#4630)	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	[legacy] move trainer to legacy (#4545)	1 year 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	[gemini] improve compatibility and add static placement policy (#4479)	1 year ago
test_pipeline_process_group.py	[test] refactor tests with spawn (#3452)	2 years ago
test_stage_manager.py	[shardformer] support interleaved pipeline (#4448)	1 year ago