Deleted torch.backends.cudnn.benchmark line, defaults are fine

This commit is contained in:
Christopher Anderson 2025-08-26 08:43:31 +10:00
parent bdb5658268
commit 110cb0a9d9

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)