mirror of
https://github.com/ROCm/jax.git
synced 2025-04-16 03:46:06 +00:00
51 lines
1.6 KiB
Python
51 lines
1.6 KiB
Python
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
|
|
|
|
# To update XLA 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/openxla/xla/archive/<git hash>.tar.gz | sha256sum
|
|
# and update the sha256 with the result.
|
|
http_archive(
|
|
name = "xla",
|
|
sha256 = "8188f528ce2c81b62f6cc01b7b30bee21c276dd79c881c06813fc408d487c48f",
|
|
strip_prefix = "xla-d10fb016209af268e86e3b0f0083fd3c283c4592",
|
|
urls = [
|
|
"https://github.com/openxla/xla/archive/d10fb016209af268e86e3b0f0083fd3c283c4592.tar.gz",
|
|
],
|
|
)
|
|
|
|
# For development, one often wants to make changes to the TF repository as well
|
|
# as the JAX repository. You can override the pinned repository above with a
|
|
# local checkout by either:
|
|
# a) overriding the TF repository on the build.py command line by passing a flag
|
|
# like:
|
|
# python build/build.py --bazel_options=--override_repository=xla=/path/to/xla
|
|
# or
|
|
# b) by commenting out the http_archive above and uncommenting the following:
|
|
# local_repository(
|
|
# name = "xla",
|
|
# path = "/path/to/xla",
|
|
# )
|
|
|
|
load("//third_party/ducc:workspace.bzl", ducc = "repo")
|
|
ducc()
|
|
|
|
load("@xla//:workspace4.bzl", "xla_workspace4")
|
|
xla_workspace4()
|
|
|
|
load("@xla//:workspace3.bzl", "xla_workspace3")
|
|
xla_workspace3()
|
|
|
|
load("@xla//:workspace2.bzl", "xla_workspace2")
|
|
xla_workspace2()
|
|
|
|
load("@xla//:workspace1.bzl", "xla_workspace1")
|
|
xla_workspace1()
|
|
|
|
load("@xla//:workspace0.bzl", "xla_workspace0")
|
|
xla_workspace0()
|
|
|
|
|
|
load("//third_party/flatbuffers:workspace.bzl", flatbuffers = "repo")
|
|
flatbuffers()
|