From e370f1cefe6aa1704e9099a2bf99fa64743399cb Mon Sep 17 00:00:00 2001 From: guopeian Date: Wed, 21 Jul 2021 17:25:16 +0800 Subject: [PATCH] fix --- .../aicpu/aicpu_node_executor_unittest.cc | 19 ++++++------------- 1 file changed, 6 insertions(+), 13 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 29a6e97c..294da594 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 @@ -265,24 +265,17 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_copy_data_to_hbm) { domi::KernelDef *kernel_def = task_def.mutable_kernel(); kernel_def->set_args(reinterpret_cast(&args), args.head.length); kernel_def->set_args_size(args.head.length); - AicpuExtInfo aicpu_ext_info; - aicpu_ext_info.infoType = aicpu::FWKAdapter::FWK_ADPT_EXT_SHAPE_TYPE; - aicpu_ext_info.infoLen = sizeof(int32_t); - memcpy_s(aicpu_ext_info.infoMsg, sizeof(int32_t), &type, sizeof(int32_t)); - char *ext_mem = (char*)malloc(sizeof(AicpuExtInfo) + sizeof(int32_t)); - memcpy_s(ext_mem, sizeof(AicpuExtInfo) + sizeof(int32_t), &aicpu_ext_info2, sizeof(AicpuExtInfo) + sizeof(int32_t)); - kernel_def->set_kernel_ext_info(ext_mem, sizeof(AicpuExtInfo) + sizeof(int32_t)); - kernel_def->set_kernel_ext_info_size(sizeof(AicpuExtInfo) + sizeof(int32_t)); hybrid_model.task_defs_[node] = std::vector({task_def, task_def}); AicpuNodeTask aicpu_node_task(node_item, task_def); - std::vector> &out_shape_hbm; + std::vector> out_shape_hbm; ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS); for (int i = 0; i < node_item->num_outputs; i++) { - aicpu_node_task.output_summary_[i].shape_data_size = 1; - aicpu_node_task.output_summary_[i].raw_data_size = 1; - aicpu_node_task.output_summary_[i].shape_data_ptr = 0; - aicpu_node_task.output_summary_[i].raw_data_ptr = 0; + const auto &summary = aicpu_node_task.output_summary_[i]; + summary.shape_data_size = 1; + summary.raw_data_size = 1; + summary.shape_data_ptr = 0; + summary.raw_data_ptr = 0; } ASSERT_EQ(aicpu_node_task.ReadResultSummaryAndPrepareMemory(*node_state->GetTaskContext(), out_shape_hbm), SUCCESS); ASSERT_EQ(aicpu_node_task.CopyDataToHbm(*node_state->GetTaskContext(), out_shape_hbm), SUCCESS);