20 lines
884 B
Makefile
20 lines
884 B
Makefile
|
TF_VERSION=1.15.2
|
||
|
|
|
||
|
|
# -march see https://gcc.gnu.org/onlinedocs/gcc-4.8.5/gcc/i386-and-x86-64-Options.html
|
||
|
|
|
||
|
|
all: libtensorflow libtensorflow-avx libtensorflow-avx2
|
||
|
|
patch:
|
||
|
|
git apply tensorflow-$(TF_VERSION).diff
|
||
|
|
download:
|
||
|
|
wget https://github.com/tensorflow/tensorflow/archive/v$(TF_VERSION).tar.gz
|
||
|
|
tar -xzf v$(TF_VERSION).tar.gz
|
||
|
|
cp .tf_configure.bazelrc Makefile ../*.sh tensorflow-$(TF_VERSION)
|
||
|
|
libtensorflow:
|
||
|
|
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so
|
||
|
|
./create_archive.sh osx-cpu $(TF_VERSION)
|
||
|
|
libtensorflow-avx:
|
||
|
|
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so --copt=-march=core-avx-i --host_copt=-march=core-avx-i
|
||
|
|
./create_archive.sh osx-avx $(TF_VERSION)
|
||
|
|
libtensorflow-avx2:
|
||
|
|
bazel build --jobs 2 --config=opt //tensorflow:libtensorflow.so --copt=-march=core-avx2 --host_copt=-march=core-avx2
|
||
|
|
./create_archive.sh osx-avx2 $(TF_VERSION)
|