feat: add ROCm 7.1.1 Windows wheel mapping

AMD uses a different version string for 7.1.1 wheels:
2.9.0+rocmsdk20251116 (date-tagged) instead of +rocm7.1.1.
Adds the 7.1.1 release folder to both install.ps1 and
install_python_stack.py so users with ROCm 7.1 get ROCm
torch instead of falling back to CPU.
This commit is contained in:
LeoBorcherding 2026-05-06 17:55:23 -05:00
commit 7e2943597d
2 changed files with 12 additions and 0 deletions

View file

@ -71,6 +71,12 @@ _ROCM_WINDOWS_RELEASES: dict[tuple[int, int], tuple[str, str, str, str]] = {
"0.24.1+rocm7.2.1",
"2.9.1+rocm7.2.1",
),
(7, 1): (
"rocm-rel-7.1.1",
"2.9.0+rocmsdk20251116",
"0.24.0+rocmsdk20251116",
"2.9.0+rocmsdk20251116",
),
}
# bitsandbytes continuous-release_main wheels with the ROCm 4-bit GEMV fix