123 lines
3.6 KiB
TOML
123 lines
3.6 KiB
TOML
[package]
|
|
name = "rtx-bench"
|
|
version = "1.0.0"
|
|
edition.workspace = true
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
description = "Comprehensive benchmarking suite for RustyTorch++ platform"
|
|
keywords = ["benchmarking", "machine-learning", "performance", "tensor"]
|
|
categories = ["development-tools::testing", "science"]
|
|
|
|
[dependencies]
|
|
# RTX Core Infrastructure
|
|
rtx-tensor = { workspace = true }
|
|
rtx-runtime = { workspace = true }
|
|
rtx-autograd = { workspace = true }
|
|
rtx-memory = { workspace = true }
|
|
rtx-kernel = { workspace = true }
|
|
rtx-graph = { workspace = true }
|
|
rtx-validation = { workspace = true }
|
|
|
|
# RTX Training
|
|
rtx-transformers = { workspace = true, optional = true }
|
|
# rtx-distributed = { workspace = true, optional = true } # Temporarily disabled - missing hpc-channels
|
|
rtx-compress = { workspace = true }
|
|
rtx-flash-attention = { workspace = true }
|
|
|
|
# RTX Models
|
|
rtx-vision = { workspace = true, optional = true }
|
|
rtx-multimodal = { workspace = true }
|
|
# rtx-nlg = { workspace = true, optional = true } # Disabled - candle incompatible with Rust 2024
|
|
|
|
# RTX Production
|
|
rtx-inference = { workspace = true }
|
|
rtx-serving-api = { workspace = true }
|
|
rtx-monitoring = { workspace = true }
|
|
|
|
# Essential dependencies
|
|
anyhow = { workspace = true }
|
|
thiserror = { workspace = true }
|
|
tracing = { workspace = true }
|
|
tracing-subscriber = { workspace = true }
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
tokio = { workspace = true }
|
|
|
|
# Benchmarking and statistics
|
|
criterion = { workspace = true, features = ["html_reports", "csv_output"] }
|
|
statistical = "1.0"
|
|
plotters = "0.3"
|
|
hdrhistogram = "7.5"
|
|
|
|
# Math and collections
|
|
nalgebra = { workspace = true }
|
|
indexmap = { workspace = true }
|
|
dashmap = { workspace = true }
|
|
rand = { workspace = true }
|
|
rand_distr = { workspace = true }
|
|
statrs = { workspace = true }
|
|
|
|
# GPU acceleration
|
|
cudarc = { workspace = true, optional = true }
|
|
# rtx-backend-cuda = { workspace = true, optional = true } # Excluded - requires CUDA/nvcc
|
|
rtx-backend-metal = { workspace = true, optional = true }
|
|
rtx-metal = { workspace = true, optional = true }
|
|
|
|
# System metrics
|
|
sysinfo = "0.29"
|
|
psutil = "3.2"
|
|
num_cpus = "1.16"
|
|
|
|
# Time handling
|
|
chrono = { workspace = true }
|
|
|
|
# Async runtime
|
|
futures = { workspace = true }
|
|
async-trait = { workspace = true }
|
|
|
|
# Concurrency
|
|
parking_lot = { workspace = true }
|
|
crossbeam = { workspace = true }
|
|
rayon = { workspace = true }
|
|
|
|
# File I/O and serialization
|
|
uuid = { workspace = true }
|
|
bincode = { workspace = true }
|
|
toml = { workspace = true }
|
|
|
|
# CLI and reporting
|
|
clap = { workspace = true }
|
|
comfy-table = { workspace = true }
|
|
colorful = { workspace = true }
|
|
indicatif = { workspace = true }
|
|
|
|
[dev-dependencies]
|
|
tokio-test = { workspace = true }
|
|
proptest = { workspace = true }
|
|
tempfile = { workspace = true }
|
|
|
|
[features]
|
|
default = ["vision"] # Removed "distributed" - dependency disabled
|
|
# cuda = ["cudarc", "rtx-backend-cuda", "rtx-flash-attention/cuda", "rtx-tensor/cuda", "rtx-runtime/cuda"] # Disabled - requires CUDA/nvcc
|
|
metal = ["rtx-backend-metal", "rtx-metal"] # Metal GPU acceleration for Apple Silicon
|
|
# distributed = ["rtx-distributed"] # Temporarily disabled - missing hpc-channels
|
|
vision = ["rtx-vision"]
|
|
# nlp = ["rtx-nlg", "rtx-transformers"] # Disabled - rtx-nlg has candle dependency incompatible with Rust 2024
|
|
competitive = [] # Enable competitive analysis benchmarks
|
|
|
|
[[bin]]
|
|
name = "rtx-bench"
|
|
path = "src/bin/main.rs"
|
|
|
|
[[bench]]
|
|
name = "tensor_ops"
|
|
harness = false
|
|
|
|
[[bench]]
|
|
name = "model_architectures"
|
|
harness = false
|
|
[lints]
|
|
workspace = true
|