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

!32709 correct the cpu kernel factory error ionformation on branch 1.6 

Merge pull request !32709 from 沈竞兴/r1.60408
This commit is contained in:
i-robot 2022-04-19 01:59:24 +00:00 committed by Gitee
parent 14be329ef8 9cf98a18ce
commit c8092b7143
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel_factory.cc
View File

@ -90,10 +90,7 @@ void CPUKernelFactory::UpdateKernelAttrs(const std::string &kernel_name, const s
std::vector<std::pair<KernelAttr, CPUKernelCreator>> attr_creators(attr_size);
auto iter = name_to_attr_creator_.find(kernel_name);
if (iter == name_to_attr_creator_.end()) {
MS_LOG(EXCEPTION) << "For '" << kernel_name
<< ", only support these types: Concat, Pack, Stack, Split, Transpose, Unpack, AddN, "
"ConcatOffset or DynamicStitch currently, but got "
<< kernel_name;
MS_LOG(EXCEPTION) << "For CPU kernel, op[" << kernel_name << "] hasn't been supported, but was called.";
}
if (attr_size <= iter->second.size()) {