mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 17:24:32 +02:00
ROCm always use numpy on cholesky
This commit is contained in:
@@ -48,9 +48,11 @@ def has_mps() -> bool:
|
||||
def has_xpu() -> bool:
|
||||
return bool(hasattr(torch, 'xpu') and torch.xpu.is_available())
|
||||
|
||||
|
||||
def has_rocm() -> bool:
|
||||
return bool(torch.version.hip is not None and torch.cuda.is_available())
|
||||
|
||||
|
||||
def has_zluda() -> bool:
|
||||
if not cuda_ok:
|
||||
return False
|
||||
|
||||
Reference in New Issue
Block a user