[package] name = "rtx-flash-metal-attention" version = "0.1.0" edition.workspace = true rust-version = "1.92" authors.workspace = true license.workspace = true repository.workspace = true description = "Flash Attention implementation for Apple Silicon via Metal" [dependencies] # Core RustyTorch dependencies rtx-tensor = { workspace = true, default-features = false, features = ["metal"] } rtx-runtime = { workspace = true, features = ["metal"] } # Core utilities thiserror = { workspace = true } tracing = { workspace = true } # Performance parking_lot = { workspace = true } # Metal bindings (macOS only) [target.'cfg(target_os = "macos")'.dependencies] objc2 = { version = "0.6", features = ["std"] } objc2-metal = { version = "0.3" } objc2-foundation = { version = "0.3", features = ["NSError", "NSString", "NSData", "NSDictionary", "NSObject"] } block2 = { version = "0.6" } [dev-dependencies] criterion = { workspace = true } tempfile = { workspace = true } [features] default = [] f16 = [] # Use float16 for Q/K/V (default is float32) [lints] workspace = true