Browse Source

fix

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

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

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

Loading…
Cancel
Save