/
workspace1.bzl
41 lines (33 loc) · 1.71 KB
/
workspace1.bzl
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
load("//third_party/android:android_configure.bzl", "android_configure")
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("@com_github_grpc_grpc//bazel:grpc_deps.bzl", "grpc_deps")
load("@com_google_benchmark//:bazel/benchmark_deps.bzl", "benchmark_deps")
load("@io_bazel_rules_closure//closure:defs.bzl", "closure_repositories")
load("@rules_cuda//cuda:dependencies.bzl", "rules_cuda_dependencies")
load("@rules_pkg//:deps.bzl", "rules_pkg_dependencies")
# buildifier: disable=unnamed-macro
def workspace(with_rules_cc = True):
"""Loads a set of TensorFlow dependencies. To be used in a WORKSPACE file.
Args:
with_rules_cc: whether to load and patch rules_cc repository.
"""
native.register_toolchains("@local_config_python//:py_toolchain")
rules_cuda_dependencies(with_rules_cc)
rules_pkg_dependencies()
closure_repositories()
http_archive(
name = "bazel_toolchains",
sha256 = "294cdd859e57fcaf101d4301978c408c88683fbc46fbc1a3829da92afbea55fb",
strip_prefix = "bazel-toolchains-8c717f8258cd5f6c7a45b97d974292755852b658",
urls = [
"http://mirror.tensorflow.org/github.com/bazelbuild/bazel-toolchains/archive/8c717f8258cd5f6c7a45b97d974292755852b658.tar.gz",
"https://github.com/bazelbuild/bazel-toolchains/archive/8c717f8258cd5f6c7a45b97d974292755852b658.tar.gz",
],
)
android_configure(name = "local_config_android")
grpc_deps()
benchmark_deps()
# Alias so it can be loaded without assigning to a different symbol to prevent
# shadowing previous loads and trigger a buildifier warning.
tf_workspace1 = workspace