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

rocm: detect hipBLASLt lazy library in per-arch subfolder and zlib form

Browse Source
This commit is contained in:
0xDELUXA
2026-07-07 23:12:04 +03:00
parent 913481e5bd
commit 6460601407
1 changed files with 9 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/rocm.py
+9 -5
View File
@@ -114,12 +114,16 @@ class Agent:
self.arch = MicroArchitecture.GCN
self.is_apu = (self.gfx_version & 0xFFF0 == 0x1150) or self.gfx_version in (0x801, 0x902, 0x90c, 0x1013, 0x1033, 0x1035, 0x1036, 0x1103,)
blaslt_filename = f"TensileLibrary_lazy_{self.name}.dat"
if blaslt_tensile_libpath is not None and os.path.exists(os.path.join(blaslt_tensile_libpath, blaslt_filename)):
self.blaslt_supported = True
elif blaslt_tensile_libpath is not None and os.path.exists(os.path.join(blaslt_tensile_libpath, self.name, blaslt_filename)):
self.blaslt_supported = True
else:
if blaslt_tensile_libpath is None:
self.blaslt_supported = False
else:
search_paths = (
os.path.join(blaslt_tensile_libpath, blaslt_filename),
os.path.join(blaslt_tensile_libpath, blaslt_filename + ".zlib"),
os.path.join(blaslt_tensile_libpath, self.name, blaslt_filename),
os.path.join(blaslt_tensile_libpath, self.name, blaslt_filename + ".zlib"),
)
self.blaslt_supported = any(os.path.exists(p) for p in search_paths)
def __str__(self) -> str:
return self.name