Install protobuf version compatible with the current TensorFlow version

Author: FloopCZ

tensorflow_cc/CMakeLists.txt

@@ -11,7 +11,7 @@ option(REQUIRE_CUDA "Make sure to find and use CUDA (implies ALLOW_CUDA)." OFF)
 set(LOCAL_RAM_RESOURCES 4096 CACHE STRING "The amount of local RAM resources passed to bazel scheduler (e.g., 4096).")
 set(LOCAL_CPU_RESOURCES HOST_CPUS CACHE STRING "The amount of local CPU cores passed to bazel scheduler (e.g., 2).")
 set(TENSORFLOW_TAG "v${version}" CACHE STRING "The tensorflow release tag to be checked out (default v${version}).")
-option(SYSTEM_PROTOBUF "Link to system protobuf." OFF)
+option(INSTALL_PROTOBUF "Install protobuf compatible with tensorflow version." OFF)
 set(CMAKE_CXX_STANDARD 14 CACHE STRING "The C++ standard for building and linking the library (e.g., 14).")
 
 # -------------
@@ -32,6 +32,9 @@ configure_file("cmake/build_tensorflow.sh.in" "build_tensorflow.sh" @ONLY)
 # ----------------------------------------------
 
 include(TensorflowBase)
+if(INSTALL_PROTOBUF)
+  include(ProtobufExternal)
+endif()
 
 # ------------------------------
 # Define Tensorflow_CC Interface
@@ -40,6 +43,10 @@ include(TensorflowBase)
 add_library(tensorflow_cc INTERFACE)
 target_compile_features(tensorflow_cc INTERFACE "cxx_std_${CMAKE_CXX_STANDARD}")
 
+if(INSTALL_PROTOBUF)
+    add_dependencies(tensorflow_cc protobuf-external)
+endif()
+
 # The include folders are sometimes contained under bazel-bin/bin/ and sometimes just bazel-bin.
 target_include_directories(
   tensorflow_cc INTERFACE
@@ -53,17 +60,6 @@ target_link_libraries(
   "${CMAKE_INSTALL_PREFIX}/lib/libtensorflow_cc.so.${PROJECT_VERSION_MAJOR}"
   dl pthread
 )
-if(SYSTEM_PROTOBUF)
-  find_package(Protobuf REQUIRED)
-  target_include_directories(
-    tensorflow_cc INTERFACE
-    "${Protobuf_INCLUDE_DIRS}"
-  )
-  target_link_libraries(
-    tensorflow_cc INTERFACE
-    "${Protobuf_LIBRARIES}"
-  )
-endif()
 
 # ----------------------------------------
 # Configure CMake Config and Version Files

tensorflow_cc/cmake/ProtobufExternal.cmake

@@ -0,0 +1,20 @@
+set(PROTOBUF_ARCHIVE https://github.com/protocolbuffers/protobuf/archive/v3.9.2.zip)
+
+ExternalProject_Add(
+  protobuf-external
+  PREFIX protobuf
+  URL "${PROTOBUF_ARCHIVE}"
+  BINARY_DIR "${CMAKE_CURRENT_BINARY_DIR}/protobuf"
+  CMAKE_CACHE_ARGS
+    "-DCMAKE_BUILD_TYPE:STRING=${CMAKE_BUILD_TYPE}"
+    "-Dprotobuf_BUILD_TESTS:BOOL=OFF"
+    "-Dprotobuf_BUILD_EXAMPLES:BOOL=OFF"
+    "-DCMAKE_CXX_COMPILER:STRING=${CMAKE_CXX_COMPILER}"
+  SOURCE_SUBDIR cmake
+  BUILD_ALWAYS 1
+  STEP_TARGETS build
+  INSTALL_COMMAND ""
+)
+
+install(SCRIPT "${CMAKE_CURRENT_BINARY_DIR}/protobuf/cmake_install.cmake")
+

tensorflow_cc/cmake/TensorflowBase.cmake

@@ -11,7 +11,8 @@ ExternalProject_Add(
   BUILD_IN_SOURCE 1
   UPDATE_COMMAND ""
   CONFIGURE_COMMAND ""
-  BUILD_COMMAND "${CMAKE_CURRENT_BINARY_DIR}/build_tensorflow.sh"
+  BUILD_COMMAND echo "*" > "${CMAKE_CURRENT_BINARY_DIR}/tensorflow/.bazelversion"
+        COMMAND "${CMAKE_CURRENT_BINARY_DIR}/build_tensorflow.sh"
   INSTALL_COMMAND "${CMAKE_CURRENT_SOURCE_DIR}/cmake/copy_links.sh" bazel-bin
           # The include folders are sometimes contained under bazel-bin/bin/ and sometimes just bazel-bin.
           # Later, we include both so let's make sure they both exist so that CMake does not complain.