From c5e98ccdaa9cd0173e968cafe7a6ddd8b9cd971a Mon Sep 17 00:00:00 2001
From: guopeian <guopeian1@hisilicon.com>
Date: Mon, 19 Jul 2021 10:53:52 +0800
Subject: [PATCH] fix ut

---
 .../ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc  | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
index 153dda85..22f9b889 100644
--- a/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
+++ b/tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc
@@ -148,7 +148,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
 
   domi::TaskDef task_def2;
   task_def2.set_type(RT_MODEL_TASK_ALL_KERNEL);
-  domi::KernelExDef *kernel_def = task_def2.mutable_kernel();
+  domi::KernelDef *kernel_def = task_def2.mutable_kernel();
   kernel_def->set_args(reinterpret_cast<const char *>(&args), args.head.length);
   kernel_def->set_args_size(args.head.length);
   AicpuExtInfo aicpu_ext_info2;
@@ -169,7 +169,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) {
   //kernel_ex_def->set_allocated_kernel_ext_info(nullptr);
 
   free(ext_mem);
-
+  free(ext_mem2);
 }
 }  // namespace ge