|
|
@@ -237,3 +237,25 @@ TEST_F(UtestSingleOpTask, test_aicpu_task_update_io_addr) { |
|
|
|
ASSERT_EQ(ret, PARAM_INVALID); |
|
|
|
} |
|
|
|
} |
|
|
|
TEST_F(UtestSingleOpTask, test_aicpu_task_launch_kernel) { |
|
|
|
AiCpuCCTask task; |
|
|
|
rtStream_t stream = nullptr; |
|
|
|
task.num_inputs_ = 2; |
|
|
|
task.num_outputs_ = 1; |
|
|
|
task.input_is_const_ = {true, false}; |
|
|
|
int total_addr = 3; |
|
|
|
uint32_t* addrs[total_addr] = {nullptr, nullptr, nullptr}; |
|
|
|
task.io_addr_ = reinterpret_cast<uintptr_t*>(addrs); |
|
|
|
task.io_addr_num_ = total_addr; |
|
|
|
|
|
|
|
{ |
|
|
|
vector<DataBuffer> inputs(2, 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)); |
|
|
|
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); |
|
|
|
} |
|
|
|
} |