From 1b6349833ff00d0d25d6928242718dca6ad8534e Mon Sep 17 00:00:00 2001
From: zhujingjing <zhujingjing6@huawei.com>
Date: Thu, 15 Jul 2021 20:10:53 +0800
Subject: [PATCH] GE supports aicore DEPEND_SHAPE_RANGE op update outputs shape

---
 tests/ut/ge/hybrid/ge_hybrid_unittest.cc | 4 ++++
 1 file changed, 4 insertions(+)

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<ComputeGraph>("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<CallbackManager>(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<std::unique_ptr<AiCoreOpTask>> 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