diff --git a/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc b/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
index 6a44f5ab..433f098a 100644
--- a/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
+++ b/tests/ut/ge/graph/partition/dynamic_shape_partition_unittest.cc
@@ -17,6 +17,10 @@
 #include <gtest/gtest.h>
 #include "graph/partition/dynamic_shape_partition.h"
 #include "compute_graph.h"
+#include "inc/framework/common/types.h"
+#include "utils/graph_utils.h"
+#include "graph/debug/ge_attr_define.h"
+
 
 #define private public
 #define protected public
@@ -75,7 +79,7 @@ class UtestDynamicShapePartition : public testing::Test {
 
 // test Init_EndGraphTaskInfo_failed
 TEST_F(UtestDynamicShapePartition, single_op_scene_success) {
-  ComputeGraphPtr computeGraph("default");
+  ComputeGraphPtr graph = shared_ptr<ComputeGraph>("default");
 
   NodePtr node1 =
       NodeBuilder("node1", CONSTANTOP).AddInputDesc({1, 1, 224, 224}).AddOutputDesc({1, 1, 224, 224}).Build(graph);