Browse Source

fix ut

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

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

@@ -240,7 +240,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task) {
kernel_def.set_args_size(args.head.length);
domi::KernelDef *kernel_def_tmp = task_def.mutable_kernel();
*kernel_def_tmp = kernel_def;
hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def});
AicpuNodeTask aicpu_node_task(node_item, task_def);
ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS);
ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS);
@@ -323,7 +323,7 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_blocking_node_task_fail) {
kernel_def.set_args_size(args.head.length);
domi::KernelDef *kernel_def_tmp = task_def.mutable_kernel();
*kernel_def_tmp = kernel_def;
hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def});
AicpuNodeTask aicpu_node_task(node_item, task_def);
RTS_STUB_RETURN_VALUE(rtGetDevice, rtError_t, 0x78000001);


Loading…
Cancel
Save