Searched defs:ncclComm (Results 1 – 3 of 3) sorted by relevance
320 auto& ncclComm = it.first; in cacheAllocatorRegisterHook() local338 auto& ncclComm = it.first; in cacheAllocatorDeregisterHook() local364 for (auto& ncclComm : allNCCLComms) { in getNCCLCommDumpMap() local1174 auto& ncclComm = it.second; in abortCommsFromMap() local1211 auto& ncclComm = it.second; in abort() local2001 std::shared_ptr<NCCLComm>& ncclComm) { in checkForNCCLErrors()2006 std::shared_ptr<NCCLComm>& ncclComm) { in checkForNCCLErrorsInternal()2112 std::shared_ptr<NCCLComm>& ncclComm = devNCCLCommMap_[devNCCLCommMapKey]; in checkForNCCLErrorsInternal() local2160 std::shared_ptr<NCCLComm> ncclComm; in checkForNCCLErrorsInternal() local2367 auto& ncclComm = i.second; in checkForNCCLErrorsInternal() local[all …]
261 explicit NCCLComm(ncclComm_t ncclComm) in NCCLComm()
52 std::shared_ptr<c10d::NCCLComm>& ncclComm) override { in checkForNCCLErrors()