50 lines
1.3 KiB
TOML
50 lines
1.3 KiB
TOML
[package]
|
|
name = "rtx-optim-core"
|
|
version = "1.0.0"
|
|
edition.workspace = true
|
|
rust-version = "1.92"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
description = "Unified model optimization framework - compression, quantization, pruning, and merging"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
keywords = ["machine-learning", "model-optimization", "compression", "quantization", "merging"]
|
|
categories = ["machine-learning", "algorithms"]
|
|
|
|
[features]
|
|
default = ["compress", "merging"]
|
|
compress = ["dep:rtx-compress"]
|
|
merging = ["dep:rtx-model-merging"]
|
|
all = ["compress", "merging"]
|
|
|
|
# GPU acceleration features
|
|
cuda = ["rtx-compress?/cuda"]
|
|
metal = ["rtx-model-merging?/metal"]
|
|
gpu = ["cuda", "metal"]
|
|
|
|
[dependencies]
|
|
# Model optimization implementations (optional based on features)
|
|
rtx-compress = { path = "../rtx-compress", optional = true }
|
|
rtx-model-merging = { path = "../rtx-model-merging", optional = true }
|
|
|
|
# Core dependencies
|
|
rtx-tensor = { path = "../../core/rtx-tensor" }
|
|
rtx-runtime = { path = "../../core/rtx-runtime" }
|
|
|
|
# Error handling
|
|
anyhow = { workspace = true }
|
|
thiserror = { workspace = true }
|
|
|
|
# Serialization
|
|
serde = { workspace = true }
|
|
serde_json = { workspace = true }
|
|
|
|
# Logging
|
|
tracing = { workspace = true }
|
|
|
|
[dev-dependencies]
|
|
tokio-test = { workspace = true }
|
|
criterion = { workspace = true }
|
|
|
|
[lints]
|
|
workspace = true
|