parent
caccf2dcf7
commit
1a6b7dd704
|
@ -0,0 +1 @@
|
|||
../Formula/libtensorflow.rb
|
|
@ -0,0 +1,80 @@
|
|||
class LibtensorflowAT1 < Formula
|
||||
include Language::Python::Virtualenv
|
||||
|
||||
desc "C interface for Google's OS library for Machine Intelligence"
|
||||
homepage "https://www.tensorflow.org/"
|
||||
url "https://github.com/tensorflow/tensorflow/archive/v1.14.0.tar.gz"
|
||||
sha256 "aa2a6a1daafa3af66807cfe0bc77bfe1144a9a53df9a96bab52e3e575b3047ed"
|
||||
|
||||
keg_only :versioned_formula
|
||||
|
||||
depends_on "bazel" => :build
|
||||
depends_on :java => ["1.8", :build]
|
||||
depends_on "python" => :build
|
||||
|
||||
# Upgrade protobuf to 3.8.0
|
||||
# The custom commit contains a fix to make protobuf.bzl compatible with Bazel 0.26 or later version.
|
||||
patch do
|
||||
url "https://github.com/tensorflow/tensorflow/commit/508f76b1d9925304cedd56d51480ec380636cb82.diff?full_index=1"
|
||||
sha256 "89f09f266ee56ee583cfffb8b4ce9333f181f497f7e04a672a68a8b611d21270"
|
||||
end
|
||||
|
||||
def install
|
||||
venv_root = "#{buildpath}/venv"
|
||||
virtualenv_create(venv_root, "python3")
|
||||
|
||||
cmd = Language::Java.java_home_cmd("1.8")
|
||||
ENV["JAVA_HOME"] = Utils.popen_read(cmd).chomp
|
||||
|
||||
ENV["PYTHON_BIN_PATH"] = "#{venv_root}/bin/python"
|
||||
ENV["CC_OPT_FLAGS"] = "-march=native"
|
||||
ENV["TF_IGNORE_MAX_BAZEL_VERSION"] = "1"
|
||||
ENV["TF_NEED_JEMALLOC"] = "1"
|
||||
ENV["TF_NEED_GCP"] = "0"
|
||||
ENV["TF_NEED_HDFS"] = "0"
|
||||
ENV["TF_ENABLE_XLA"] = "0"
|
||||
ENV["USE_DEFAULT_PYTHON_LIB_PATH"] = "1"
|
||||
ENV["TF_NEED_OPENCL"] = "0"
|
||||
ENV["TF_NEED_CUDA"] = "0"
|
||||
ENV["TF_NEED_MKL"] = "0"
|
||||
ENV["TF_NEED_VERBS"] = "0"
|
||||
ENV["TF_NEED_MPI"] = "0"
|
||||
ENV["TF_NEED_S3"] = "1"
|
||||
ENV["TF_NEED_GDR"] = "0"
|
||||
ENV["TF_NEED_KAFKA"] = "0"
|
||||
ENV["TF_NEED_OPENCL_SYCL"] = "0"
|
||||
ENV["TF_NEED_ROCM"] = "0"
|
||||
ENV["TF_DOWNLOAD_CLANG"] = "0"
|
||||
ENV["TF_SET_ANDROID_WORKSPACE"] = "0"
|
||||
ENV["TF_CONFIGURE_IOS"] = "0"
|
||||
system "./configure"
|
||||
|
||||
system "bazel", "build", "--jobs", ENV.make_jobs, "--compilation_mode=opt", "--copt=-march=native", "tensorflow:libtensorflow.so"
|
||||
lib.install Dir["bazel-bin/tensorflow/*.so*", "bazel-bin/tensorflow/*.dylib*"]
|
||||
(include/"tensorflow/c").install %w[
|
||||
tensorflow/c/c_api.h
|
||||
tensorflow/c/c_api_experimental.h
|
||||
tensorflow/c/tf_attrtype.h
|
||||
]
|
||||
|
||||
(lib/"pkgconfig/tensorflow.pc").write <<~EOS
|
||||
Name: tensorflow
|
||||
Description: Tensorflow library
|
||||
Version: #{version}
|
||||
Libs: -L#{lib} -ltensorflow
|
||||
Cflags: -I#{include}
|
||||
EOS
|
||||
end
|
||||
|
||||
test do
|
||||
(testpath/"test.c").write <<~EOS
|
||||
#include <stdio.h>
|
||||
#include <tensorflow/c/c_api.h>
|
||||
int main() {
|
||||
printf("%s", TF_Version());
|
||||
}
|
||||
EOS
|
||||
system ENV.cc, "-L#{lib}", "-I#{include}", "-ltensorflow", "-o", "test_tf", "test.c"
|
||||
assert_equal version, shell_output("./test_tf")
|
||||
end
|
||||
end
|
Loading…
Reference in New Issue