[package] name = "rtx-flash-attention" version.workspace = true edition.workspace = true 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 } rtx-quantum = { workspace = true, optional = true } rtx-neuromorphic = { workspace = true, optional = true } rtx-edge = { workspace = true, optional = true } # GPU acceleration cudarc = { workspace = true, optional = true } # rustg = { workspace = true } # Temporarily commented out # 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 } [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" [features] default = ["cuda"] cuda = ["cudarc"] quantum = ["rtx-quantum"] neuromorphic = ["rtx-neuromorphic"] edge = ["rtx-edge"] experimental = []