| @@ -24,7 +24,7 @@ | |||||
| #undef protected | #undef protected | ||||
| #undef private | #undef private | ||||
| #include "graph/build/logical_stream_allocator.h" | |||||
| #include "graph/build/graph_builder.h" | |||||
| #include "common/types.h" | #include "common/types.h" | ||||
| #include "common/util.h" | #include "common/util.h" | ||||
| @@ -101,8 +101,8 @@ TEST_F(UtestGraphBuilder, BuildForKnownShapeGraph_test) { | |||||
| ge::ComputeGraphPtr graph = make_shared<ge::ComputeGraph>(""); | ge::ComputeGraphPtr graph = make_shared<ge::ComputeGraph>(""); | ||||
| MakeGraph(graph); | 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); | builder.BuildForKnownShapeGraph(graph, ge_model_ptr, 0); | ||||
| } | } | ||||