Browse Source

fix ut

pull/2039/head
guopeian 3 years ago
parent
commit
c5e98ccdaa
1 changed files with 2 additions and 2 deletions
  1. +2
    -2
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc

+ 2
- 2
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -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

Loading…
Cancel
Save