diff --git a/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc b/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc index 068ef5f4..d5229b2c 100644 --- a/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc +++ b/tests/ut/ge/graph/preprocess/graph_preprocess_unittest.cc @@ -22,6 +22,7 @@ #include "common/util.h" #include "graph/passes/graph_builder_utils.h" #include "graph/utils/attr_utils.h" +#include "graph/debug/ge_attr_define.h" #define private public #define protected public @@ -65,7 +66,6 @@ TEST_F(UtestGraphPreproces, test_dynamic_input_shape_parse) { EXPECT_EQ(ret, ge::SUCCESS); // check data node output shape_range and shape auto data_node = graph_prepare.compute_graph_->FindNode("data1"); -) auto data_output_desc = data_node->GetOpDesc()->GetOutputDescPtr(0); vector expect_shape = {3,-1,-1,5}; auto result_shape = data_output_desc->GetShape();