Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
00af5bdcbe
1 changed files with 5 additions and 5 deletions
  1. +5
    -5
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc

+ 5
- 5
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -271,11 +271,11 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_copy_data_to_hbm) {
std::vector<std::unique_ptr<TensorBuffer>> out_shape_hbm;
ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS);
for (int i = 0; i < node_item->num_outputs; i++) {
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;
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);


Loading…
Cancel
Save