add macro

4 years ago · ef922e7c00
parent c47a89e236
commit ef922e7c00
1 changed files with 4 additions and 0 deletions
--- a/tests/ut/ge/CMakeLists.txt
+++ b/tests/ut/ge/CMakeLists.txt
@ -885,6 +885,10 @@ target_compile_options(ut_libge_others_utest PRIVATE
    -g --coverage -fprofile-arcs -ftest-coverage
 )

+target_compile_definitions(ut_libge_others_utest PRIVATE
+    $<$<STREQUAL:${ENABLE_OPEN_SRC},True>:ONLY_COMPILE_OPEN_SRC>
+)
+
 target_link_libraries(ut_libge_others_utest
    $<BUILD_INTERFACE:intf_pub>
    ge_load_common ge_execute_common ge_ut_common gtest gtest_main ascend_protobuf ${COMMON_SHARED_LIBRARIES} json -lrt -ldl -lgcov