mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Move XLA dependency to using tf_http_archive to enable the ability for bazel to use a GCS mirror for the XLA dependency.

Browse Source 
PiperOrigin-RevId: 562783418
This commit is contained in:
Michael Hudgins 2023-09-05 08:15:21 -07:00 committed by jax authors
parent daa8ec2c4b
commit 1ac7e75ee6
1 changed files with 3 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
third_party/xla/workspace.bzl vendored
View File

@ -12,7 +12,7 @@
# See the License for the specific language governing permissions and
# limitations under the License.
load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
load("//third_party:repo.bzl", "tf_http_archive", "tf_mirror_urls")
# To update XLA to a new revision,
# a) update XLA_COMMIT to the new git commit hash
@ -24,13 +24,11 @@ XLA_COMMIT = "09ee9c8ff80c66fffa1774cc89b20060698d3d52"
XLA_SHA256 = "bc7ff5691e881dd80fc9794685e731e0e53ff459b0f7a8e380f4006f42da0d25"
def repo():
http_archive(
tf_http_archive(
name = "xla",
sha256 = XLA_SHA256,
strip_prefix = "xla-{commit}".format(commit = XLA_COMMIT),
urls = [
"https://github.com/openxla/xla/archive/{commit}.tar.gz".format(commit = XLA_COMMIT),
],
urls = tf_mirror_urls("https://github.com/openxla/xla/archive/{commit}.tar.gz".format(commit = XLA_COMMIT)),
)
# For development, one often wants to make changes to the TF repository as well