[package] name = "rtx-flash-attention" version = "1.0.0" edition.workspace = true rust-version = "1.92" authors.workspace = true license.workspace = true repository.workspace = true description = "Revolutionary Flash Attention implementation with Quantum/Neuromorphic/Edge variants" [dependencies] # Core RustyTorch dependencies rtx-tensor = { workspace = true } rtx-runtime = { workspace = true } rtx-kernel = { workspace = true } # GPU acceleration cudarc = { workspace = true, optional = true } # Core utilities anyhow = { workspace = true } thiserror = { workspace = true } tracing = { workspace = true } serde = { workspace = true } # Performance and memory management parking_lot = { workspace = true } rayon = { workspace = true } dashmap = { workspace = true } # Math and numerics nalgebra = { workspace = true } # Async support async-trait = { workspace = true } tokio = { workspace = true } [target.'cfg(target_os = "macos")'.dependencies] objc2 = { version = "0.6", optional = true, features = ["std"] } objc2-metal = { version = "0.3", optional = true } objc2-foundation = { version = "0.3", optional = true } block2 = { version = "0.6", optional = true } [dev-dependencies] criterion = { workspace = true } proptest = { workspace = true } tokio-test = { workspace = true } tempfile = { workspace = true } [build-dependencies] bindgen = "0.69" cc = "1.0" [[bench]] name = "flash_attention_bench" harness = false [[example]] name = "flash_attention_demo" path = "examples/flash_attention_demo.rs" required-features = ["disabled_tests"] [features] # Note: No default GPU features - select cuda or metal explicitly based on your platform default = [] cuda = ["rtx-kernel/cuda", "rtx-tensor/cuda", "rtx-runtime/cuda", "dep:cudarc"] metal = ["rtx-kernel/metal", "rtx-tensor/metal", "dep:objc2", "dep:objc2-metal", "dep:objc2-foundation", "dep:block2"] variants = [] experimental = [] disabled_tests = [] [lints] workspace = true