53 lines
1.2 KiB
TOML
53 lines
1.2 KiB
TOML
[package]
|
|
name = "rtx-backend-rocm"
|
|
version = "1.0.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch++ Team"]
|
|
license = "Apache-2.0"
|
|
description = "ROCm (AMD GPU) backend implementation for RustyTorch++ using HIP runtime"
|
|
|
|
[dependencies]
|
|
# Backend trait definitions
|
|
rtx-backend = { workspace = true }
|
|
|
|
# Core utilities
|
|
anyhow = { workspace = true }
|
|
thiserror = "1.0"
|
|
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"] }
|
|
|
|
# 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 ROCm-specific, only use on AMD GPU systems
|
|
default = []
|
|
# Enable full ROCm support (all features)
|
|
rocm = ["hip-runtime", "rocblas", "miopen"]
|
|
# Enable HIP runtime integration
|
|
hip-runtime = []
|
|
# Enable rocBLAS for optimized GEMM/BLAS
|
|
rocblas = ["hip-runtime"]
|
|
# Enable MIOpen for CNN operations
|
|
miopen = ["hip-runtime"]
|
|
# Enable profiling
|
|
profiling = []
|
|
|
|
[lib]
|
|
name = "rtx_backend_rocm"
|
|
path = "src/lib.rs"
|
|
|
|
[lints]
|
|
workspace = true
|