| 123456789101112131415161718192021 |
- #if !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
- #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
- #else
- #error "This file should not be included when either TORCH_STABLE_ONLY or TORCH_TARGET_VERSION is defined."
- #endif // !defined(TORCH_STABLE_ONLY) && !defined(TORCH_TARGET_VERSION)
|
