From 36aec1f9019a16e7d458a2377471ba36fa11f40b Mon Sep 17 00:00:00 2001 From: guopeian Date: Thu, 22 Jul 2021 11:22:26 +0800 Subject: [PATCH] fix ut --- .../node_executor/aicpu/aicpu_node_executor_unittest.cc | 16 ++++++++++------ 1 file changed, 10 insertions(+), 6 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 1fb5f91d..36dcbdb6 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 @@ -271,13 +271,17 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_copy_data_to_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++) { - 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; + auto &summary = output_summary_host_[i]; + summary.shape_data_ptr = 0; + summary.shape_data_size = 1; + summary.raw_data_ptr = 0; + summary.raw_data_size = 1; + } + for (int i = 0; i < node_item->num_outputs; i++) { + std::unique_ptr shape_buffer; + AllocTensorBuffer(1, shape_buffer); + out_shape_hbm.emplace_back(std::move(shape_buffer)); } - 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); free(ext_mem); }