|
|
@@ -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()), |
|
|
|