Update zluda.py

This commit is contained in:
patientx 2025-02-28 16:41:56 +03:00 committed by GitHub
parent af43425ab5
commit c235a51d82
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -46,7 +46,7 @@ if torch.cuda.get_device_name().endswith("[ZLUDA]"):
print(" :: ZLUDA detected, disabling non-supported functions. ")
torch.backends.cudnn.enabled = False
print(" :: CuDNN, flash_sdp, mem_efficient_sdp disabled). ")
torch.backends.cuda.enable_flash_sdp(False)
torch.backends.cuda.enable_flash_sdp(False) # enable if using 6.2 with latest nightly zluda
torch.backends.cuda.enable_math_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(False)
print("***--------------------------------------------------------***")