|
|
|
@ -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);
|
|
|
|
|