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