From bdcbe20915cea02285d376ba8506ee5a230a82b3 Mon Sep 17 00:00:00 2001 From: guopeian Date: Wed, 21 Jul 2021 15:21:20 +0800 Subject: [PATCH] fix ut --- .../node_executor/aicpu/aicpu_node_executor_unittest.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 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 1b4b87f9..61427594 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 @@ -199,13 +199,14 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) { node_item->num_outputs = 0; AicpuNodeTask aicpu_node_task(node_item, task_def); ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), SUCCESS); - aicpu_node_task.node_item_->num_outputs = 1; + node_item->num_outputs = 1; + AicpuNodeTask aicpu_node_task2(node_item, task_def); kernel_def->set_args_size(0); - ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED); + ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED); char* args2 = "123"; kernel_def->set_args(reinterpret_cast(&args2), 3); kernel_def->set_args_size(3); - ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED); + ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED); } TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {