Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 01:04:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Merge pull request #5039 from resonantsky/dev

Browse Source 
Added ROCm Multi-Arch installation support
This commit is contained in:
Vladimir Mandic
2026-08-19 16:24:04 +02:00
committed by GitHub
parent 85235a8bcd 50494ee6d4
commit a393373b3a
3 changed files with 23 additions and 26 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/rocm.py
+9 -22
View File
@@ -133,28 +133,15 @@ class Agent:
@property
def therock(self) -> str | None:
if (self.gfx_version & 0xFFF0) == 0x1200:
return "v2/gfx120X-all"
if (self.gfx_version & 0xFFF0) == 0x1100:
return "v2/gfx110X-all"
if self.gfx_version == 0x1150:
return "v2-staging/gfx1150"
if self.gfx_version == 0x1151:
return "v2/gfx1151"
if self.gfx_version == 0x1152:
return "v2-staging/gfx1152"
if self.gfx_version == 0x1153:
return "v2-staging/gfx1153"
if self.gfx_version in (0x1030, 0x1031, 0x1032, 0x1034,):
return "v2-staging/gfx103X-dgpu"
#if (self.gfx_version & 0xFFF0) == 0x1010:
# return "gfx101X-dgpu"
#if (self.gfx_version & 0xFFF0) == 0x900:
# return "gfx90X-dcgpu"
#if (self.gfx_version & 0xFFF0) == 0x940:
# return "gfx94X-dcgpu"
#if self.gfx_version == 0x950:
# return "gfx950-dcgpu"
if self.gfx_version is None:
return None
gfx = self.name if self.name.startswith("gfx") else f"gfx{self.gfx_version:04x}"
if self.gfx_version & 0xFFF0 in (0x1200, 0x1100):
return f"whl-multi-arch/amd-torch-device-{gfx}"
if self.gfx_version in (0x1150, 0x1151, 0x1152, 0x1153):
return f"whl-multi-arch/amd-torch-device-{gfx}"
if self.gfx_version in (0x1030, 0x1031, 0x1032, 0x1033, 0x1034, 0x1035, 0x1036):
return f"whl-multi-arch/amd-torch-device-{gfx}"
return None
def get_gfx_version(self) -> str | None: