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 036ea8c7..5d24cd81 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 @@ -178,6 +178,13 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) { } TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) { + ComputeGraphPtr graph = std::make_shared("test"); + GeModelPtr ge_sub_model = std::make_shared(); + GeRootModelPtr ge_root_model = std::make_shared(graph); + ge_root_model->SetModelName("test_name"); + ge_root_model->SetSubgraphInstanceNameToModel("sub", ge_sub_model); + HybridModel hybrid_model(ge_root_model); + NodePtr node = CreateNode(graph, "frameworkop", FRAMEWORK_OP_TYPE, 4, 2); std::unique_ptr new_node; ASSERT_EQ(NodeItem::Create(node, new_node), SUCCESS); NodeItem *node_item = new_node.get(); @@ -193,13 +200,13 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) { node_item->num_outputs = 0; AicpuNodeTask aicpu_node_task(node_item, task_def); ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), SUCCESS); - aicpu_node_task->node_item_->num_outputs = 1; + aicpu_node_task.node_item_->num_outputs = 1; kernel_def->set_args_size(0); - ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED); + ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED); char* args2 = "123"; kernel_def->set_args(reinterpret_cast(&args2), 3); kernel_def->set_args_size(3); - ASSERT_EQ(aicpu_nod_task.SetMemCopyTask(task_def), FAILED); + ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED); } TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {