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 setup.py in preparation for 0.3.15 jax/jaxlib release

Browse Source
This commit is contained in:
Skye Wanderman-Milne 2022-07-14 10:12:59 -07:00
parent 86ab8a84ee
commit 9149c38e1e
2 changed files with 5 additions and 5 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 = "634e5ee7fba57ba8e1e6af18c8521d9849d631248979f0fa4824e772ecabdb79",
strip_prefix = "tensorflow-6e406f0bd8d08be3fa43daa306e8565cdb29a546",
sha256 = "e3243973e04c182690e4a3b269b7daf1b94ee2e903afd06646d2486dcb79feae",
strip_prefix = "tensorflow-9ad5167145151eeb41eea01beb8d59441f47595d",
urls = [
"https://github.com/tensorflow/tensorflow/archive/6e406f0bd8d08be3fa43daa306e8565cdb29a546.tar.gz",
"https://github.com/tensorflow/tensorflow/archive/9ad5167145151eeb41eea01beb8d59441f47595d.tar.gz",
],
)

4
setup.py
View File

@ -14,14 +14,14 @@
from setuptools import setup, find_packages
_current_jaxlib_version = '0.3.14'
_current_jaxlib_version = '0.3.15'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.3.14'
_available_cuda_versions = ['11']
_default_cuda_version = '11'
_available_cudnn_versions = ['82', '805']
_default_cudnn_version = '82'
_libtpu_version = '0.1.dev20220627'
_libtpu_version = '0.1.dev20220714'
_dct = {}
with open('jax/version.py') as f: