BAZEL_BIN_DIR := extra-deps/tensorflow-haskell/third_party/tensorflow/bazel-bin

TF_LIB_SO_BASE := $(BAZEL_BIN_DIR)/tensorflow/libtensorflow.so

TF_LIB_LINKS := $(TF_LIB_SO_BASE) $(TF_LIB_SO_BASE).2 $(TF_LIB_SO_BASE).2.10

.PHONY: all help setup submodules tflitebuild tf-lib-links libedgetpu libedgetpu-lib-links cabal.project

all: help

help:

@echo "This Makefile defines the following targets:"

@echo " 'setup': all of the following:"

@echo " 'submodules': Sets up the Git submodules"

@echo " 'tflitebuild': Builds tensorflow lite inside build/"

@echo " 'tfbuild': Builds full tensorflow with bazel"

@echo " 'tf-lib-links': Additional .so symlinks in bazel bin dir"

@echo " 'libedgetpu': Build libedgetpu.so"

@echo " 'libedgetpu-lib-links': Additional .so symlinks in libedgetpu bin dir"

@echo " 'cabal.project': Creates cabal.project from cabal.project.in (with envsubst)"

@echo "Note: DO NOT USE -j with this Makefile. Parallelism is exploited already, and you can only break things."

setup:

$(MAKE) submodules

$(MAKE) tflitebuild tfbuild

$(MAKE) tf-lib-links

$(MAKE) libedgetpu

$(MAKE) libedgetpu-lib-links

$(MAKE) cabal.project

submodules:

@if git status --porcelain | grep extra-deps >/dev/null; then \

echo "'git status' detects changes in the extra-deps directory, refusing to init submodules"; \

exit 1; \

fi

git submodule update --init --recursive

tflitebuild:

mkdir -p build

cd build && cmake ../extra-deps/tensorflow-haskell/third_party/tensorflow/tensorflow/lite -DBUILD_SHARED_LIBS=1

cd build && cmake --build . -j

tfbuild: bazel511

cd extra-deps/tensorflow-haskell/third_party/tensorflow && ../../../../bazel511 build //tensorflow:all //tensorflow/tools/pip_package:build_pip_package

cd extra-deps/tensorflow-haskell/third_party/tensorflow && ../../../../run-with-PATH-dir.sh python=python3 -- bazel-bin/tensorflow/tools/pip_package/build_pip_package ../../../../accelerate-tensorflow-lite/tf-python-venv

virtualenv accelerate-tensorflow-lite/tf-python-venv

accelerate-tensorflow-lite/tf-python-venv/bin/pip3 install --force-reinstall accelerate-tensorflow-lite/tf-python-venv/tensorflow-2.10.1-*.whl

bazel511:

curl -L https://github.com/bazelbuild/bazel/releases/download/5.1.1/bazel-5.1.1-linux-x86_64 >$@

chmod +x $@

tf-lib-links: $(TF_LIB_LINKS)

$(TF_LIB_LINKS):

ln -s libtensorflow.so.2.10.1 $@

ABSL_LD_FLAGS := $(addprefix -L$(PWD)/build/_deps/abseil-cpp-build/absl/,flags hash container strings)

libedgetpu:

cd extra-deps/libedgetpu && \

env TFROOT=$(PWD)/extra-deps/tensorflow-haskell/third_party/tensorflow \

make -f makefile_build/Makefile \

CXXFLAGS='-I$(PWD)/build/flatbuffers/include -I$(PWD)/build/abseil-cpp' \

LDFLAGS='-L$(PWD)/build/_deps/flatbuffers-build $(ABSL_LD_FLAGS)' \

FLATC=$(PWD)/extra-deps/tensorflow-haskell/third_party/tensorflow/bazel-bin/external/flatbuffers/flatc \

libedgetpu-throttled \

-j$(shell nproc)

libedgetpu-lib-links: extra-deps/libedgetpu/out/throttled/k8/libedgetpu.so

extra-deps/libedgetpu/out/throttled/k8/libedgetpu.so:

ln -s libedgetpu.so.1.0 $@

cabal.project: cabal.project.in

envsubst '$$PWD' <$< >$@