|
|
@@ -271,13 +271,17 @@ 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++) {
|
|
|
|
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<TensorBuffer> 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);
|
|
|
|
}
|
|
|
|