9c6594c
|
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
#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
};
}
|
