[package]
name = "rtx-distributed"
version.workspace = true
edition.workspace = true
authors.workspace = true
license.workspace = true
repository.workspace = true
description = "Distributed training support for RustyTorch++"

[dependencies]
# Workspace dependencies
anyhow.workspace = true
thiserror.workspace = true
tracing.workspace = true
tokio.workspace = true
serde.workspace = true
bincode.workspace = true
serde_json.workspace = true

# Local crates
rtx-runtime = { path = "../rtx-runtime" }
rtx-tensor = { path = "../rtx-tensor" }

# NCCL binding
nccl = "1.0"

# Network and process management
mpi = { version = "0.6", optional = true }
crossbeam.workspace = true
parking_lot.workspace = true
uuid.workspace = true

# Configuration
config.workspace = true
clap = { version = "4.0", features = ["derive"] }

# Async trait support
async-trait.workspace = true

# Additional dependencies for multi-node support
rand = "0.8"
futures = "0.3"

[dev-dependencies]
proptest.workspace = true
criterion.workspace = true
tempfile = "3.8"

[features]
default = ["nccl"]
nccl = []
mpi = ["dep:mpi"]

[lib]
name = "rtx_distributed"
path = "src/lib.rs"