torch.nn.attention.cudnn_sdp_python_enabled

torch.nn.attention.cudnn_sdp_python_enabled()

Warning

This API is beta and subject to change.

Returns whether cuDNN scaled dot product attention is currently served by the cuDNN frontend Python API rather than the built-in C++ implementation.

Return type:

bool