wehub-resource-sync
100 lines
4.0 KiB
Python
100 lines
4.0 KiB
Python
"""TensorFlow workspace initialization. Consult the WORKSPACE on how to use it."""
|
|
|
|
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
|
|
|
|
# buildifier: disable=function-docstring
|
|
# buildifier: disable=unnamed-macro
|
|
def workspace():
|
|
tf_http_archive(
|
|
name = "rules_proto",
|
|
sha256 = "20b240eba17a36be4b0b22635aca63053913d5c1ee36e16be36499d167a2f533",
|
|
strip_prefix = "rules_proto-11bf7c25e666dd7ddacbcd4d4c4a9de7a25175f8",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_proto/archive/11bf7c25e666dd7ddacbcd4d4c4a9de7a25175f8.tar.gz",
|
|
),
|
|
patch_file = ["//third_party:rules_proto.patch"],
|
|
)
|
|
|
|
# https://github.com/bazelbuild/bazel-skylib/releases
|
|
tf_http_archive(
|
|
name = "bazel_skylib",
|
|
sha256 = "bc283cdfcd526a52c3201279cda4bc298652efa898b10b4db0837dc51652756f",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/bazel-skylib/releases/download/1.7.1/bazel-skylib-1.7.1.tar.gz",
|
|
),
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "rules_license",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_license/releases/download/0.0.7/rules_license-0.0.7.tar.gz",
|
|
),
|
|
sha256 = "4531deccb913639c30e5c7512a054d5d875698daeb75d8cf90f284375fe7c360",
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "rules_pkg",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/rules_pkg/releases/download/0.7.1/rules_pkg-0.7.1.tar.gz",
|
|
),
|
|
sha256 = "451e08a4d78988c06fa3f9306ec813b836b1d076d0f055595444ba4ff22b867f",
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "bazel_features",
|
|
sha256 = "4fd9922d464686820ffd8fcefa28ccffa147f7cdc6b6ac0d8b07fde565c65d66",
|
|
strip_prefix = "bazel_features-1.25.0",
|
|
urls = tf_mirror_urls("https://github.com/bazel-contrib/bazel_features/releases/download/v1.25.0/bazel_features-v1.25.0.tar.gz"),
|
|
)
|
|
|
|
tf_http_archive(
|
|
name = "rules_cc",
|
|
urls = tf_mirror_urls("https://github.com/bazelbuild/rules_cc/releases/download/0.2.0/rules_cc-0.2.0.tar.gz"),
|
|
strip_prefix = "rules_cc-0.2.0",
|
|
sha256 = "ae244f400218f4a12ee81658ff246c0be5cb02c5ca2de5519ed505a6795431e9",
|
|
patch_file = [
|
|
"@xla//third_party/py:rules_cc_protobuf.patch",
|
|
],
|
|
)
|
|
|
|
# Toolchains for ML projects hermetic builds.
|
|
# Details: https://github.com/google-ml-infra/rules_ml_toolchain
|
|
tf_http_archive(
|
|
name = "rules_ml_toolchain",
|
|
sha256 = "6b1b294cbecb98a08bcfe59641461f72e113db30fa91b44fcc49d98099b5ba65",
|
|
strip_prefix = "rules_ml_toolchain-b11745590f513ec55b32e2d126073576fde18c71",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/google-ml-infra/rules_ml_toolchain/archive/b11745590f513ec55b32e2d126073576fde18c71.tar.gz",
|
|
),
|
|
)
|
|
|
|
# Maven dependencies.
|
|
RULES_JVM_EXTERNAL_TAG = "4.3"
|
|
tf_http_archive(
|
|
name = "rules_jvm_external",
|
|
strip_prefix = "rules_jvm_external-%s" % RULES_JVM_EXTERNAL_TAG,
|
|
sha256 = "6274687f6fc5783b589f56a2f1ed60de3ce1f99bc4e8f9edef3de43bdf7c6e74",
|
|
urls = tf_mirror_urls("https://github.com/bazelbuild/rules_jvm_external/archive/%s.zip" % RULES_JVM_EXTERNAL_TAG),
|
|
)
|
|
|
|
# Platforms
|
|
tf_http_archive(
|
|
name = "platforms",
|
|
urls = tf_mirror_urls(
|
|
"https://github.com/bazelbuild/platforms/releases/download/0.0.11/platforms-0.0.11.tar.gz",
|
|
),
|
|
sha256 = "29742e87275809b5e598dc2f04d86960cc7a55b3067d97221c9abbc9926bff0f",
|
|
)
|
|
|
|
# clang-tidy checks.
|
|
tf_http_archive(
|
|
name = "bazel_clang_tidy",
|
|
strip_prefix = "bazel_clang_tidy-c4d35e0d0b838309358e57a2efed831780f85cd0",
|
|
sha256 = "96da6e935ccc91045cf928dbc57f22508a2729c51f7fb3f56178017b0deb9b3c",
|
|
urls = tf_mirror_urls("https://github.com/erenon/bazel_clang_tidy/archive/c4d35e0d0b838309358e57a2efed831780f85cd0.tar.gz"),
|
|
)
|
|
|
|
# Alias so it can be loaded without assigning to a different symbol to prevent
|
|
# shadowing previous loads and trigger a buildifier warning.
|
|
xla_workspace3 = workspace
|