mirror of
https://github.com/ROCm/jax.git
synced 2025-04-18 12:56:07 +00:00
79a7f7bca8
Update XLA. CUDA 11.1 wheels are compatible with CUDA versions 11.1+, since NVidia now promises enhanced version compatibility between CUDA minor releases starting with CUDA 11.1
35 lines
1.2 KiB
Python
35 lines
1.2 KiB
Python
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
# 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 = "ce033795d4d58ecb92a0188d6881c29d5a126e62cd21dd1882eb902e2b5ac226",
|
|
strip_prefix = "tensorflow-559047cb46f6805dfc50cba6d91b5a1e2d8d1b68",
|
|
urls = [
|
|
"https://github.com/tensorflow/tensorflow/archive/559047cb46f6805dfc50cba6d91b5a1e2d8d1b68.tar.gz",
|
|
],
|
|
)
|
|
|
|
# For development, one can use a local TF repository instead.
|
|
# local_repository(
|
|
# name = "org_tensorflow",
|
|
# path = "tensorflow",
|
|
# )
|
|
|
|
load("//third_party/pocketfft:workspace.bzl", pocketfft = "repo")
|
|
pocketfft()
|
|
|
|
# Initialize TensorFlow's external dependencies.
|
|
load("@org_tensorflow//tensorflow:workspace3.bzl", "workspace")
|
|
workspace()
|
|
load("@org_tensorflow//tensorflow:workspace2.bzl", "workspace")
|
|
workspace()
|
|
load("@org_tensorflow//tensorflow:workspace1.bzl", "workspace")
|
|
workspace()
|
|
load("@org_tensorflow//tensorflow:workspace0.bzl", "workspace")
|
|
workspace()
|