[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 = "2"
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]
workspace = true