Script to build TFLite and its dependencies (#79)

This script builds TFLite together with all its dependencies and
generates a CMake cache file that can be used from other projects.

Author: petrhosek

buildbot/build_tflite.sh

@@ -0,0 +1,145 @@
+# Copyright 2020 Google LLC
+#
+# Licensed under the Apache License, Version 2.0 (the "License");
+# you may not use this file except in compliance with the License.
+# You may obtain a copy of the License at
+#
+#      http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing, software
+# distributed under the License is distributed on an "AS IS" BASIS,
+# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+# See the License for the specific language governing permissions and
+# limitations under the License.
+
+#!/bin/bash
+
+set -e
+
+readonly CPUINFO_REPOSITORY="https://github.com/pytorch/cpuinfo"
+readonly CPUINFO_TAG="5e63739504f0f8e18e941bd63b2d6d42536c7d90"
+
+readonly RUY_REPOSITORY="https://github.com/google/ruy"
+readonly RUY_TAG="841ea4172ba904fe3536789497f9565f2ef64129"
+
+readonly ABSEIL_REPOSITORY="https://github.com/abseil/abseil-cpp"
+readonly ABSEIL_TAG="dc370a82467cb35066475537b797197aee3e5164"
+
+readonly EIGEN_REPOSITORY="https://gitlab.com/libeigen/eigen"
+readonly EIGEN_TAG="0e187141679fdb91da33249d18cb79a011c0e2ea"
+
+readonly NEON_2_SSE_REPOSITORY="https://github.com/intel/ARM_NEON_2_x86_SSE"
+readonly NEON_2_SSE_TAG="cef9501d1d1c47223466bf2b8cd43f3368c37773"
+
+readonly FLATBUFFERS_REPOSITORY="https://github.com/google/flatbuffers"
+readonly FLATBUFFERS_TAG="615616cb5549a34bdf288c04bc1b94bd7a65c396"
+
+readonly PROTOBUF_REPOSITORY="https://github.com/protocolbuffers/protobuf"
+readonly PROTOBUF_TAG="c9869dc7803eb0a21d7e589c40ff4f9288cd34ae"
+
+readonly TENSORFLOW_REPOSITORY="https://github.com/tensorflow/tensorflow"
+readonly TENSORFLOW_TAG="d74d591e8f9e42be640dd93cfa0584914e25e98e"
+
+# cpuinfo
+git clone --filter=tree:0 --no-checkout ${CPUINFO_REPOSITORY} cpuinfo/src/cpuinfo
+git -C cpuinfo/src/cpuinfo checkout ${CPUINFO_TAG}
+cmake -GNinja -S cpuinfo/src/cpuinfo -B cpuinfo/src/cpuinfo-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/cpuinfo \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -DCPUINFO_BUILD_UNIT_TESTS:BOOL=OFF \
+  -DCPUINFO_BUILD_MOCK_TESTS:BOOL=OFF \
+  -DCPUINFO_BUILD_BENCHMARKS:BOOL=OFF
+ninja -C cpuinfo/src/cpuinfo-build install
+
+# ruy
+git clone --filter=tree:0 --no-checkout ${RUY_REPOSITORY} ruy/src/ruy
+git -C ruy/src/ruy checkout ${RUY_TAG}
+cmake -GNinja -S ruy/src/ruy -B ruy/src/ruy-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/ruy \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -Dcpuinfo_DIR:PATH=${PWD}/cpuinfo/share/cpuinfo \
+  -DRUY_MINIMAL_BUILD:BOOL=ON \
+  -DRUY_ENABLE_INSTALL:BOOL=ON \
+  -DRUY_FIND_CPUINFO:BOOL=ON
+ninja -C ruy/src/ruy-build install
+
+# absl
+git clone --filter=tree:0 --no-checkout ${ABSEIL_REPOSITORY} abseil-cpp/src/abseil-cpp
+git -C abseil-cpp/src/abseil-cpp checkout ${ABSEIL_TAG}
+cmake -GNinja -S abseil-cpp/src/abseil-cpp -B abseil-cpp/src/abseil-cpp-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/abseil-cpp \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -DABSL_BUILD_TESTING:BOOL=OFF \
+  -DABSL_ENABLE_INSTALL:BOOL=ON
+ninja -C abseil-cpp/src/abseil-cpp-build install
+
+# eigen
+git clone --filter=tree:0 --no-checkout ${EIGEN_REPOSITORY} eigen/src/eigen
+git -C eigen/src/eigen checkout ${EIGEN_TAG}
+cmake -GNinja -S eigen/src/eigen -B eigen/src/eigen-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/eigen \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -DEIGEN_BUILD_DOC:BOOL=OFF \
+  -DEIGEN_BUILD_TESTING:BOOL=OFF
+ninja -C eigen/src/eigen-build install
+
+# ARM_NEON_2_x86_SSE
+git clone --filter=tree:0 --no-checkout ${NEON_2_SSE_REPOSITORY} ARM_NEON_2_x86_SSE/src/ARM_NEON_2_x86_SSE
+git -C ARM_NEON_2_x86_SSE/src/ARM_NEON_2_x86_SSE checkout ${NEON_2_SSE_TAG}
+cmake -GNinja -S ARM_NEON_2_x86_SSE/src/ARM_NEON_2_x86_SSE -B ARM_NEON_2_x86_SSE/src/ARM_NEON_2_x86_SSE-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/ARM_NEON_2_x86_SSE \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON
+ninja -C ARM_NEON_2_x86_SSE/src/ARM_NEON_2_x86_SSE-build install
+
+# flatbuffers
+git clone --filter=tree:0 --no-checkout ${FLATBUFFERS_REPOSITORY} flatbuffers/src/flatbuffers
+git -C flatbuffers/src/flatbuffers checkout ${FLATBUFFERS_TAG}
+cmake -GNinja -S flatbuffers/src/flatbuffers -B flatbuffers/src/flatbuffers-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/flatbuffers \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -DFLATBUFFERS_BUILD_TESTS:BOOL=OFF
+ninja -C flatbuffers/src/flatbuffers-build install
+
+# tflite
+git clone --filter=tree:0 --no-checkout ${TENSORFLOW_REPOSITORY} tensorflow/src/tensorflow
+git -C tensorflow/src/tensorflow checkout ${TENSORFLOW_TAG}
+cmake -GNinja -S tensorflow/src/tensorflow/tensorflow/lite -B tensorflow/src/tensorflow-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/tensorflow \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DTFLITE_ENABLE_INSTALL:BOOL=ON \
+  -DTFLITE_ENABLE_XNNPACK:BOOL=OFF \
+  -Dcpuinfo_DIR:PATH=${PWD}/cpuinfo/share/cpuinfo \
+  -Druy_DIR:PATH=${PWD}/ruy/lib/cmake/ruy \
+  -Dabsl_DIR:PATH=${PWD}/abseil-cpp/lib/cmake/absl \
+  -DEigen3_DIR:PATH=${PWD}/eigen/share/eigen3/cmake \
+  -DNEON_2_SSE_DIR:PATH=${PWD}/ARM_NEON_2_x86_SSE/lib/cmake/NEON_2_SSE \
+  -DFlatbuffers_DIR:PATH=${PWD}/flatbuffers/lib/cmake/flatbuffers
+ninja -C tensorflow/src/tensorflow-build install
+
+# protobuf
+git clone --filter=tree:0 --no-checkout ${PROTOBUF_REPOSITORY} protobuf/src/protobuf
+git -C protobuf/src/protobuf checkout ${PROTOBUF_TAG} 
+cmake -GNinja -S protobuf/src/protobuf/cmake -B protobuf/src/protobuf-build \
+  -DCMAKE_FIND_PACKAGE_PREFER_CONFIG:BOOL=ON \
+  -DCMAKE_INSTALL_PREFIX:PATH=${PWD}/protobuf \
+  -DCMAKE_POSITION_INDEPENDENT_CODE:BOOL=ON \
+  -Dprotobuf_BUILD_TESTS:BOOL=OFF
+ninja -C protobuf/src/protobuf-build install
+
+# CMake cache file
+cat <<EOF >>tflite.cmake
+set(cpuinfo_DIR "${PWD}/cpuinfo/share/cpuinfo" CACHE PATH "")
+set(ruy_DIR "${PWD}/ruy/lib/cmake/ruy" CACHE PATH "")
+set(absl_DIR "${PWD}/abseil-cpp/lib/cmake/absl" CACHE PATH "")
+set(Eigen3_DIR "${PWD}/eigen/share/eigen3/cmake" CACHE PATH "")
+set(NEON_2_SSE_DIR "${PWD}/ARM_NEON_2_x86_SSE/lib/cmake/NEON_2_SSE" CACHE PATH "")
+set(Flatbuffers_DIR "${PWD}/flatbuffers/lib/cmake/flatbuffers" CACHE PATH "")
+set(protobuf_DIR "${PWD}/output/lib/cmake/protobuf" CACHE PATH "")
+EOF