add macro

4 years ago · d986f9aa5e
parent f5390707a4
commit d986f9aa5e
1 changed files with 1 additions and 0 deletions
--- 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