Searched defs:user_comms (Results 1 – 2 of 2) sorted by relevance
152 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_reduce() local185 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_all_reduce() local216 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_broadcast() local248 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_all_gather() local281 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_reduce_scatter() local
577 const comm_list& user_comms) { in broadcast()624 const comm_list& user_comms) { in reduce()672 const comm_list& user_comms) { in reduce()681 const comm_list& user_comms) { in all_reduce()723 const comm_list& user_comms) { in reduce_scatter()764 const comm_list& user_comms) { in all_gather()