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:
parent
ec40e9f207
commit
7e2943597d
2 changed files with 12 additions and 0 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue