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