Home
last modified time | relevance | path

Searched refs:devNCCLCommMapKey (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DProcessGroupNCCL.cpp2103 void ProcessGroupNCCL::destroyNCCLComms(const std::string& devNCCLCommMapKey) { in checkForNCCLErrorsInternal() argument
2105 if (devNCCLCommMap_.find(devNCCLCommMapKey) == devNCCLCommMap_.end()) { in checkForNCCLErrorsInternal()
2109 devNCCLCommMapKey, in checkForNCCLErrorsInternal()
2112 std::shared_ptr<NCCLComm>& ncclComm = devNCCLCommMap_[devNCCLCommMapKey]; in checkForNCCLErrorsInternal()
2117 devNCCLCommMap_.erase(devNCCLCommMapKey); in checkForNCCLErrorsInternal()
H A DProcessGroupNCCL.hpp853 void destroyNCCLComms(const std::string& devNCCLCommMapKey);