|
|
@@ -199,13 +199,14 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_memcopy_task) { |
|
|
|
node_item->num_outputs = 0;
|
|
|
|
AicpuNodeTask aicpu_node_task(node_item, task_def);
|
|
|
|
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), SUCCESS);
|
|
|
|
aicpu_node_task.node_item_->num_outputs = 1;
|
|
|
|
node_item->num_outputs = 1;
|
|
|
|
AicpuNodeTask aicpu_node_task2(node_item, task_def);
|
|
|
|
kernel_def->set_args_size(0);
|
|
|
|
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED);
|
|
|
|
ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
|
|
|
|
char* args2 = "123";
|
|
|
|
kernel_def->set_args(reinterpret_cast<const char *>(&args2), 3);
|
|
|
|
kernel_def->set_args_size(3);
|
|
|
|
ASSERT_EQ(aicpu_node_task.SetMemCopyTask(task_def), FAILED);
|
|
|
|
ASSERT_EQ(aicpu_node_task2.SetMemCopyTask(task_def), FAILED);
|
|
|
|
}
|
|
|
|
|
|
|
|
TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {
|
|
|
|