diff --git a/comfy/customzluda/zluda.py b/comfy/customzluda/zluda.py index d9b6f48f2..57050627f 100644 --- a/comfy/customzluda/zluda.py +++ b/comfy/customzluda/zluda.py @@ -437,8 +437,7 @@ try: print(" :: Enabled cuDNN") else: print(" :: Disabled cuDNN") - torch.backends.cudnn.benchmark = False - + @triton.jit def _zluda_kernel_test(x_ptr, y_ptr, n_elements, BLOCK_SIZE: tl.constexpr): pid = tl.program_id(axis=0)