qsdy
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix applymomentum cpu kernel select

This commit is contained in:
zhaosida 2021-10-08 15:32:46 +08:00
parent faf3258745
commit 16f1de96d6
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/runtime/device/cpu/kernel_select_cpu.cc
View File

@ -280,7 +280,7 @@ bool SelectKernel(const CNodePtr &kernel_node, KernelAttr *selected_kernel_attr,
}
size_t output_num = AnfAlgo::GetOutputTensorNum(kernel_node);
if (kernel_attr.GetOutputSize() != output_num) {
MS_LOG(EXCEPTION) << "Output num is not equal!";
MS_LOG(DEBUG) << "Output num is not equal!";
continue;
}
int input_dtype_matched_num =