From a6140ebc17e7da227ef7d04f5bf483e3d4ba35b6 Mon Sep 17 00:00:00 2001 From: guopeian Date: Wed, 21 Jul 2021 13:59:16 +0800 Subject: [PATCH] fix ut --- tests/ut/ge/single_op/single_op_model_unittest.cc | 12 +++++------- 1 file changed, 5 insertions(+), 7 deletions(-) diff --git a/tests/ut/ge/single_op/single_op_model_unittest.cc b/tests/ut/ge/single_op/single_op_model_unittest.cc index 77f3e3f1..a6466972 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -422,15 +422,13 @@ TEST_F(UtestSingleOpModel, build_memcpoy_task) { AicpuTaskStruct args; args.head.length = sizeof(args); args.head.ioAddrNum = 6; - domi::TaskDef *task_def = model_task_def->add_task(); - task_def->set_type(RT_MODEL_TASK_KERNEL); - domi::KernelDef *kernel_def = task_def->mutable_kernel(); - kernel_def->set_args(reinterpret_cast(&args), args.head.length); - kernel_def->set_args_size(0); + domi::KernelDef kernel_def; + kernel_def.set_args(reinterpret_cast(&args), args.head.length); + kernel_def.set_args_size(0); AiCpuCCTask aicpu_task; ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); char* args2 = "123"; - kernel_def->set_args(reinterpret_cast(&args2), 3); - kernel_def->set_args_size(3); + kernel_def.set_args(reinterpret_cast(&args2), 3); + kernel_def.set_args_size(3); ASSERT_EQ(aicpu_task.SetMemCopyTask(kernel_def), FAILED); }