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

Update TF commit for release

Browse Source 
PiperOrigin-RevId: 446555288
This commit is contained in:
Yash Katariya 2022-05-04 14:39:00 -07:00 committed by jax authors
parent 72971151a4
commit 38ce6d027b
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 = "bee3000a7892008b32c65a5d6ba5d0a1f490e9e749d6789c5100d47068b1bbd8",
strip_prefix = "tensorflow-46cb56d405f0800e1741f68bba098d8eea5d0025",
sha256 = "76d7cbb4ce7d6b506e1845bb1bf7fe0ed89888d19069363629992da5059dc700",
strip_prefix = "tensorflow-994bca455e4d15b843115d4e43b0989726b7d767",
urls = [
"https://github.com/tensorflow/tensorflow/archive/46cb56d405f0800e1741f68bba098d8eea5d0025.tar.gz",
"https://github.com/tensorflow/tensorflow/archive/994bca455e4d15b843115d4e43b0989726b7d767.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.dev20220503'
_libtpu_version = '0.1.dev20220504'
_dct = {}
with open('jax/version.py') as f: