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

---
 tests/ut/ge/CMakeLists.txt | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tests/ut/ge/CMakeLists.txt b/tests/ut/ge/CMakeLists.txt
index a3d803b7..bf625bf9 100755
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@@ -856,6 +856,7 @@ 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