!638 modify LoadCustAicpuSo

From: @HW_KK
Reviewed-by: @ji_chen,@youui
Signed-off-by: @ji_chen,@youui
pull/638/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
commit 6b1b3f3f83

@ -1265,8 +1265,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