Browse Source

update UT

pull/1313/head
chuxing 4 years ago
parent
commit
59c9181cdb
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      tests/ut/ge/single_op/single_op_model_unittest.cc

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

@@ -241,6 +241,7 @@ TEST_F(UtestSingleOpModel, test_build_memcpy_task) {
param.memory_size = 1024;
auto ret = RtsKernelTaskBuilder::BuildMemcpyAsyncTask(op_desc, kernel_def, param, task);
ASSERT_EQ(ret, SUCCESS);
ASSERT_EQ(task->LaunchKernel(nullptr), SUCCESS);

op_desc->SetInputOffset({0, 0});
op_desc->AddOutputDesc("y2", tensor_desc);


Loading…
Cancel
Save