|
|
@@ -645,8 +645,8 @@ Status AicpuNodeTask::Init(const HybridModel &model) { |
|
|
|
auto kernel_type = static_cast<ccKernelType>(context.kernel_type()); |
|
|
|
if (kernel_type == ccKernelType::CUST_AI_CPU) { |
|
|
|
bool loaded = false; |
|
|
|
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc, so_name, loaded), |
|
|
|
"load cust aicpu so failed."); |
|
|
|
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc, so_name, loaded), |
|
|
|
"load cust aicpu so failed."); |
|
|
|
if (!loaded) { |
|
|
|
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LaunchCustAicpuSo(), "Launch cust aicpu so failed."); |
|
|
|
} |
|
|
|