65 lines
1.6 KiB
Plaintext
65 lines
1.6 KiB
Plaintext
[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 = [] |