46 lines
1.2 KiB
TOML
46 lines
1.2 KiB
TOML
[package]
|
|
name = "pinn_mre_helmholtz"
|
|
version = "0.1.0"
|
|
edition = "2024"
|
|
rust-version = "1.92"
|
|
description = "Physics-Informed Neural Network (PINN) for 1D MRE Helmholtz equation"
|
|
authors = ["RustyTorch Team"]
|
|
|
|
# Standalone example - not part of workspace
|
|
[workspace]
|
|
|
|
[dependencies]
|
|
# RustyTorch++ crates
|
|
rtx-tensor = { path = "../../crates/core/rtx-tensor", features = ["cpu"] }
|
|
rtx-autograd = { path = "../../crates/core/rtx-autograd" }
|
|
rtx-nn = { path = "../../crates/core/rtx-nn" }
|
|
rtx-transformers = { path = "../../crates/training/rtx-transformers" }
|
|
rtx-losses = { path = "../../crates/core/rtx-losses" }
|
|
rtx-runtime = { path = "../../crates/core/rtx-runtime", optional = true }
|
|
|
|
# External dependencies
|
|
num-complex = "0.4"
|
|
anyhow = "1.0"
|
|
thiserror = "1.0"
|
|
fastrand = "2.0"
|
|
cudarc = { version = "0.18.1", optional = true }
|
|
|
|
[features]
|
|
default = ["openblas"]
|
|
cuda = ["rtx-tensor/cuda", "rtx-nn/cuda", "rtx-runtime", "cudarc"]
|
|
openblas = ["rtx-tensor/openblas"]
|
|
mkl = ["rtx-tensor/mkl"]
|
|
|
|
[dev-dependencies]
|
|
criterion = { version = "0.5", features = ["html_reports"] }
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
|
|
[[bench]]
|
|
name = "pinn_benchmark"
|
|
harness = false
|
|
|
|
[profile.release]
|
|
lto = true
|
|
opt-level = 3
|