Searched defs:_comms (Results 1 – 1 of 1) sorted by relevance
133 *_comms = nullptr; in THCPModule_nccl_reduce() local166 *_comms = nullptr; in THCPModule_nccl_all_reduce() local198 PyObject *_inputs = nullptr, *_streams = nullptr, *_comms = nullptr; in THCPModule_nccl_broadcast() local230 *_comms = nullptr; in THCPModule_nccl_all_gather() local262 *_comms = nullptr; in THCPModule_nccl_reduce_scatter() local