guopeian
4 years ago
1 changed files with 5 additions and 2 deletions
Unified View
Diff Options
+ 5
- 2
tests/ut/ge/single_op/single_op_task_unittest.cc
View File
| @@ -202,6 +202,7 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { | |||||
| auto op_desc = make_shared<OpDesc>("deque", "Deque"); | auto op_desc = make_shared<OpDesc>("deque", "Deque"); | ||||
| ge::AttrUtils::SetBool(op_desc, ATTR_NAME_IS_BLOCKING_OP, false); | ge::AttrUtils::SetBool(op_desc, ATTR_NAME_IS_BLOCKING_OP, false); | ||||
| AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE); | AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE); | ||||
| GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); | |||||
| task.SetOpDesc(op_desc); | task.SetOpDesc(op_desc); | ||||
| task.num_inputs_ = 2; | task.num_inputs_ = 2; | ||||
| task.num_outputs_ = 1; | task.num_outputs_ = 1; | ||||
| @@ -221,11 +222,13 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { | |||||
| std::string ext_info_str(ext_mem, sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); | std::string ext_info_str(ext_mem, sizeof(ge::hybrid::AicpuExtInfo) + sizeof(int32_t)); | ||||
| vector<DataBuffer> inputs(2, DataBuffer()); | vector<DataBuffer> inputs(2, DataBuffer()); | ||||
| vector<DataBuffer> outputs(1, 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)); | |||||
| vector<GeTensorDesc> inputs_desc(2, tensor); | |||||
| vector<GeTensorDesc> outputs_desc(1, tensor); | |||||
| 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; | ||||
| ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS); | ASSERT_EQ(task.InitForSummaryAndCopy(), SUCCESS); | ||||
| task.num_inputs_ = 0; | |||||
| task.num_outputs_ = 0; | |||||
| ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS); | ASSERT_EQ(task.LaunchKernel(inputs_desc, inputs, outputs_desc, outputs, stream), SUCCESS); | ||||
| } | } | ||||
| TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) { | TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) { | ||||