team-10/env/Lib/site-packages/torch/include/ATen/SDPBackend.h
2025-08-02 07:34:44 +02:00

16 lines
253 B
C++

#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