| @@ -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(reinterpret_cast<const char *>(&args), args.head.length); | ||||
| task_def2.mutable_kernel()->set_args_size(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); | 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); | ASSERT_EQ(aicpu_node_task.LaunchTask(*node_state->GetTaskContext()), SUCCESS); | ||||
| //kernel_ex_def->set_allocated_kernel_ext_info(nullptr); | //kernel_ex_def->set_allocated_kernel_ext_info(nullptr); | ||||
| free(ext_mem); | free(ext_mem); | ||||
| } | } | ||||
| } // namespace ge | } // namespace ge | ||||