Browse Source

fix

pull/1879/head
guopeian 4 years ago
parent
commit
459e804cab
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/task/aicpu_kernel_task_builder.cc

+ 1
- 1
ge/single_op/task/aicpu_kernel_task_builder.cc View File

@@ -78,7 +78,7 @@ Status AiCpuCCTaskBuilder::BuildTask(AiCpuCCTask &task, uint64_t kernel_id, cons
task.is_custom_ = true; task.is_custom_ = true;
task.dump_flag_ |= RT_KERNEL_CUSTOM_AICPU; task.dump_flag_ |= RT_KERNEL_CUSTOM_AICPU;
bool loaded = false; bool loaded = false;
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc_, so_name, loaded),
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc_, so_name, loaded),
"[Load][CustAicpuSo] failed."); "[Load][CustAicpuSo] failed.");
if (!loaded) { if (!loaded) {
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LaunchCustAicpuSo(), "[Launch][CustAicpuSo] failed."); GE_CHK_STATUS_RET(ModelManager::GetInstance()->LaunchCustAicpuSo(), "[Launch][CustAicpuSo] failed.");


Loading…
Cancel
Save