Browse Source

fix ut

pull/2039/head
guopeian 3 years ago
parent
commit
3f4ba35e20
2 changed files with 22 additions and 7 deletions
  1. +0
    -7
      tests/ut/ge/single_op/single_op_model_unittest.cc
  2. +22
    -0
      tests/ut/ge/single_op/single_op_task_unittest.cc

+ 0
- 7
tests/ut/ge/single_op/single_op_model_unittest.cc View File

@@ -410,13 +410,6 @@ TEST_F(UtestSingleOpModel, build_dynamic_task02) {
model.model_helper_.model_ = ge_model;
auto op_desc = std::make_shared<ge::OpDesc>("add", "Add");
AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT);
for (int i = 0; i < 4; i++) {
op_desc->AddInputDesc(tensor);
}
for (int i = 0; i < 2; i++) {
op_desc->AddOutputDesc(tensor);
}
NodePtr node = graph->AddNode(op_desc);
model.op_list_[0] = node;
StreamResource *res = new (std::nothrow) StreamResource(1);


+ 22
- 0
tests/ut/ge/single_op/single_op_task_unittest.cc View File

@@ -237,3 +237,25 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) {
ASSERT_EQ(ret, PARAM_INVALID);
}
}
TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
AiCpuCCTask task;
rtStream_t stream = nullptr;
task.num_inputs_ = 2;
task.num_outputs_ = 1;
task.input_is_const_ = {true, false};
int total_addr = 3;
uint32_t* addrs[total_addr] = {nullptr, nullptr, nullptr};
task.io_addr_ = reinterpret_cast<uintptr_t*>(addrs);
task.io_addr_num_ = total_addr;

{
vector<DataBuffer> inputs(2, DataBuffer());
vector<DataBuffer> outputs(1, DataBuffer());
vector<GeTensorDesc> inputs_desc(2, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT));
vector<GeTensorDesc> outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT));
task.unknown_type_ = ge::DEPEND_COMPUTE;
task.num_outputs_ = 1;
ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS);
ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS);
}
}

Loading…
Cancel
Save