• Kirigaya Kazuto's avatar
    [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy (#1497) · 5a6fd71f
    Kirigaya Kazuto authored
    * support p2p communication with any type of object | pass test
    
    * reconstruct pipeline schedule with p2p_v2.py(support communication with List[Any]) | pass test
    
    * [engin/schedule] use p2p_v2 to recontruct pipeline_schedule
    
    * [pipeline/rpc] implement a demo for PP with cuda rpc framework
    
    * [pipeline/rpc] support interleaving | fix checkpoint bug | change logic when dispatch data in work_list to ensure steady 1F1B
    
    * [pipeline/rpc] implement distributed optimizer | test with assert_close
    
    * [pipeline/rpc] implement distributed optimizer | test with assert_close
    
    * [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy
    
    * [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy
    
    * [pipeline/rpc] update outstanding mechanism | optimize dispatching strategy
    5a6fd71f
test_cuda_rpc_optimizer.py 2.44 KB