|
|
@@ -203,6 +203,9 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { |
|
|
|
ge::AttrUtils::SetBool(op_desc, ATTR_NAME_IS_BLOCKING_OP, false); |
|
|
|
AttrUtils::SetInt(op_desc, ::ge::ATTR_NAME_UNKNOWN_SHAPE_TYPE, ge::DEPEND_COMPUTE); |
|
|
|
GeTensorDesc tensor(GeShape(), FORMAT_NCHW, DT_FLOAT); |
|
|
|
op_desc->AddInputDesc(tensor); |
|
|
|
op_desc->AddInputDesc(tensor); |
|
|
|
op_desc->AddOutputDesc(tensor); |
|
|
|
task.SetOpDesc(op_desc); |
|
|
|
task.num_inputs_ = 2; |
|
|
|
task.num_outputs_ = 1; |
|
|
@@ -227,8 +230,6 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { |
|
|
|
ASSERT_EQ(task.SetExtInfoAndType(ext_info_str, 0), SUCCESS); |
|
|
|
task.unknown_type_ = ge::DEPEND_COMPUTE; |
|
|
|
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); |
|
|
|
} |
|
|
|
TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) { |
|
|
|