Searched refs:unpack_comms (Results 1 – 1 of 1) sorted by relevance
76 static std::vector<ncclComm_t> unpack_comms(PyObject* obj, size_t size) { in unpack_comms() function152 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_reduce()185 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_all_reduce()216 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_broadcast()248 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_all_gather()281 auto user_comms = unpack_comms(_comms, inputs.size()); in THCPModule_nccl_reduce_scatter()