Skip to content
Permalink
Branch: master
Find file Copy path
Find file Copy path
1 recent contributor

Users who have contributed to this file recently

86 lines (76 sloc) 2.78 KB
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.15.1.tar.gz"
sha256 "19b6e72bc8675937f618cede364d7228a71c2eeaffc42801bcefd98dda7ca056"
bottle do
cellar :any
sha256 "964aeb5885784c1a69cda633a2fa74e0c4bbb46639654711d438c7803bd45026" => :catalina
sha256 "964aeb5885784c1a69cda633a2fa74e0c4bbb46639654711d438c7803bd45026" => :mojave
sha256 "96f60a7e019d96effa1c5f51dee2fea199e1cc49dfa36fbf7b99ac86389df825" => :high_sierra
end
keg_only :versioned_formula
depends_on "bazel" => :build
depends_on :java => ["1.8", :build]
depends_on "python@3.8" => :build
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"
bazel_compatibility_flags = %w[
--noincompatible_remove_legacy_whole_archive
]
system "bazel", "build", "--jobs", ENV.make_jobs, "--compilation_mode=opt", "--copt=-march=native", *bazel_compatibility_flags, "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
tensorflow/c/tf_datatype.h
tensorflow/c/tf_status.h
tensorflow/c/tf_tensor.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
You can’t perform that action at this time.