| @@ -807,6 +807,7 @@ TEST_F(UtestGeHybrid, TestParseDependencies) { | |||||
| } | } | ||||
| TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { | TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { | ||||
| dlog_setlevel(0, 0, 0); | |||||
| auto graph = make_shared<ComputeGraph>("graph"); | auto graph = make_shared<ComputeGraph>("graph"); | ||||
| OpDescPtr op_desc = CreateOpDesc("Add", "Add"); | OpDescPtr op_desc = CreateOpDesc("Add", "Add"); | ||||
| GeShape shape({2, 16}); | GeShape shape({2, 16}); | ||||
| @@ -821,6 +822,8 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { | |||||
| node_item->output_start = 0; | node_item->output_start = 0; | ||||
| GraphExecutionContext execution_context; | GraphExecutionContext execution_context; | ||||
| execution_context.callback_manager = | |||||
| std::unique_ptr<CallbackManager>(new (std::nothrow) CallbackManager()); | |||||
| GraphItem graph_item; | GraphItem graph_item; | ||||
| SubgraphContext subgraph_context(&graph_item, &execution_context); | SubgraphContext subgraph_context(&graph_item, &execution_context); | ||||
| ASSERT_EQ(subgraph_context.Init(), SUCCESS); | ASSERT_EQ(subgraph_context.Init(), SUCCESS); | ||||
| @@ -833,5 +836,6 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { | |||||
| std::vector<std::unique_ptr<AiCoreOpTask>> tasks; | std::vector<std::unique_ptr<AiCoreOpTask>> tasks; | ||||
| AiCoreNodeTask node_task(std::move(tasks)); | AiCoreNodeTask node_task(std::move(tasks)); | ||||
| ASSERT_EQ(node_task.ExecuteAsync(task_context, nullptr), SUCCESS); | ASSERT_EQ(node_task.ExecuteAsync(task_context, nullptr), SUCCESS); | ||||
| dlog_setlevel(0, 3, 0); | |||||
| } | } | ||||
| } // namespace ge | } // namespace ge | ||||