Browse Source

RunHostCpuKernel

tags/v1.3.0
zhaozhixuan 4 years ago
parent
commit
f192012ddf
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      ge/hybrid/node_executor/host_cpu/host_cpu_node_executor.cc

+ 1
- 1
ge/hybrid/node_executor/host_cpu/host_cpu_node_executor.cc View File

@@ -170,7 +170,7 @@ Status HostCpuNodeExecutor::LoadTask(const HybridModel &model, const NodePtr &no
GELOGE(INTERNAL_ERROR, "[Get][Handle]Get constant folding handle failed.");
return INTERNAL_ERROR;
}
auto run_cpu_kernel = (uint32_t (*)(void *))mmDlsym(handle, "RunCpuKernel");
auto run_cpu_kernel = (uint32_t (*)(void *))mmDlsym(handle, "RunHostCpuKernel");
if (run_cpu_kernel != nullptr) {
host_aicpu_task->SetRunKernel(run_cpu_kernel);
} else {


Loading…
Cancel
Save