40 lines
1.1 KiB
TOML
40 lines
1.1 KiB
TOML
[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
|