guopeian
4 years ago
1 changed files with 6 additions and 2 deletions
Unified View
Diff Options
+ 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) { | TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { | ||||
| AiCpuCCTask task; | 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_inputs_ = 2; | ||||
| task.num_outputs_ = 1; | task.num_outputs_ = 1; | ||||
| task.input_is_const_ = {true, false}; | 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)); | vector<GeTensorDesc> outputs_desc(1, GeTensorDesc(GeShape(), FORMAT_NCHW, DT_FLOAT)); | ||||
| ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS); | ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS); | ||||
| task.unknown_type_ = ge::DEPEND_COMPUTE; | task.unknown_type_ = ge::DEPEND_COMPUTE; | ||||
| task.num_outputs_ = 1; | |||||
| ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS); | ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS); | ||||
| ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS); | ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS); | ||||
| } | } | ||||