From bccd5893dcf0afe3c000765dc805f35f70090a98 Mon Sep 17 00:00:00 2001 From: guopeian Date: Tue, 20 Jul 2021 19:26:10 +0800 Subject: [PATCH] fix ut --- .../ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc | 4 ++-- 1 file changed, 2 insertions(+), 2 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 734ee2be..d7ad0932 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 @@ -240,7 +240,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) { kernel_def.set_args_size(args.head.length); domi::KernelDef *kernel_def_tmp = task_def.mutable_kernel(); *kernel_def_tmp = kernel_def; - + hybrid_model.task_defs_[node] = std::vector({task_def}); AicpuNodeTask aicpu_node_task(node_item, task_def); ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS); ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS); @@ -323,7 +323,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task_fail) { kernel_def.set_args_size(args.head.length); domi::KernelDef *kernel_def_tmp = task_def.mutable_kernel(); *kernel_def_tmp = kernel_def; - + hybrid_model.task_defs_[node] = std::vector({task_def}); AicpuNodeTask aicpu_node_task(node_item, task_def); RTS_STUB_RETURN_VALUE(rtGetDevice, rtError_t, 0x78000001);