diff --git a/inc/framework/common/taskdown_common.h b/inc/framework/common/taskdown_common.h index 28f6f2b0..f2f731be 100644 --- a/inc/framework/common/taskdown_common.h +++ b/inc/framework/common/taskdown_common.h @@ -45,8 +45,8 @@ typedef enum tagccKernelType { TE_AI_CPU = 5, /* te aicpu operator */ AI_CPU = 6, /* aicpu */ CUST_AI_CPU = 7, /* custom aicpu*/ - INVALID = 8, /* unknown kernel type */ - HOST_CPU = 9 /* host cpu */ + HOST_CPU = 8, /* host cpu */ + INVALID = 10000 /* unknown kernel type */ } ccKernelType; typedef struct tagOpContext { diff --git a/tests/ut/ge/hybrid/node_executor/host_cpu/host_cpu_node_task_unittest.cc b/tests/ut/ge/hybrid/node_executor/host_cpu/host_cpu_node_task_unittest.cc index 74fd7519..b113fa9b 100644 --- a/tests/ut/ge/hybrid/node_executor/host_cpu/host_cpu_node_task_unittest.cc +++ b/tests/ut/ge/hybrid/node_executor/host_cpu/host_cpu_node_task_unittest.cc @@ -82,7 +82,7 @@ TEST_F(UtestHostCpuNodeTask, test_load) { host_task_def.set_type(RT_MODEL_TASK_KERNEL); ASSERT_EQ(node_executor.LoadTask(hybrid_model, node, task), INTERNAL_ERROR); domi::KernelContext *context = host_task_def.mutable_kernel()->mutable_context(); - context->set_kernel_type(9); // ccKernelType::HOST_CPU + context->set_kernel_type(8); // ccKernelType::HOST_CPU ASSERT_EQ(node_executor.LoadTask(hybrid_model, node, task), INTERNAL_ERROR); HostCpuEngine::GetInstance().constant_folding_handle_ = (void *)0x01; ASSERT_EQ(node_executor.LoadTask(hybrid_model, node, task), INTERNAL_ERROR);