Enables or disables flash sdp.
torch.backends.cuda.math_sdp_enabled()
Warning:
 This flag is experimental and subject to change.

Returns whether math sdp is enabled or not.
torch.backends.cuda.enable_math_sdp(enabled)
Warning:
 This flag is experimental and subject to change.

Enables or disables math sdp.
torch.backends.cuda.sdp_kernel(enable_flash=True, enable_math=True, enable_mem_efficient=True)
Warning:
 This flag is experimental and subject to change.

This context manager can be used to temporarily enable or disable
   flash/memory efficient sdp and math sdp. Upon exiting the context
   manager, the previous state of the flags will be restored.
torch.backends.cudnn
torch.backends.cudnn.version()
Returns the version of cuDNN
torch.backends.cudnn.is_available()
Returns a bool indicating if CUDNN is currently available.
torch.backends.cudnn.enabled
A "bool" that controls whether cuDNN is enabled.