58 lines
1.1 KiB
Plaintext
58 lines
1.1 KiB
Plaintext
[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"
|
|
|