From d986f9aa5e44824edae3734105ca4ef62af1761c Mon Sep 17 00:00:00 2001
From: chenyemeng <chenyemeng@huawei.com>
Date: Wed, 30 Dec 2020 11:55:07 +0800
Subject: [PATCH] add macro

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

diff --git a/tests/ut/common/graph/CMakeLists.txt b/tests/ut/common/graph/CMakeLists.txt
index 99b21182..e2490150 100644
--- a/tests/ut/common/graph/CMakeLists.txt
+++ b/tests/ut/common/graph/CMakeLists.txt
@@ -110,6 +110,7 @@ target_compile_options(ut_libgraph PRIVATE
 
 target_compile_definitions(ut_libgraph PRIVATE
     google=ascend_private
+    $<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
 )
 
 target_link_libraries(ut_libgraph