Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h31 static std::map<std::string, ncclDataType_t> kNcclDtypeMap = { variable
52 ncclDataType_t nccl_dtype(const TypeId &type_id) { return kNcclDtypeMap[TypeIdLabel(type_id)]; } in nccl_dtype()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc144 auto nccl_data_type_iter = kernel::kNcclDtypeMap.find(TypeIdLabel(type)); in LaunchAllReduce()
145 if (nccl_data_type_iter == kernel::kNcclDtypeMap.end()) { in LaunchAllReduce()