mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!32591 optimize the cpu kernel select error info 

Merge pull request !32591 from limingqi107/bug_fix3
This commit is contained in:
i-robot 2022-04-06 18:35:17 +00:00 committed by Gitee
parent 391d3d9f7c 056afa975c
commit 7cd2dcbdfd
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 10 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
mindspore/ccsrc/plugin/device/cpu/hal/device/kernel_select_cpu.cc
View File

@ -187,8 +187,15 @@ void SetKernelBuildInfo(const std::vector<std::string> &input_formats, const std
}
void KernelNotSupportException(const AnfNodePtr &kernel_node, const std::vector<TypeId> &input_types,
const std::vector<TypeId> &infer_output_types) {
const std::vector<TypeId> &infer_output_types, bool is_kernel_exist) {
std::string kernel_name = common::AnfAlgo::GetCNodeName(kernel_node);
if (!is_kernel_exist) {
MS_LOG(EXCEPTION) << "Unsupported op [" << kernel_name
<< "] on CPU, Please confirm whether the device target setting is correct, or refer to the "
"official website to query the operator support list.\n"
<< trace::DumpSourceLines(kernel_node);
}
std::stringstream operator_info;
operator_info << "Operator[" << kernel_name << "] ";
size_t input_num = common::AnfAlgo::GetInputTensorNum(kernel_node);
@ -415,13 +422,13 @@ void SetKernelInfo(const CNodePtr &kernel_node) {
if (!SelectKernel(kernel_node, &selected_kernel_attr, kernel_attrs, input_types, input_not_cnode_indexes,
output_types, &matched, true)) {
if (op_name == "Cast") {
KernelNotSupportException(kernel_node, input_types, output_types);
KernelNotSupportException(kernel_node, input_types, output_types, !kernel_attrs.empty());
}
matched = std::make_pair(false, false);
(void)SelectKernel(kernel_node, &selected_kernel_attr, kernel_attrs, input_types, input_not_cnode_indexes,
output_types, &matched, false);
if (!matched.first) {
KernelNotSupportException(kernel_node, input_types, output_types);
KernelNotSupportException(kernel_node, input_types, output_types, !kernel_attrs.empty());
}
}