Browse Source

Feature: Cancel default set aicpu engine

tags/v1.1.0
l00444296 4 years ago
parent
commit
c9eb4f3ab7
1 changed files with 4 additions and 0 deletions
  1. +4
    -0
      ge/graph/passes/compile_nodes_pass.cc

+ 4
- 0
ge/graph/passes/compile_nodes_pass.cc View File

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


Loading…
Cancel
Save