mirror of
https://github.com/invoke-ai/InvokeAI.git
synced 2026-10-12 02:16:43 +00:00
Alexander Eichhornand0xDELUXA
7001bf3780
The rocm extra moves from PyTorch's 2.13.0+rocm7.2 (Linux only) to AMD's 2.13.0+rocm10.0.0 for Linux x86_64 and Windows. The lock carries the GPU kernel packages of every target AMD builds, so the launcher can leave out the ones a machine does not need. Linux loses gfx900/gfx906, which ROCm 10 dropped. bitsandbytes moves to 0.50.2, the first release with a Windows ROCm binary. A new setting, rocm_aotriton_experimental (auto/on/off), sets TORCH_ROCM_AOTRITON_ENABLE_EXPERIMENTAL before anything runs attention. auto turns AOTriton's experimental fused kernels on only for a ROCm 10 build on GPUs where they were measured correct (gfx1200 so far); under torch 2.12+rocm7.14 the same switch made every fused call fail on that GPU. Builds on the dependency and documentation changes of #9606. Co-authored-by: 0xDELUXA <[email protected]>
18 lines
492 B
JSON
18 lines
492 B
JSON
{
|
|
"python": "3.12",
|
|
"torchIndexUrl": {
|
|
"win32": {
|
|
"cpu": "https://download.pytorch.org/whl/cpu",
|
|
"cuda": "https://download.pytorch.org/whl/cu130",
|
|
"xpu": "https://download.pytorch.org/whl/xpu"
|
|
},
|
|
"linux": {
|
|
"cpu": "https://download.pytorch.org/whl/cpu",
|
|
"rocm": "https://stable.repo.amd.com/rocm/whl-next/",
|
|
"cuda": "https://download.pytorch.org/whl/cu130",
|
|
"xpu": "https://download.pytorch.org/whl/xpu"
|
|
},
|
|
"darwin": {}
|
|
}
|
|
}
|