wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Added env_var for cudnn.benchmark

Browse Source
This commit is contained in:
Christopher Anderson 2025-08-28 09:00:08 +10:00
parent eeab23fc0b
commit cf22cbd8d5
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
comfy/customzluda/zluda.py
View File

@ -438,6 +438,13 @@ try:
else:
print(" :: Disabled cuDNN")
if os.environ.get("TORCH_BACKENDS_CUDNN_BENCHMARK"):
torch.backends.cudnn.benchmark = os.environ.get("TORCH_BACKENDS_CUDNN_BENCHMARK", "1").strip().lower() not in {"0", "off", "false", "disable", "disabled", "no"}
if torch.backends.cudnn.benchmark:
print(" :: Enabled cuDNN Benchmark")
else:
print(" :: Disabled cuDNN Benchmark")
@triton.jit
def _zluda_kernel_test(x_ptr, y_ptr, n_elements, BLOCK_SIZE: tl.constexpr):
pid = tl.program_id(axis=0)