mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Update WORKSPACE and libtpu version for jaxlib 0.3.15, take 3

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2022-07-22 11:41:39 -07:00
parent e121e811ab
commit 26fbeb6e2a
2 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
WORKSPACE
View File

@ -7,10 +7,10 @@ load("@bazel_tools//tools/build_defs/repo:http.bzl", "http_archive")
# and update the sha256 with the result.
http_archive(
name = "org_tensorflow",
sha256 = "4cda98e3ba9cb853f7a6b4b195c04155d35a31ed27b21e9fe5473bf4f6a9add8",
strip_prefix = "tensorflow-e04beef36a5bf65594f310fc798d98545a393aee",
sha256 = "6b2b714e00db7b82fd79aaa1deacd321c7f0aeab8b17cf8c32062a80070842a8",
strip_prefix = "tensorflow-579c67bd60220fe66a29318191f0850ff0aa825f",
urls = [
"https://github.com/tensorflow/tensorflow/archive/e04beef36a5bf65594f310fc798d98545a393aee.tar.gz",
"https://github.com/tensorflow/tensorflow/archive/579c67bd60220fe66a29318191f0850ff0aa825f.tar.gz",
],
)

2
setup.py
View File

@ -21,7 +21,7 @@ _available_cuda_versions = ['11']
_default_cuda_version = '11'
_available_cudnn_versions = ['82', '805']
_default_cudnn_version = '82'
_libtpu_version = '0.1.dev20220719'
_libtpu_version = '0.1.dev20220722'
_dct = {}
with open('jax/version.py') as f: