Merge pull request #276 from sfinktah/sfink-cudnn-benchmark-env

Deleted torch.backends.cudnn.benchmark line, defaults are fine
This commit is contained in:
patientx 2025-08-26 19:34:35 +03:00 committed by GitHub
commit c1aef0126d
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194

View File

@ -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)