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 #4986 from 0xDELUXA/fix-hipblaslt-detect

Browse Source 
rocm: detect hipBLASLt lazy library in per-arch subfolder and zlib form
This commit is contained in:
Vladimir Mandic
2026-07-08 00:47:54 +02:00
committed by GitHub
parent 913481e5bd 6460601407
commit c3f9865fcc
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