Browse Source

!573 Fix code check

From: @dong-duo
Reviewed-by: @xchu42,@ji_chen
Signed-off-by: @ji_chen
tags/v1.2.0
mindspore-ci-bot Gitee 4 years ago
parent
commit
55bdf67732
3 changed files with 5 additions and 5 deletions
  1. +1
    -1
      ge/graph/load/new_model_manager/task_info/kernel_task_info.cc
  2. +2
    -2
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc
  3. +2
    -2
      ge/single_op/task/aicpu_kernel_task_builder.cc

+ 1
- 1
ge/graph/load/new_model_manager/task_info/kernel_task_info.cc View File

@@ -884,7 +884,7 @@ Status KernelTaskInfo::InitAicpuTask(uint32_t op_index, const domi::KernelDef &k
if (kernel_type_ == ccKernelType::CUST_AI_CPU) {
bool loaded = false;
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc, so_name_, loaded),
"launch cust aicpu so failed");
"launch cust aicpu so failed");
}

// copy args to new host memory


+ 2
- 2
ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc View File

@@ -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.");
}


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

@@ -63,8 +63,8 @@ Status AiCpuCCTaskBuilder::BuildTask(AiCpuCCTask &task, uint64_t kernel_id) {
task.is_custom_ = true;
task.dump_flag_ |= RT_KERNEL_CUSTOM_AICPU;
bool loaded = false;
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc_, so_name, loaded),
"launch cust aicpu so failed");
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LoadCustAicpuSo(op_desc_, so_name, loaded),
"launch cust aicpu so failed");
if (!loaded) {
GE_CHK_STATUS_RET(ModelManager::GetInstance()->LaunchCustAicpuSo(), "launch cust aicpu so failed.");
}


Loading…
Cancel
Save