Browse Source

fix ut

pull/2052/head
guopeian 3 years ago
parent
commit
bdcbe20915
1 changed files with 4 additions and 3 deletions
  1. +4
    -3
      tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc

+ 4
- 3
tests/ut/ge/hybrid/node_executor/aicpu/aicpu_node_executor_unittest.cc View File

@@ -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) {


Loading…
Cancel
Save