Searched refs:AllReduceAlgo (Results 1 – 3 of 3) sorted by relevance
675 AllReduceAlgo IntraNodeComm::selectAllReduceAlgo(const at::Tensor& input) { in selectAllReduceAlgo()678 return AllReduceAlgo::NONE; in selectAllReduceAlgo()689 return AllReduceAlgo::HCM; in selectAllReduceAlgo()697 return AllReduceAlgo::ONE_SHOT; in selectAllReduceAlgo()704 return AllReduceAlgo::TWO_SHOT; in selectAllReduceAlgo()707 return AllReduceAlgo::NONE; in selectAllReduceAlgo()714 AllReduceAlgo algo) { in allReduce()722 case AllReduceAlgo::ONE_SHOT: in allReduce()724 case AllReduceAlgo::TWO_SHOT: in allReduce()726 case AllReduceAlgo::HCM: in allReduce()
27 enum class AllReduceAlgo : uint8_t { enum66 AllReduceAlgo selectAllReduceAlgo(const at::Tensor& input);68 at::Tensor allReduce(const at::Tensor& input, AllReduceAlgo algo);
3454 if (algo != intra_node_comm::AllReduceAlgo::NONE) { in checkForNCCLErrorsInternal()