namespace at { | |
constexpr int32_t num_sdp_backends = 5; | |
enum class SDPBackend { | |
error = -1, | |
math = 0, | |
flash_attention = 1, | |
efficient_attention = 2, | |
cudnn_attention = 3, | |
overrideable = 4 | |
}; | |
} // namespace at | |
namespace at { | |
constexpr int32_t num_sdp_backends = 5; | |
enum class SDPBackend { | |
error = -1, | |
math = 0, | |
flash_attention = 1, | |
efficient_attention = 2, | |
cudnn_attention = 3, | |
overrideable = 4 | |
}; | |
} // namespace at | |