proper zluda detection

2026-09-20 01:31:13 +02:00 · 2025-03-20 23:03:23 +09:00
parent d5cfd61e50
commit 0c890b50e0
1 changed files with 2 additions and 1 deletions
@@ -52,7 +52,8 @@ def has_zluda() -> bool:
        return False
    try:
        dev = torch.device("cuda")
-        return torch.cuda.get_device_name(dev).endswith("[ZLUDA]")
+        cc = torch.cuda.get_device_capability(dev)
+        return cc == (8, 8)
    except Exception:
        return False