diff options
Diffstat (limited to 'tensorflow/contrib/cmake/tf_core_framework.cmake')
| -rw-r--r-- | tensorflow/contrib/cmake/tf_core_framework.cmake | 31 |
1 files changed, 15 insertions, 16 deletions
diff --git a/tensorflow/contrib/cmake/tf_core_framework.cmake b/tensorflow/contrib/cmake/tf_core_framework.cmake index f38697097c..cad3b7864d 100644 --- a/tensorflow/contrib/cmake/tf_core_framework.cmake +++ b/tensorflow/contrib/cmake/tf_core_framework.cmake @@ -146,6 +146,14 @@ file(GLOB_RECURSE tf_core_lib_test_srcs list(REMOVE_ITEM tf_core_lib_srcs ${tf_core_lib_test_srcs}) +if(NOT tensorflow_ENABLE_SSL_SUPPORT) + file(GLOB_RECURSE tf_core_lib_cloud_srcs + "${tensorflow_source_dir}/tensorflow/core/platform/cloud/*.h" + "${tensorflow_source_dir}/tensorflow/core/platform/cloud/*.cc" + ) + list(REMOVE_ITEM tf_core_lib_srcs ${tf_core_lib_cloud_srcs}) +endif() + add_library(tf_core_lib OBJECT ${tf_core_lib_srcs}) target_include_directories(tf_core_lib PUBLIC ${tensorflow_source_dir} @@ -153,9 +161,7 @@ target_include_directories(tf_core_lib PUBLIC ${jpeg_INCLUDE_DIR} ${png_INCLUDE_DIR} ${eigen_INCLUDE_DIRS} - ${re2_EXTRA_INCLUDE_DIR} ${jsoncpp_INCLUDE_DIR} - ${boringssl_INCLUDE_DIR} ) target_compile_options(tf_core_lib PRIVATE -fno-exceptions @@ -171,12 +177,16 @@ add_dependencies(tf_core_lib gif_copy_headers_to_destination jpeg_copy_headers_to_destination png_copy_headers_to_destination - re2_copy_headers_to_destination eigen tf_protos_cc jsoncpp - boringssl -) + ) + +if(tensorflow_ENABLE_SSL_SUPPORT) + target_include_directories(tf_core_lib PUBLIC ${boringssl_INCLUDE_DIR}) + add_dependencies(tf_core_lib boringssl) +endif() + # Tricky setup to force always rebuilding # force_rebuild always runs forcing ${VERSION_INFO_CC} target to run @@ -230,18 +240,7 @@ add_library(tf_core_framework OBJECT target_include_directories(tf_core_framework PUBLIC ${tensorflow_source_dir} ${eigen_INCLUDE_DIRS} - ${re2_INCLUDES} ) -#target_link_libraries(tf_core_framework -# ${CMAKE_THREAD_LIBS_INIT} -# ${PROTOBUF_LIBRARIES} -# #${re2_STATIC_LIBRARIES} -# re2_lib -# ${jpeg_STATIC_LIBRARIES} -# ${png_STATIC_LIBRARIES} -# tf_protos_cc -# tf_core_lib -#) add_dependencies(tf_core_framework tf_core_lib proto_text |