| 12345678910111213141516 |
- #pragma once
- #include <cstdint>
- 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
|