From 67282b45307f503173ce95a4cd47a40bc7198bec Mon Sep 17 00:00:00 2001
From: zhaoxinxin <zhaoxinxin1@huawei.com>
Date: Tue, 16 Mar 2021 20:26:48 +0800
Subject: [PATCH] 	modified:   tests/ut/common/graph/CMakeLists.txt 
 modified:   tests/ut/ge/CMakeLists.txt

---
 tests/ut/common/graph/CMakeLists.txt | 3 ++-
 tests/ut/ge/CMakeLists.txt           | 1 +
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/tests/ut/common/graph/CMakeLists.txt b/tests/ut/common/graph/CMakeLists.txt
index 44a2a97c..4aac5995 100644
--- a/tests/ut/common/graph/CMakeLists.txt
+++ b/tests/ut/common/graph/CMakeLists.txt
@@ -96,11 +96,12 @@ set(SRC_FILES
     "${GE_CODE_DIR}/metadef/graph/utils/tensor_utils.cc"
     "${GE_CODE_DIR}/metadef/ops/op_imp.cpp"
     "${GE_CODE_DIR}/metadef/graph/opsproto/opsproto_manager.cc"
-    "${GE_CODE_DIR}/metadef/graph/utils/transformer_utils.cc"
     "${GE_CODE_DIR}/metadef/graph/runtime_inference_context.cc"
     "${GE_CODE_DIR}/metadef/graph/ref_relation.cc"
     "${GE_CODE_DIR}/metadef/third_party/transformer/src/transfer_shape_according_to_format.cc"
     "${GE_CODE_DIR}/metadef/third_party/transformer/src/axis_util.cc"
+    "${GE_CODE_DIR}/metadef/third_party/transformer/src/expand_dimension.cc"
+    "${GE_CODE_DIR}/metadef/graph/utils/transformer_utils.cc"
 )
 
 #add_executable(ut_libgraph ${UT_FILES} ${SRC_FILES} ${PROTO_SRCS} ${PROTO_HDRS})
diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt
index 80636a20..aeedfedc 100755
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@@ -90,6 +90,7 @@ set(GRAPH_SRC_FILES
     "${GE_CODE_DIR}/metadef/graph/op_desc.cc"
     "${GE_CODE_DIR}/metadef/third_party/transformer/src/transfer_shape_according_to_format.cc"
     "${GE_CODE_DIR}/metadef/third_party/transformer/src/axis_util.cc"
+    "${GE_CODE_DIR}/metadef/third_party/transformer/src/expand_dimension.cc"
     "${GE_CODE_DIR}/metadef/graph/operator.cc"
     "${GE_CODE_DIR}/metadef/graph/operator_factory.cc"
     "${GE_CODE_DIR}/metadef/graph/operator_factory_impl.cc"