fix

2026-09-18 00:34:33 +02:00 · 2024-07-18 01:00:50 +09:00
parent 14569aca24
commit 7eb7dd587d
1 changed files with 1 additions and 1 deletions
@@ -565,7 +565,7 @@ def install_rocm_zluda(torch_command):
                    continue
                supported_archs.append(file[6:-3])
            for gpu in hip_visible_devices:
-                if gpu not in supported_archs:
+                if gpu[1] not in supported_archs:
                    hipblaslt_available = False
                    break
            log.debug(f'hipBLASLt supported_archs={supported_archs}, available={hipblaslt_available}')