This commit is contained in:
JettHu 2026-07-17 23:28:16 -07:00 committed by GitHub
commit 49038fff4f
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 1 deletions

View File

@ -519,7 +519,7 @@ except:
pass
if ENABLE_PYTORCH_ATTENTION:
if ENABLE_PYTORCH_ATTENTION and cpu_state != CPUState.CPU:
torch.backends.cuda.enable_math_sdp(True)
torch.backends.cuda.enable_flash_sdp(True)
torch.backends.cuda.enable_mem_efficient_sdp(True)