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

Merge pull request #15945 from skye:version

Browse Source 
PiperOrigin-RevId: 530722158
This commit is contained in:
jax authors 2023-05-09 14:59:20 -07:00
commit 1b9180167b
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 = "xla",
sha256 = "0e32c717ec6091ac9d2a620ca895a90f1b996657a36fca952ade4ccda54e5eea",
strip_prefix = "xla-48b5796215744b2de022250720731c782039c39d",
sha256 = "3d9ca2c8cd81b820b63052409867a81102c42bb1382b971f6402bad21288df52",
strip_prefix = "xla-dad64948516e3672b3e2518945831a70b5e90b81",
urls = [
"https://github.com/openxla/xla/archive/48b5796215744b2de022250720731c782039c39d.tar.gz",
"https://github.com/openxla/xla/archive/dad64948516e3672b3e2518945831a70b5e90b81.tar.gz",
],
)

4
setup.py
View File

@ -19,13 +19,13 @@ import sys
from setuptools import setup, find_packages
_current_jaxlib_version = '0.4.7'
_current_jaxlib_version = '0.4.9'
# The following should be updated with each new jaxlib release.
_latest_jaxlib_version_on_pypi = '0.4.7'
_available_cuda11_cudnn_versions = ['82', '86']
_default_cuda11_cudnn_version = '86'
_default_cuda12_cudnn_version = '88'
_libtpu_version = '0.1.dev20230327'
_libtpu_version = '0.1.dev20230509'
_dct = {}
with open('jax/version.py', encoding='utf-8') as f: