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

Replace _ with - because wheel.py normalizes it to .

Browse Source 
PiperOrigin-RevId: 404049619
This commit is contained in:
Yash Katariya 2021-10-18 13:47:08 -07:00 committed by jax authors
parent 0f0bfcaef5
commit 93fe3ab492
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/setup.py
View File

@ -23,7 +23,7 @@ with open('jaxlib/version.py') as f:
cuda_version = os.environ.get("JAX_CUDA_VERSION")
cudnn_version = os.environ.get("JAX_CUDNN_VERSION")
if cuda_version and cudnn_version:
__version__ += f"+cuda{cuda_version.replace('.', '')}_cudnn{cudnn_version.replace('.', '')}"
__version__ += f"+cuda{cuda_version.replace('.', '')}-cudnn{cudnn_version.replace('.', '')}"
setup(
name='jaxlib',