mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 21:06:06 +00:00
f60d927df8
Make specification of CUDA environment variables optional. Fixes #627. Fixes #276, although the fix requires a new Jaxlib release.
45 lines
1.7 KiB
Python
45 lines
1.7 KiB
Python
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
http_archive(
|
|
name = "io_bazel_rules_closure",
|
|
sha256 = "43c9b882fa921923bcba764453f4058d102bece35a37c9f6383c713004aacff1",
|
|
strip_prefix = "rules_closure-9889e2348259a5aad7e805547c1a0cf311cfcd91",
|
|
urls = [
|
|
"https://mirror.bazel.build/github.com/bazelbuild/rules_closure/archive/9889e2348259a5aad7e805547c1a0cf311cfcd91.tar.gz",
|
|
"https://github.com/bazelbuild/rules_closure/archive/9889e2348259a5aad7e805547c1a0cf311cfcd91.tar.gz", # 2018-12-21
|
|
],
|
|
)
|
|
|
|
http_archive(
|
|
name = "bazel_skylib",
|
|
sha256 = "2ef429f5d7ce7111263289644d233707dba35e39696377ebab8b0bc701f7818e",
|
|
urls = ["https://github.com/bazelbuild/bazel-skylib/releases/download/0.8.0/bazel-skylib.0.8.0.tar.gz"],
|
|
) # https://github.com/bazelbuild/bazel-skylib/releases
|
|
|
|
# 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 = "732ccdb272a275014c28e757c30a4575dc100865a8cb30ffc4c39b7d4bc761a2",
|
|
strip_prefix = "tensorflow-afeb3a07811b201f53e365807ac5173e4bad1c2e",
|
|
urls = [
|
|
"https://github.com/tensorflow/tensorflow/archive/afeb3a07811b201f53e365807ac5173e4bad1c2e.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_workspace(
|
|
path_prefix = "",
|
|
tf_repo_name = "org_tensorflow",
|
|
)
|