diff --git a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc index b09211cb..e8c13a82 100644 --- a/tests/ut/ge/hybrid/ge_hybrid_unittest.cc +++ b/tests/ut/ge/hybrid/ge_hybrid_unittest.cc @@ -807,6 +807,7 @@ TEST_F(UtestGeHybrid, TestParseDependencies) { } TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { + dlog_setlevel(0, 0, 0); auto graph = make_shared("graph"); OpDescPtr op_desc = CreateOpDesc("Add", "Add"); GeShape shape({2, 16}); @@ -821,6 +822,8 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { node_item->output_start = 0; GraphExecutionContext execution_context; + execution_context.callback_manager = + std::unique_ptr(new (std::nothrow) CallbackManager()); GraphItem graph_item; SubgraphContext subgraph_context(&graph_item, &execution_context); ASSERT_EQ(subgraph_context.Init(), SUCCESS); @@ -833,5 +836,6 @@ TEST_F(UtestGeHybrid, TestTaskExecuteAsync) { std::vector> tasks; AiCoreNodeTask node_task(std::move(tasks)); ASSERT_EQ(node_task.ExecuteAsync(task_context, nullptr), SUCCESS); + dlog_setlevel(0, 3, 0); } } // namespace ge \ No newline at end of file