SDPBackend.h 253 B

12345678910111213141516
  1. #pragma once
  2. #include <cstdint>
  3. namespace at {
  4. constexpr int32_t num_sdp_backends = 5;
  5. enum class SDPBackend {
  6. error = -1,
  7. math = 0,
  8. flash_attention = 1,
  9. efficient_attention = 2,
  10. cudnn_attention = 3,
  11. overrideable = 4
  12. };
  13. } // namespace at