Browse Source

fix

pull/1985/head
guopeian 3 years ago
parent
commit
f3c04de30b
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/aicpu/aicpu_node_executor.cc

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

@@ -961,7 +961,7 @@ Status AicpuNodeTask::LaunchTask(TaskContext &context) {
auto kernel_type = static_cast<ccKernelType>(kcontext.kernel_type());
uint32_t flag = RT_KERNEL_DEFAULT;
if (kernel_type == ccKernelType::CUST_AI_CPU) {
flag |= static_cast<uint32_t>(RT_KERNEL_C USTOM_AICPU);
flag |= static_cast<uint32_t>(RT_KERNEL_CUSTOM_AICPU);
}
auto rt_ret = rtCpuKernelLaunchWithFlag(reinterpret_cast<const void *>(so_name.c_str()),
reinterpret_cast<const void *>(kernel_name.c_str()),


Loading…
Cancel
Save