torch.nn.attention.current_flash_attention_impl

torch.nn.attention.current_flash_attention_impl()

Return the currently activated flash attention impl name, if any.

None indicates that no custom impl has been activated.

Return type:

str | None