Searched refs:kNcclDtypeMap (Results 1 – 2 of 2) sorted by relevance
31 static std::map<std::string, ncclDataType_t> kNcclDtypeMap = { variable52 ncclDataType_t nccl_dtype(const TypeId &type_id) { return kNcclDtypeMap[TypeIdLabel(type_id)]; } in nccl_dtype()
144 auto nccl_data_type_iter = kernel::kNcclDtypeMap.find(TypeIdLabel(type)); in LaunchAllReduce()145 if (nccl_data_type_iter == kernel::kNcclDtypeMap.end()) { in LaunchAllReduce()