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

Merge pull request #16869 from skye:version

Browse Source 
PiperOrigin-RevId: 551628701
This commit is contained in:
jax authors 2023-07-27 13:47:46 -07:00
commit c75e85da16
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 = "082fd460b11e9a672d4b1cab99807128958094dc56c8fff9ea40ad19fe1a313f",
strip_prefix = "xla-ed7886d05343d2a787829a380b723a675499067c",
sha256 = "60558f5c2a2ff46f46194febdc808180b6536a9d42b2835cd738afff462a496b",
strip_prefix = "xla-81b357b48e9b1c9dc3381a6ab4a2a64b06c34fc7",
urls = [
"https://github.com/openxla/xla/archive/ed7886d05343d2a787829a380b723a675499067c.tar.gz",
"https://github.com/openxla/xla/archive/81b357b48e9b1c9dc3381a6ab4a2a64b06c34fc7.tar.gz",
],
)

4
setup.py
View File

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