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

rocm&zluda handle apu

Browse Source
This commit is contained in:
Seunghoon Lee
2024-09-26 13:34:11 +09:00
parent 9ba7f4b1a0
commit 96922d70a7
2 changed files with 11 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/rocm.py
+5 -3
View File
@@ -53,18 +53,20 @@ class MicroArchitecture(Enum):
class Agent:
name: str
arch: MicroArchitecture
is_apu: bool
if sys.platform != "win32":
blaslt_supported: bool
def __init__(self, name: str):
self.name = name
gfx_version = name[3:7]
if len(gfx_version) == 4:
gfx = name[3:7]
if len(gfx) == 4:
self.arch = MicroArchitecture.RDNA
elif gfx_version in ("908", "90a", "942",):
elif gfx in ("908", "90a", "942",):
self.arch = MicroArchitecture.CDNA
else:
self.arch = MicroArchitecture.GCN
self.is_apu = gfx in ("801", "902", "90c", "1013", "1033", "1035", "1036", "1103",)
if sys.platform != "win32":
self.blaslt_supported = os.path.exists(os.path.join(HIPBLASLT_TENSILE_LIBPATH, f"extop_{name}.co"))