|
|
@@ -151,18 +151,16 @@ TEST_F(UtestAicpuNodeExecutor, aicpu_tf_node_task) { |
|
|
|
task_def2.mutable_kernel()->set_args(reinterpret_cast<const char *>(&args), args.head.length);
|
|
|
|
task_def2.mutable_kernel()->set_args_size(args.head.length);
|
|
|
|
|
|
|
|
hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2});
|
|
|
|
hybrid_model.task_defs_[node] = std::vector<domi::TaskDef>({task_def2, task_def2});
|
|
|
|
|
|
|
|
AicpuNodeTask aicpu_node_task(node_item, task_def);
|
|
|
|
ASSERT_EQ(aicpu_node_task.Init(hybrid_model), FAILED);
|
|
|
|
ASSERT_EQ(aicpu_node_task.Init(hybrid_model), SUCCESS);
|
|
|
|
ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS);
|
|
|
|
|
|
|
|
|
|
|
|
//kernel_ex_def->set_allocated_kernel_ext_info(nullptr);
|
|
|
|
|
|
|
|
free(ext_mem);
|
|
|
|
|
|
|
|
}
|
|
|
|
|
|
|
|
} // namespace ge
|
|
|
|
|