Browse Source

For const input in dynamic aicpu single_op.

tags/v1.2.0
unknown 3 years ago
parent
commit
97fac611e1
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/single_op/task/aicpu_task_builder.cc

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

@@ -105,7 +105,7 @@ namespace ge {
fwk_op_kernel.fwkKernelBase.fwk_kernel.extInfoAddr = reinterpret_cast<uintptr_t>(task.ext_info_addr_dev_);
fwk_op_kernel.fwkKernelBase.fwk_kernel.extInfoLen = kernel_ext_info_size;
}
GE_CHK_STATUS_RET(task.SetInputConst(), "AicpuTask set input_const failed.");
GE_CHK_STATUS_RET(task.SetInputConst(), "AiCpuTask set input_const failed.");
GE_CHK_STATUS_RET(task.InitForSummaryAndCopy(), "AiCpuTask init for summary and copy task failed.");

fwk_op_kernel.fwkKernelBase.fwk_kernel.sessionID = ULLONG_MAX;


Loading…
Cancel
Save