From 84c6c10ceb2561b5a27751d43765fc873ea51312 Mon Sep 17 00:00:00 2001
From: chenyemeng <chenyemeng@huawei.com>
Date: Tue, 22 Dec 2020 20:51:10 +0800
Subject: [PATCH] GeTensor aligned addr & zero copy support

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

diff --git a/ge/graph/passes/inplace_support_check_pass.cc b/ge/graph/passes/inplace_support_check_pass.cc
index 06986677..c98aeda8 100644
--- a/ge/graph/passes/inplace_support_check_pass.cc
+++ b/ge/graph/passes/inplace_support_check_pass.cc
@@ -78,6 +78,7 @@ Status InplaceSupportCheckPass::Run(NodePtr &node) {
       return FAILED;
     }
     AddRePassNode(node);
+    break;
   }
 
   GELOGD("InplaceSupportCheckPass success");
diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt
index bf625bf9..4499041b 100755
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@@ -776,6 +776,7 @@ add_library(ge_load_common STATIC ${GRAPH_LOAD_COMMON_SRC_FILES} ${PROTO_SRCS} $
 
 target_compile_definitions(ge_load_common PRIVATE
     google=ascend_private
+	$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_link_libraries(ge_load_common PRIVATE
@@ -790,6 +791,7 @@ add_library(ge_execute_common STATIC ${GRAPH_EXECUTE_COMMON_SRC_FILES} ${PROTO_S
 
 target_compile_definitions(ge_execute_common PRIVATE
     google=ascend_private
+	$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_link_libraries(ge_execute_common PRIVATE
@@ -856,7 +858,6 @@ target_compile_options(ut_libge_multiparts_utest PRIVATE
 
 target_compile_definitions(ut_libge_multiparts_utest PRIVATE
     google=ascend_private
-	$<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_link_libraries(ut_libge_multiparts_utest