andreas/ComfyUI
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Log message for cudnn disable on AMD. (#10418)

Browse Source
This commit is contained in:
comfyanonymous
2025-10-20 12:43:24 -07:00
committed by GitHub
parent a4787ac83b
commit 2c2aa409b0
1 changed files with 1 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/model_management.py
View File

@@ -333,6 +333,7 @@ SUPPORT_FP8_OPS = args.supports_fp8_compute
try: try:
if is_amd(): if is_amd():
torch.backends.cudnn.enabled = False # Seems to improve things a lot on AMD torch.backends.cudnn.enabled = False # Seems to improve things a lot on AMD
logging.info("Set: torch.backends.cudnn.enabled = False for better AMD performance.")
try: try:
rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2])) rocm_version = tuple(map(int, str(torch.version.hip).split(".")[:2]))
except: except: