• W
    [Cherry-pick] Collective communication APIs (#46922) · 5fba2a98
    Wen Sun 提交于
    * Support both use_calc_stream and sync_op in send recv APIs (#46023)
    
    * Support both use_calc_stream and sync_op in allgather API (#46295)
    
    * Support both use_calc_stream and sync_op in collective communication API (#46761)
    
    * Move group and all reduce from collective to communication (#45848)
    
    * Completes bfloat16 dtype for collective api in eager mode (#45844)
    
    * Fix collective APIs cannot be recognized when building docs (#46962)
    Co-authored-by: NLiYuRio <63526175+LiYuRio@users.noreply.github.com>
    5fba2a98
reduce.py 3.8 KB