[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