Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

cuda exception handler

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-09 11:15:31 -04:00
parent 8a541d21fa
commit 3341a34549
1 changed files with 5 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/zluda.py
+5 -2
View File
@@ -18,8 +18,11 @@ def _join_rocm_home(*paths) -> str:
def is_zluda(device: DeviceLikeType):
device = torch.device(device)
return torch.cuda.get_device_name(device).endswith("[ZLUDA]")
try:
device = torch.device(device)
return torch.cuda.get_device_name(device).endswith("[ZLUDA]")
except Exception:
return False
def test(device: DeviceLikeType) -> Union[Exception, None]: