From 42f04ecbbdbdc8d08c0b9791bf2f8b8109ca3cf5 Mon Sep 17 00:00:00 2001
From: wangxiaotian22 <wangxiaotian4@huawei.com>
Date: Thu, 4 Mar 2021 15:16:34 +0800
Subject: [PATCH] ut compile

---
 tests/ut/ge/graph/build/graph_builder_unittest.cc | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tests/ut/ge/graph/build/graph_builder_unittest.cc b/tests/ut/ge/graph/build/graph_builder_unittest.cc
index f44c23a6..07615d3d 100644
--- a/tests/ut/ge/graph/build/graph_builder_unittest.cc
+++ b/tests/ut/ge/graph/build/graph_builder_unittest.cc
@@ -24,7 +24,7 @@
 #undef protected
 #undef private
 
-#include "graph/build/logical_stream_allocator.h"
+#include "graph/build/graph_builder.h"
 
 #include "common/types.h"
 #include "common/util.h"
@@ -101,8 +101,8 @@ TEST_F(UtestGraphBuilder, BuildForKnownShapeGraph_test) {
   ge::ComputeGraphPtr graph = make_shared<ge::ComputeGraph>("");
   MakeGraph(graph);
 
-  GraphBuilder builder;
-  GeModelPtr ge_model_ptr = nullptr;
+  ge::GraphBuilder builder;
+  ge::GeModelPtr ge_model_ptr = nullptr;
   builder.BuildForKnownShapeGraph(graph, ge_model_ptr, 0);
  }