Skip to content

cuda_specs.py: fix version tag string for ROCm/CUDA library lookup#1889

Open
eliasmagn wants to merge 1 commit intobitsandbytes-foundation:mainfrom
eliasmagn:main
Open

cuda_specs.py: fix version tag string for ROCm/CUDA library lookup#1889
eliasmagn wants to merge 1 commit intobitsandbytes-foundation:mainfrom
eliasmagn:main

Conversation

@eliasmagn
Copy link

Replaced the incorrect major * 10 + minor logic (e.g. ROCm 7.12 → "82"; CUDA breaks when minor > 9) with simple string concatenation, producing the expected tags:

  • ROCm 7.12 → "712"
  • CUDA 12.4 → "124"

This makes version tag derivation consistent across CUDA and ROCm and fixes ROCm shared library name resolution (e.g. libbitsandbytes_rocm712.so). We should not use zero-padding (e.g. {minor:02d}), as that would change CUDA tags (e.g. 12.4 → "1204") .

Hope this helps.

…82"; CUDA breaks when minor > 9) with simple string concatenation, producing the expected tags:

- ROCm 7.12 → "712"
- CUDA 12.4 → "124"

This makes version tag derivation consistent across CUDA and ROCm and fixes ROCm shared library name resolution (e.g. libbitsandbytes_rocm712.so).
We should not use zero-padding (e.g. `{minor:02d}`), as that would change CUDA tags (e.g. 12.4 → "1204") .
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant