mirror of
https://github.com/vladmandic/automatic
synced 2026-09-19 01:04:32 +02:00
zluda fix old cards
This commit is contained in:
+5
-2
@@ -34,8 +34,11 @@ def initialize_zluda():
|
||||
do_hijack()
|
||||
|
||||
torch.backends.cudnn.enabled = zluda_installer.MIOpen_enabled
|
||||
if not zluda_installer.MIOpen_enabled:
|
||||
torch.backends.cuda.enable_cudnn_sdp(False)
|
||||
if hasattr(torch.backends.cuda, "enable_cudnn_sdp"):
|
||||
if not zluda_installer.MIOpen_enabled:
|
||||
torch.backends.cuda.enable_cudnn_sdp(False)
|
||||
torch.backends.cuda.enable_cudnn_sdp = do_nothing
|
||||
else:
|
||||
torch.backends.cuda.enable_cudnn_sdp = do_nothing
|
||||
torch.backends.cuda.enable_flash_sdp(False)
|
||||
torch.backends.cuda.enable_flash_sdp = torch.backends.cuda.enable_cudnn_sdp
|
||||
|
||||
Reference in New Issue
Block a user