m53297601
/
graphengine
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

modify LoadCustAicpuSo

Browse Source
pull/638/head
wuweikang 4 years ago
parent ef63b2207a
commit 425b9b7a58
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File

4
ge/graph/load/new_model_manager/model_manager.cc
Unescape View File

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

Write Preview
Loading…
Cancel
Save