Repair incomplete operator selection log

pull/4300/head
jjfeing 5 years ago
parent 2dc4dae41c
commit 37d18c3f73

@ -70,9 +70,8 @@ void KernelQueryAll(const CNodePtr &kernel_node,
std::vector<std::shared_ptr<kernel::KernelBuildInfo>> *kernel_info_list) {
MS_EXCEPTION_IF_NULL(kernel_node);
MS_EXCEPTION_IF_NULL(kernel_info_list);
std::string op_name = AnfAlgo::GetCNodeName(kernel_node);
TbeMetadataInfo(kernel_node, kernel_info_list);
if (kernel_info_list->empty()) {
AicpuMetadataInfo(kernel_node, kernel_info_list);
if (!kernel_info_list->empty()) {
@ -81,16 +80,17 @@ void KernelQueryAll(const CNodePtr &kernel_node,
AnfAlgo::SetNodeAttr(kAttrIsAICPUKernel, MakeValue(true), kernel_node);
}
}
if (kernel_info_list->empty()) {
GetRtKelInfo(kernel_node, kernel_info_list);
}
if (kernel_info_list->empty()) {
HcclMetadataInfo(kernel_node, kernel_info_list);
}
if (kernel_info_list->empty()) {
MS_LOG(EXCEPTION) << "Op " << kernel_node->DebugString() << "kernel query fail!";
MS_EXCEPTION(NotExistsError)
<< "Failed to obtain operator info, Please check whether the operator info is registered, Op full name:"
<< kernel_node->fullname_with_scope() << "Node Type: " << op_name
<< ", Node DebugString: " << kernel_node->DebugString();
}
}
@ -99,8 +99,6 @@ void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel
MS_EXCEPTION_IF_NULL(kernel_node);
MS_EXCEPTION_IF_NULL(kernel_info_list);
std::string op_name = AnfAlgo::GetCNodeName(kernel_node);
auto context_ptr = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(context_ptr);
if (context_ptr->enable_graph_kernel() && IsPrimitiveCNode(kernel_node, prim::kPrimBatchMatMul)) {
@ -117,7 +115,9 @@ void KernelQuery(const CNodePtr &kernel_node, std::vector<std::shared_ptr<kernel
}
if (kernel_info_list->empty()) {
MS_EXCEPTION(NotExistsError) << "Op[" << kernel_node->DebugString() << "] kernel query fail!";
MS_EXCEPTION(NotExistsError)
<< "Failed to obtain operator info. Please check whether the operator info is registered, Op full name:"
<< kernel_node->fullname_with_scope() << ". Node DebugString: " << kernel_node->DebugString();
}
// check output
FilterInvalidKernelInfo(kernel_node, kernel_info_list);

Loading…
Cancel
Save