56 lines
1.1 KiB
TOML
56 lines
1.1 KiB
TOML
[package]
|
|
name = "rtx-onnx-codegen"
|
|
version = "1.0.0"
|
|
edition.workspace = true
|
|
authors = ["RustyTorch Team"]
|
|
license = "MIT OR Apache-2.0"
|
|
description = "Compile-time ONNX model import and Rust code generation for RustyTorch"
|
|
repository = "https://github.com/rustytorch/rustytorch"
|
|
keywords = ["onnx", "deep-learning", "code-generation", "machine-learning"]
|
|
categories = ["science", "development-tools::build-utils"]
|
|
|
|
[features]
|
|
default = ["cli"]
|
|
cli = ["clap"]
|
|
|
|
[dependencies]
|
|
# ONNX parsing
|
|
prost = "0.13"
|
|
|
|
# Code generation
|
|
proc-macro2 = "1.0"
|
|
quote = "1.0"
|
|
syn = { version = "2.0", features = ["full"] }
|
|
|
|
# Formatting
|
|
prettyplease = "0.2"
|
|
|
|
# CLI (optional)
|
|
clap = { version = "4.5", features = ["derive"], optional = true }
|
|
|
|
# Utilities
|
|
thiserror = "2.0"
|
|
log = "0.4"
|
|
env_logger = "0.11"
|
|
|
|
# Serialization
|
|
serde = { version = "1.0", features = ["derive"] }
|
|
serde_json = "1.0"
|
|
|
|
# File handling
|
|
memmap2 = "0.9"
|
|
|
|
[build-dependencies]
|
|
prost-build = "0.13"
|
|
|
|
[[bin]]
|
|
name = "rtx-onnx-import"
|
|
path = "src/bin/rtx-onnx-import.rs"
|
|
required-features = ["cli"]
|
|
|
|
[dev-dependencies]
|
|
tempfile = "3.10"
|
|
|
|
[lints]
|
|
workspace = true
|