Browse Source

Pre Merge pull request !342 from 赵之轩/development

pull/342/MERGE
赵之轩 Gitee 4 years ago
parent
commit
b28e620a9c
1 changed files with 1 additions and 6 deletions
  1. +1
    -6
      ge/single_op/task/aicpu_task_builder.cc

+ 1
- 6
ge/single_op/task/aicpu_task_builder.cc View File

@@ -139,12 +139,7 @@ namespace ge {
GE_CHK_STATUS_RET(task.InitForSummaryAndCopy(), "AiCpuTask init for summary and copy task failed.");

// Create session
fwk_op_kernel.fwkKernelBase.fwk_kernel.sessionID = session_id;
GELOGI("Begin to CreateAicpuSession, session id: %lu", session_id);
GE_CHECK_NOTNULL(ModelManager::GetInstance());
GE_IF_BOOL_EXEC(ModelManager::GetInstance()->CreateAicpuSession(session_id) != SUCCESS,
GELOGE(FAILED, "CreateAicpuSession error. session id: %lu", session_id);
return FAILED;)
fwk_op_kernel.fwkKernelBase.fwk_kernel.sessionID = ULLONG_MAX;
ret = SetKernelArgs(&task.args_, fwk_op_kernel);
if (ret != SUCCESS) {
return ret;


Loading…
Cancel
Save