Browse Source

fix

pull/2052/head
guopeian 3 years ago
parent
commit
c8e2ac3c13
1 changed files with 6 additions and 2 deletions
  1. +6
    -2
      tests/ut/ge/single_op/single_op_task_unittest.cc

+ 6
- 2
tests/ut/ge/single_op/single_op_task_unittest.cc View File

@@ -197,7 +197,12 @@ TEST_F(UtestSingleOpTask, test_atomic_exec) {
}
TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
AiCpuCCTask task;
rtStream_t stream = nullptr;
rtStream_t stream;
ASSERT_EQ(rtStreamCreate(&stream, 0), RT_ERROR_NONE);
auto op_desc = make_shared<OpDesc>("deque", "Deque");
ge::AttrUtils::SetBool(op_desc, ATTR_NAME_IS_BLOCKING_OP, false);
AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE);
task.SetOpDesc(op_desc);
task.num_inputs_ = 2;
task.num_outputs_ = 1;
task.input_is_const_ = {true, false};
@@ -220,7 +225,6 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) {
vector<GeTensorDesc> outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT));
ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS);
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