Searched refs:numRanks (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | NCCLUtils.hpp | 287 int numRanks, in create() argument
292 ncclCommInitRank(&(comm->ncclComm_), numRanks, commId, rank), in create()
302 int numRanks, in create() argument
314 &(comm->ncclComm_), numRanks, commId, rank, &config), in create()
319 &(comm->ncclComm_), numRanks, commId, rank, &config), in create()
|
| H A D | ProcessGroupNCCL.cpp | 2203 int numRanks, rank; in checkForNCCLErrorsInternal() local
2207 numRanks = getSize(); in checkForNCCLErrorsInternal()
2211 numRanks = 1; in checkForNCCLErrorsInternal()
2216 numRanks = 2; in checkForNCCLErrorsInternal()
2268 ncclComm = NCCLComm::create(numRanks, rank, ncclID, options_->config); in checkForNCCLErrorsInternal()
2270 ncclComm = NCCLComm::create(numRanks, rank, ncclID); in checkForNCCLErrorsInternal()
|