From c9eb4f3ab79e4eb8e91b9d5bf4dfc010c44b9d9a Mon Sep 17 00:00:00 2001 From: l00444296 Date: Thu, 19 Nov 2020 19:42:32 +0800 Subject: [PATCH] Feature: Cancel default set aicpu engine --- ge/graph/passes/compile_nodes_pass.cc | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/ge/graph/passes/compile_nodes_pass.cc b/ge/graph/passes/compile_nodes_pass.cc index 3d7ed73f..eff5ed78 100755 --- a/ge/graph/passes/compile_nodes_pass.cc +++ b/ge/graph/passes/compile_nodes_pass.cc @@ -115,6 +115,10 @@ graphStatus CompileNodesPass::GetSupportedKernel(const NodePtr &node, const std: string aicpu_kernel_lib_name = kAICPUKernelLibName; OpsKernelInfoStorePtr aicpu_kernel_info = instance->OpsKernelManagerObj().GetOpsKernelInfoStore(aicpu_kernel_lib_name); + if (aicpu_kernel_info == nullptr) { + GELOGE(ge::GE_GRAPH_PARAM_NULLPTR, "Get aicpu kernel info store failed."); + return ge::GE_GRAPH_PARAM_NULLPTR; + } if (!CheckAccuracySupport(aicpu_kernel_info, instance, op_desc)) { GELOGE(GRAPH_FAILED, "AICPU engine does not support node:%s, type:%s , get kernel lib failed.", node->GetName().c_str(), op_desc->GetType().c_str());