Browse Source

modify LoadCustAicpuSo

pull/439/head
wuweikang taoxiangdong 5 years ago
parent
commit
63f93cb4a5
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      ge/graph/load/new_model_manager/model_manager.cc

+ 2
- 2
ge/graph/load/new_model_manager/model_manager.cc View File

@@ -1266,8 +1266,8 @@ Status ModelManager::LoadCustAicpuSo(const OpDescPtr &op_desc, const string &so_
std::lock_guard<std::mutex> lock(cust_aicpu_mutex_);
CustAICPUKernelPtr aicpu_kernel = op_desc->TryGetExtAttr(OP_EXTATTR_CUSTAICPU_KERNEL, CustAICPUKernelPtr());
if (aicpu_kernel == nullptr) {
GELOGE(INTERNAL_ERROR, "cust aicpu op %s can't find kernel!", op_desc->GetName().c_str());
return INTERNAL_ERROR;
GELOGI("cust aicpu op %s has no corresponding kernel!", op_desc->GetName().c_str());
return SUCCESS;
}

// get current context


Loading…
Cancel
Save