Browse Source

Feature: Cancel default set aicpu engine

pull/346/head
l00444296 4 years ago
parent
commit
83bb471103
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      ge/graph/passes/compile_nodes_pass.cc

+ 3
- 1
ge/graph/passes/compile_nodes_pass.cc View File

@@ -123,10 +123,12 @@ graphStatus CompileNodesPass::GetSupportedKernel(const NodePtr &node, const std:
OpsKernelInfoStorePtr tmp_kernel_info = it->second;
if (CheckAccuracySupport(tmp_kernel_info, instance, op_desc)) {
kernel_lib_name = tmp_kernel_name;
GELOGD("Find kernel lib %s support node:%s, type:%s , get kernel lib success.", tmp_kernel_name.c_str(),
node->GetName().c_str(), op_desc->GetType().c_str());
return GRAPH_SUCCESS;
}
}
GELOGE(GRAPH_FAILED, "Cannot find engine support node:%s, type:%s , get kernel lib failed.",
GELOGE(GRAPH_FAILED, "Cannot find kernel lib support node:%s, type:%s , get kernel lib failed.",
node->GetName().c_str(), op_desc->GetType().c_str());
return GRAPH_FAILED;
}


Loading…
Cancel
Save