From 646060140779db9737f76625fca92a16b694cfcc Mon Sep 17 00:00:00 2001 From: 0xDELUXA Date: Tue, 7 Jul 2026 23:12:04 +0300 Subject: [PATCH] rocm: detect hipBLASLt lazy library in per-arch subfolder and zlib form --- modules/rocm.py | 14 +++++++++----- 1 file changed, 9 insertions(+), 5 deletions(-) diff --git a/modules/rocm.py b/modules/rocm.py index a514ee9ef..349f9cdc5 100644 --- a/modules/rocm.py +++ b/modules/rocm.py @@ -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