| @@ -17,6 +17,10 @@ | |||||
| #include <gtest/gtest.h> | #include <gtest/gtest.h> | ||||
| #include "graph/partition/dynamic_shape_partition.h" | #include "graph/partition/dynamic_shape_partition.h" | ||||
| #include "compute_graph.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 private public | ||||
| #define protected public | #define protected public | ||||
| @@ -75,7 +79,7 @@ class UtestDynamicShapePartition : public testing::Test { | |||||
| // test Init_EndGraphTaskInfo_failed | // test Init_EndGraphTaskInfo_failed | ||||
| TEST_F(UtestDynamicShapePartition, single_op_scene_success) { | TEST_F(UtestDynamicShapePartition, single_op_scene_success) { | ||||
| ComputeGraphPtr computeGraph("default"); | |||||
| ComputeGraphPtr graph = shared_ptr<ComputeGraph>("default"); | |||||
| NodePtr node1 = | NodePtr node1 = | ||||
| NodeBuilder("node1", CONSTANTOP).AddInputDesc({1, 1, 224, 224}).AddOutputDesc({1, 1, 224, 224}).Build(graph); | NodeBuilder("node1", CONSTANTOP).AddInputDesc({1, 1, 224, 224}).AddOutputDesc({1, 1, 224, 224}).Build(graph); | ||||