From 941dcdc7aa95b5ad541e05c860c9745df99a84ea Mon Sep 17 00:00:00 2001
From: chenyemeng <chenyemeng@huawei.com>
Date: Wed, 30 Dec 2020 14:38:18 +0800
Subject: [PATCH] add macro

---
 ge/offline/CMakeLists.txt | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/ge/offline/CMakeLists.txt b/ge/offline/CMakeLists.txt
index d195e06f..cb6a3a50 100644
--- a/ge/offline/CMakeLists.txt
+++ b/ge/offline/CMakeLists.txt
@@ -30,6 +30,7 @@ target_compile_definitions(atc PRIVATE
     COMPILE_OMG_PACKAGE
     google=ascend_private
     LOG_CPP
+    $<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_include_directories(atc PRIVATE
@@ -92,6 +93,7 @@ target_compile_definitions(atc_atc.bin PRIVATE
     COMPILE_OMG_PACKAGE
     google=ascend_private
     LOG_CPP
+    $<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_include_directories(atc_atc.bin PRIVATE
@@ -152,6 +154,7 @@ target_compile_options(fwk_atc.bin PRIVATE
     -O2
     -Wno-deprecated-declarations
     -fno-common
+    $<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_compile_definitions(fwk_atc.bin PRIVATE