mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 20:06:05 +00:00
52 lines
1.9 KiB
Python
52 lines
1.9 KiB
Python
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
http_archive(
|
|
name = "io_bazel_rules_closure",
|
|
sha256 = "5b00383d08dd71f28503736db0500b6fb4dda47489ff5fc6bed42557c07c6ba9",
|
|
strip_prefix = "rules_closure-308b05b2419edb5c8ee0471b67a40403df940149",
|
|
urls = [
|
|
"https://storage.googleapis.com/mirror.tensorflow.org/github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz",
|
|
"https://github.com/bazelbuild/rules_closure/archive/308b05b2419edb5c8ee0471b67a40403df940149.tar.gz", # 2019-06-13
|
|
],
|
|
)
|
|
|
|
|
|
# https://github.com/bazelbuild/bazel-skylib/releases
|
|
http_archive(
|
|
name = "bazel_skylib",
|
|
sha256 = "1dde365491125a3db70731e25658dfdd3bc5dbdfd11b840b3e987ecf043c7ca0",
|
|
urls = [
|
|
"http://mirror.tensorflow.org/github.com/bazelbuild/bazel-skylib/releases/download/0.9.0/bazel_skylib-0.9.0.tar.gz",
|
|
"https://github.com/bazelbuild/bazel-skylib/releases/download/0.9.0/bazel_skylib-0.9.0.tar.gz",
|
|
],
|
|
)
|
|
|
|
# To update TensorFlow to a new revision,
|
|
# a) update URL and strip_prefix to the new git commit hash
|
|
# b) get the sha256 hash of the commit by running:
|
|
# curl -L https://github.com/tensorflow/tensorflow/archive/<git hash>.tar.gz | sha256sum
|
|
# and update the sha256 with the result.
|
|
http_archive(
|
|
name = "org_tensorflow",
|
|
sha256 = "4ce0e08aa014fafa7a0e8fb3531bdc914bd8a49828e1f5c31bb8adfb751ad73d",
|
|
strip_prefix = "tensorflow-210649dd56d7c4b75e3e8e2a851b61c80ae13dbb",
|
|
urls = [
|
|
"https://github.com/tensorflow/tensorflow/archive/210649dd56d7c4b75e3e8e2a851b61c80ae13dbb.tar.gz",
|
|
],
|
|
)
|
|
|
|
# For development, one can use a local TF repository instead.
|
|
# local_repository(
|
|
# name = "org_tensorflow",
|
|
# path = "tensorflow",
|
|
# )
|
|
|
|
load("@org_tensorflow//tensorflow:workspace.bzl", "tf_workspace", "tf_bind")
|
|
|
|
tf_workspace(
|
|
path_prefix = "",
|
|
tf_repo_name = "org_tensorflow",
|
|
)
|
|
|
|
tf_bind()
|