Browse Source

Feature: Cancel default set aicpu engine

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

+ 2
- 2
ge/graph/passes/compile_nodes_pass.cc View File

@@ -112,6 +112,8 @@ graphStatus CompileNodesPass::GetSupportedKernel(const NodePtr &node, const std:
// begin accuracy supported check
if (!CheckAccuracySupport(kernel_info, instance, op_desc)) {
// if check accuracy support failed , try to go to other engine.
GELOGW("Check Accuracy Supported return not support, node name is %s, reason: %s. Try to go to other engine.",
op_desc->GetName().c_str(), reason.c_str());
string kernel_name_origin = kernel_lib_name;
OpsKernelManager &ops_kernel_manager = instance->OpsKernelManagerObj();
auto kernel_map = ops_kernel_manager.GetAllOpsKernelInfoStores();
@@ -144,8 +146,6 @@ bool CompileNodesPass::CheckAccuracySupport(const OpsKernelInfoStorePtr &kernel_
}
string reason;
if (!(kernel_info->CheckAccuracySupported(*ge_desc, reason, true))) {
GELOGW("Check Accuracy Supported return not support, node name is %s, reason: %s. Try to go to AICPU engine.",
op_desc->GetName().c_str(), reason.c_str());
return false;
}
return true;


Loading…
Cancel
Save