Home
last modified time | relevance | path

Searched refs:TypeIdLabel (Results 1 – 25 of 64) sorted by relevance

123

/third_party/mindspore/mindspore/ccsrc/frontend/parallel/
Dcontext.cc154 if (!group.empty() && group.find(TypeIdLabel(kNumberTypeFloat)) == std::string::npos && in SetAllReduceFusionSplitIndices()
155 group.find(TypeIdLabel(kNumberTypeFloat16)) == std::string::npos && in SetAllReduceFusionSplitIndices()
156 group.find(TypeIdLabel(kNumberTypeFloat32)) == std::string::npos) { in SetAllReduceFusionSplitIndices()
157 all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat)] = indices; in SetAllReduceFusionSplitIndices()
158 all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat16)] = indices; in SetAllReduceFusionSplitIndices()
159 all_reduce_fusion_split_indices_[group + TypeIdLabel(kNumberTypeFloat32)] = indices; in SetAllReduceFusionSplitIndices()
173 if (!group.empty() && group.find(TypeIdLabel(kNumberTypeFloat)) == std::string::npos && in SetAllReduceFusionSplitSizes()
174 group.find(TypeIdLabel(kNumberTypeFloat16)) == std::string::npos && in SetAllReduceFusionSplitSizes()
175 group.find(TypeIdLabel(kNumberTypeFloat32)) == std::string::npos) { in SetAllReduceFusionSplitSizes()
176 all_reduce_fusion_split_sizes_[group + TypeIdLabel(kNumberTypeFloat)] = sizes; in SetAllReduceFusionSplitSizes()
[all …]
/third_party/mindspore/mindspore/ccsrc/runtime/device/cpu/
Dcpu_device_address.cc94 …OG(ERROR) << "Types not match. Device type: " << TypeIdLabel(type_id_) << ", host type: " << TypeI… in SyncDeviceToHost()
139 …OG(ERROR) << "Types not match. Device type: " << TypeIdLabel(type_id_) << ", host type: " << TypeI… in SyncHostToDevice()
Dkernel_select_cpu.cc199 operator_info << TypeIdLabel(input_types[i]); in KernelNotSupportException()
210 operator_info << TypeIdLabel(infer_output_types[i]); in KernelNotSupportException()
/third_party/mindspore/mindspore/ccsrc/runtime/device/ascend/
Dascend_device_address.cc196 …MS_LOG(INFO) << "SyncDeviceToHost, Device(format:" << format_ << ", type_id:" << TypeIdLabel(type_… in SyncDeviceToHost()
197 … << ", size:" << size_ << "), Host(type_id:" << TypeIdLabel(type) << ", size:" << size << ")"; in SyncDeviceToHost()
234 …MS_LOG(ERROR) << "Unsupported to trans, dev_format:" << format_ << ", dev_type:" << TypeIdLabel(ty… in SyncDeviceToHost()
235 << ", host_type:" << TypeIdLabel(type); in SyncDeviceToHost()
310 …ncDeviceToHostAndConvertFormat, Device(format:" << format_ << ", type_id:" << TypeIdLabel(type_id_) in SyncDeviceToHostAndConvertFormat()
311 … << ", size:" << size_ << "), Host(type_id:" << TypeIdLabel(type) << ", size:" << size << ")"; in SyncDeviceToHostAndConvertFormat()
362 …MS_LOG(INFO) << "SyncHostToDevice, Device(format:" << format_ << ", type_id:" << TypeIdLabel(type_… in SyncHostToDevice()
363 … << ", size:" << size_ << "), Host(format:" << format << ", type_id:" << TypeIdLabel(type) in SyncHostToDevice()
401 …MS_LOG(ERROR) << "Unsupported trans, dev_format:" << format_ << ", dev_type:" << TypeIdLabel(type_… in SyncHostToDevice()
402 << ", host_type:" << TypeIdLabel(type); in SyncHostToDevice()
[all …]
Dge_types_convert.cc36 MS_LOG(INFO) << "Vm origin type_id:" << type_id << ": " << TypeIdLabel(type_id); in GetGeDataType()
39 … MS_LOG(EXCEPTION) << "MindSpore data type:" << TypeIdLabel(type_id) << " can't been found in GE."; in GetGeDataType()
56 MS_LOG(EXCEPTION) << "Invalid data type:" << type_id << ": " << TypeIdLabel(type_id); in TransTypeIdToGeDataType()
/third_party/mindspore/mindspore/core/ir/dtype/
Dtype.h45 MS_CORE_API const std::string &TypeIdLabel(const TypeId &v);
85 std::string ToString() const override { return TypeIdLabel(meta_type_); } in ToString()
120 …std::string ToString() const override { return std::string("Object:") + TypeIdLabel(object_type_);… in ToString()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/format_type/
Dcheck_consistency.cc76 …MS_LOG(ERROR) << "Found inconsistent dtype! input dtype " << input_index << ": " << TypeIdLabel(in… in CheckDataTypeForConsistency()
77 << ", selected dtype: " << TypeIdLabel(selected_data_type); in CheckDataTypeForConsistency()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/math/
Dmatmul_gpu_kernel.h100 dtype_a_ = GetCudaDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
101 dtype_b_ = GetCudaDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 1))); in Init()
102 dtype_c_ = GetCudaDataType(TypeIdLabel(AnfAlgo::GetOutputDeviceDataType(kernel_node, 0))); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nccl/
Dnccl_gpu_kernel.h52 ncclDataType_t nccl_dtype(const TypeId &type_id) { return kNcclDtypeMap[TypeIdLabel(type_id)]; } in nccl_dtype()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/
Dexpm1_cpu_kernel.cc46 …MS_LOG(EXCEPTION) << "Only support float, half, but actual data type is " << TypeIdLabel(input_dty… in Launch()
Drange_cpu_kernel.cc42 MS_LOG(EXCEPTION) << "Only support int, float, but actual data type is " << TypeIdLabel(dtype_); in Launch()
Delu_grad_cpu_kernel.cc48 MS_LOG(EXCEPTION) << "Data type is " << TypeIdLabel(dtype_) << "is not support."; in Launch()
Dpad_cpu_kernel.cc76 MS_LOG(EXCEPTION) << "Data type is " << TypeIdLabel(dtype_) << " which is not supported."; in Launch()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dconcat_outputs_for_all_gather.cc75 MS_LOG(EXCEPTION) << "Input dtype is not same, value: " << TypeIdLabel(*dtype_iter) in GenerateKernelBuildInfo()
76 << ", need dtype: " << TypeIdLabel(*cmp_dtype); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/enhancer/
Dconcat_outputs_for_all_gather.cc75 MS_LOG(EXCEPTION) << "Input dtype is not same, value: " << TypeIdLabel(*dtype_iter) in GenerateKernelBuildInfo()
76 << ", need dtype: " << TypeIdLabel(*cmp_dtype); in GenerateKernelBuildInfo()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/tbe/
Dtbe_convert_utils.cc92 MS_LOG(EXCEPTION) << "Illegal input dtype: " << TypeIdLabel(type_id); in TypeIdToString()
/third_party/mindspore/mindspore/ccsrc/runtime/device/
Dconvert_tensor_utils.cc138 MS_LOG(EXCEPTION) << "Invalid Type: " << TypeIdLabel(type); in ConvertSameType()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/nn/
Dactivation_gpu_kernel.h66 … cudnn_data_type_ = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
Dbias_add_gpu_kernel.h66 … cudnn_data_type_ = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
Dactivation_grad_kernel.h75 … cudnn_data_type_ = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
Dsoftmax_cross_entropy_with_logits_gpu_kernel.h92 … cudnn_data_type_ = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
Dsparse_softmax_cross_entropy_with_logits_gpu_kernel.h97 … cudnn_data_type_ = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kernel_node, 0))); in Init()
/third_party/mindspore/mindspore/ccsrc/backend/session/
Dcpu_session.cc341 operator_info << TypeIdLabel(kernel_build_Info->GetInputDeviceType(i)); in KernelNotSupportException()
352 operator_info << TypeIdLabel(kernel_build_Info->GetOutputDeviceType(i)); in KernelNotSupportException()
/third_party/mindspore/mindspore/ccsrc/runtime/device/gpu/
Dgpu_bucket.cc144 auto nccl_data_type_iter = kernel::kNcclDtypeMap.find(TypeIdLabel(type)); in LaunchAllReduce()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/quant/
Dbatchnorm_fold_gpu_kernel.h145 …cudnnDataType_t cudnnDataType = GetCudnnDataType(TypeIdLabel(AnfAlgo::GetInputDeviceDataType(kerne… in Init()

123