From 752da1b1da820bd6b25fd7ba5d6663a0d6b1a1f8 Mon Sep 17 00:00:00 2001 From: guopeian Date: Mon, 19 Jul 2021 11:17:44 +0800 Subject: [PATCH] fix ut --- tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc | 1 - tests/ut/ge/single_op/single_op_model_unittest.cc | 2 +- 2 files changed, 1 insertion(+), 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 22f9b889..072f4daa 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 @@ -154,7 +154,6 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) { AicpuExtInfo aicpu_ext_info2; aicpu_ext_info2.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; aicpu_ext_info2.infoLen = sizeof(int32_t); - int32_t type = node_item->shape_inference_type; memcpy_s(aicpu_ext_info2.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); char *ext_mem2 = (char*)malloc(sizeof(AicpuExtInfo) + sizeof(int32_t)); memcpy_s(ext_mem2, sizeof(AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info2, sizeof(AicpuExtInfo) + sizeof(int32_t)); 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 41b70900..f678752b 100644 --- a/tests/ut/ge/single_op/single_op_model_unittest.cc +++ b/tests/ut/ge/single_op/single_op_model_unittest.cc @@ -394,7 +394,7 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) { StreamResource *res = new (std::nothrow) StreamResource(1); ASSERT_EQ(model.ParseTasks(), SUCCESS); - model.node_tasks_[node] = { *task_def2 }; + model.node_tasks_[node] = { *task_def, *task_def }; op_desc->SetOpKernelLibName(kEngineNameAiCpu); model.BuildTaskListForDynamicOp(res, single_op); } \ No newline at end of file