diff --git a/mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel_factory.cc b/mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel_factory.cc index d6db9c96c23..be9b8a5d933 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel_factory.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/cpu/cpu_kernel_factory.cc @@ -54,6 +54,9 @@ void CPUKernelFactory::SetKernelAttrs(const std::shared_ptr op_i std::vector *kernel_attrs) { auto inputs_ptr = op_info->inputs_ptr(); auto outputs_ptr = op_info->outputs_ptr(); + if (inputs_ptr.empty()) { + MS_LOG(EXCEPTION) << "op " << op_info->op_name() << " input size is zero."; + } auto first_input_dtypes = inputs_ptr[0]->dtypes(); auto input_formats = inputs_ptr[0]->formats(); @@ -82,8 +85,7 @@ void CPUKernelFactory::UpdateKernelAttrs(const std::string &kernel_name, const s std::vector> attr_creators(attr_size); auto iter = name_to_attr_creator_.find(kernel_name); if (iter == name_to_attr_creator_.end()) { - MS_LOG(ERROR) << "CPUKernelFactory has not registered operator: " << kernel_name; - return; + MS_LOG(EXCEPTION) << "CPUKernelFactory has not registered operator: " << kernel_name; } if (attr_size <= iter->second.size()) { @@ -113,7 +115,7 @@ std::pair CPUKernelFactory::CPUKernelAttrCheck(const std::string & if (kernel_attrs[0].GetInputSize() == 0 && kernel_attrs[0].GetOutputSize() == 0) { auto op_info_ptr = mindspore::kernel::OpLib::FindOp(kernel_name, kernel::OpImplyType::kCPU); if (op_info_ptr == nullptr) { - MS_LOG(ERROR) << "Not find op[" << kernel_name << "] in cpu"; + MS_LOG(EXCEPTION) << "Not find op[" << kernel_name << "] in cpu"; } kernel_attrs.clear(); SetKernelAttrs(op_info_ptr, &kernel_attrs); @@ -152,8 +154,7 @@ std::vector CPUKernelFactory::GetSupportedKernelAttrList(const std:: std::vector result; auto iter = name_to_attr_creator_.find(kernel_name); if (iter == name_to_attr_creator_.end()) { - MS_LOG(WARNING) << "Not registered CPU kernel: op[" << kernel_name << "]!"; - return result; + MS_LOG(EXCEPTION) << "Not registered CPU kernel: op[" << kernel_name << "]!"; } auto creators = iter->second; for (size_t index = 0; index < creators.size(); ++index) { diff --git a/mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.cc b/mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.cc index 7694e3f73c8..ee446aa81af 100644 --- a/mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.cc +++ b/mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.cc @@ -188,6 +188,7 @@ bool OpLib::RegOpFromLocalInfo() { (void)OpLib::RegOp(line, ""); } } + file.close(); MS_LOG(INFO) << "End"; return true; } diff --git a/mindspore/ccsrc/runtime/device/cpu/kernel_select_cpu.cc b/mindspore/ccsrc/runtime/device/cpu/kernel_select_cpu.cc index b9e4c5057dd..bfc4bf7c6c9 100644 --- a/mindspore/ccsrc/runtime/device/cpu/kernel_select_cpu.cc +++ b/mindspore/ccsrc/runtime/device/cpu/kernel_select_cpu.cc @@ -244,7 +244,7 @@ void SetKernelInfo(const CNodePtr &kernel_node) { std::string op_name = AnfAlgo::GetCNodeName(kernel_node); auto op_info_ptr = mindspore::kernel::OpLib::FindOp(op_name, kernel::OpImplyType::kCPU); if (op_info_ptr == nullptr) { - MS_LOG(ERROR) << "Not find op[" << op_name << "] in cpu"; + MS_LOG(EXCEPTION) << "Not find op[" << op_name << "] in cpu"; } kernel_attrs.clear(); kernel::CPUKernelFactory::GetInstance().SetKernelAttrs(op_info_ptr, &kernel_attrs); @@ -260,8 +260,8 @@ void SetKernelInfo(const CNodePtr &kernel_node) { KernelNotSupportException(kernel_node, input_types, output_types); } matched = std::make_pair(false, false); - SelectKernel(kernel_node, &selected_kernel_attr, kernel_attrs, input_types, input_not_cnode_indexes, output_types, - &matched, 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); } diff --git a/mindspore/ops/_op_impl/cpu/minimum_grad.py b/mindspore/ops/_op_impl/cpu/minimum_grad.py index 80d77e23c4e..ca88f07992e 100644 --- a/mindspore/ops/_op_impl/cpu/minimum_grad.py +++ b/mindspore/ops/_op_impl/cpu/minimum_grad.py @@ -24,10 +24,16 @@ minimum_grad_op_info = CpuRegOp("MinimumGrad") \ .output(1, "y2", "required") \ .dtype_format(DataType.I32_Default, DataType.I32_Default, DataType.I32_Default, \ DataType.I32_Default, DataType.I32_Default) \ - .dtype_format(DataType.F16_Default, DataType.F16_Default, DataType.F16_Default, \ - DataType.F16_Default, DataType.F16_Default) \ + .dtype_format(DataType.U32_Default, DataType.U32_Default, DataType.U32_Default, \ + DataType.U32_Default, DataType.U32_Default) \ .dtype_format(DataType.F32_Default, DataType.F32_Default, DataType.F32_Default, \ DataType.F32_Default, DataType.F32_Default) \ + .dtype_format(DataType.I64_Default, DataType.I64_Default, DataType.I64_Default, \ + DataType.I64_Default, DataType.I64_Default) \ + .dtype_format(DataType.U64_Default, DataType.U64_Default, DataType.U64_Default, \ + DataType.U64_Default, DataType.U64_Default) \ + .dtype_format(DataType.F64_Default, DataType.F64_Default, DataType.F64_Default, \ + DataType.F64_Default, DataType.F64_Default) \ .get_op_info()