Searched defs:ncclComm_ (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/torch/csrc/distributed/c10d/ | ||
| D | ProcessGroupNCCL.hpp | 355 std::shared_ptr<NCCLComm> ncclComm_; member in c10d::ProcessGroupNCCL::WorkNCCL |
| D | NCCLUtils.hpp | 515 ncclComm_t ncclComm_; member in c10d::NCCLComm |