Omar Sobh
6a03aeba61
Co-authored-by: Omar Sobh <[email protected]> Co-committed-by: Omar Sobh <[email protected]>
56 lines
1.3 KiB
TOML
56 lines
1.3 KiB
TOML
[package]
|
|
name = "rtx-backend-cuda"
|
|
version = "1.0.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch++ Team"]
|
|
license = "Apache-2.0"
|
|
description = "CUDA backend implementation for RustyTorch++ with hand-optimized kernels"
|
|
|
|
[dependencies]
|
|
# Backend trait definitions
|
|
rtx-backend = { workspace = true }
|
|
|
|
# Core utilities
|
|
anyhow = { workspace = true }
|
|
thiserror = "2"
|
|
tracing = { workspace = true }
|
|
parking_lot = { workspace = true }
|
|
once_cell = "1.19"
|
|
|
|
# Numeric types
|
|
half = { version = "2.3", features = ["serde"] }
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
|
|
# CUDA acceleration via cudarc (optional - only builds on CUDA-capable systems)
|
|
cudarc = { workspace = true, optional = true }
|
|
|
|
# Flash Attention integration (optional)
|
|
rtx-flash-attention = { workspace = true, features = ["cuda"], optional = true }
|
|
|
|
# Random number generation
|
|
rand = { workspace = true }
|
|
rand_distr = { workspace = true }
|
|
|
|
[dev-dependencies]
|
|
proptest = "1.4"
|
|
tokio-test = "0.4"
|
|
|
|
[features]
|
|
# No default - this crate is CUDA-specific, only use on CUDA-capable systems
|
|
default = []
|
|
cuda = ["dep:cudarc"]
|
|
# Enable rtx-flash-attention integration for optimized attention
|
|
flash-attention = ["dep:rtx-flash-attention"]
|
|
|
|
[lib]
|
|
name = "rtx_backend_cuda"
|
|
path = "src/lib.rs"
|
|
|
|
[lints]
|
|
workspace = true
|
|
|
|
[[test]]
|
|
name = "backend_parity_tests"
|
|
required-features = ["cuda"]
|