xinyun/ComfyUI
1
0
Fork 0
mirror of https://git.datalinker.icu/comfyanonymous/ComfyUI synced 2026-08-16 00:43:31 +08:00
Code Issues Packages Projects Releases Wiki Activity

Enable Convolution AutoTuning

Browse Source
This commit is contained in:
lspindler 2025-08-12 14:57:47 +02:00
parent 9126c0cfe4
commit f91a510287
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ops.py
View File

@ -26,6 +26,9 @@ import contextlib
cast_to = comfy.model_management.cast_to #TODO: remove once no more references
if torch.cuda.is_available() and torch.backends.cudnn.is_available():
torch.backends.cudnn.benchmark = True
def cast_to_input(weight, input, non_blocking=False, copy=True):
return comfy.model_management.cast_to(weight, input.dtype, input.device, non_blocking=non_blocking, copy=copy)