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

Turn off cuda malloc by default when --fast autotune is turned on. (#10393)

Browse Source
This commit is contained in:
comfyanonymous
2025-10-18 19:35:46 -07:00
committed by GitHub
parent 9da397ea2f
commit 5b80addafd
3 changed files with 7 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
comfy/ops.py
View File

@@ -67,9 +67,6 @@ except:
cast_to = comfy.model_management.cast_to #TODO: remove once no more references
if torch.cuda.is_available() and torch.backends.cudnn.is_available() and PerformanceFeature.AutoTune in args.fast:
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)