59 lines
1.6 KiB
TOML
59 lines
1.6 KiB
TOML
[package]
|
|
name = "rtx-neural-operator-demo"
|
|
version = "1.0.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
description = "Neural operator demo for RustyTorch++ - interactive PDE solving with FNO"
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
|
|
[dependencies]
|
|
rtx-neural-operator-shared = { workspace = true }
|
|
rtx-neural-operator = { workspace = true }
|
|
rtx-tensor = { workspace = true }
|
|
rtx-backend = { workspace = true }
|
|
rtx-backend-cpu = { workspace = true }
|
|
rtx-nn = { workspace = true }
|
|
rtx-autograd = { workspace = true }
|
|
rtx-losses = { workspace = true }
|
|
rtx-hub = { workspace = true }
|
|
|
|
# GPU backends (optional - enable via features)
|
|
# Note: CUDA requires nvcc to be installed, so it uses a path reference
|
|
rtx-backend-cuda = { path = "../../crates/core/rtx-backend-cuda", optional = true }
|
|
rtx-backend-metal = { workspace = true, optional = true }
|
|
rtx-backend-rocm = { workspace = true, optional = true }
|
|
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
thiserror = "1.0"
|
|
safetensors = "0.5"
|
|
dirs = "5"
|
|
tracing = "0.1"
|
|
tokio = { version = "1.0", features = ["sync", "rt", "rt-multi-thread"] }
|
|
rand = "0.9"
|
|
rand_distr = "0.5"
|
|
|
|
[features]
|
|
default = []
|
|
# GPU acceleration features
|
|
cuda = ["dep:rtx-backend-cuda"]
|
|
metal = ["dep:rtx-backend-metal"]
|
|
rocm = ["dep:rtx-backend-rocm"]
|
|
# Enable all available GPU backends
|
|
gpu = ["cuda", "metal", "rocm"]
|
|
|
|
[dev-dependencies]
|
|
proptest = "1.4"
|
|
approx = "0.5"
|
|
tempfile = "3.10"
|
|
|
|
[lints.rust]
|
|
unsafe_code = "forbid"
|
|
missing_docs = "warn"
|
|
|
|
[lints.clippy]
|
|
all = "warn"
|
|
pedantic = "warn"
|