mirror of
https://github.com/pykeio/ort
synced 2026-04-25 16:34:55 +02:00
63890c58e2
The code still targets 1.81 but I'm tired of the CI failing and I refuse to commit `Cargo.lock` to pin `base64ct`. Thankfully, Cargo's resolver has started taking `rust-version` into account since 1.84, so this won't be a problem in the future.
107 lines
3.3 KiB
TOML
107 lines
3.3 KiB
TOML
[workspace]
|
|
members = [ 'ort-sys' ]
|
|
default-members = [ '.' ]
|
|
exclude = [
|
|
'backends/candle',
|
|
'backends/tract',
|
|
'examples/async-gpt2-api',
|
|
'examples/cudarc',
|
|
'examples/custom-ops',
|
|
'examples/gpt2',
|
|
'examples/model-info',
|
|
'examples/modnet',
|
|
'examples/phi-3-vision',
|
|
'examples/sentence-transformers',
|
|
'examples/training',
|
|
'examples/wasm-emscripten',
|
|
'examples/yolov8',
|
|
'tests/leak-check'
|
|
]
|
|
|
|
[package]
|
|
name = "ort"
|
|
description = "A safe Rust wrapper for ONNX Runtime 1.22 - Optimize and accelerate machine learning inference & training"
|
|
version = "2.0.0-rc.10"
|
|
edition = "2021"
|
|
rust-version = "1.85"
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/pykeio/ort"
|
|
homepage = "https://ort.pyke.io/"
|
|
readme = "README.md"
|
|
keywords = [ "machine-learning", "ai", "ml", "onnxruntime" ]
|
|
categories = [ "algorithms", "mathematics", "science" ]
|
|
authors = [
|
|
"pyke.io <contact@pyke.io>",
|
|
"Nicolas Bigaouette <nbigaouette@gmail.com>"
|
|
]
|
|
include = [ "/src/", "/LICENSE-APACHE", "/LICENSE-MIT", "/README.md" ]
|
|
|
|
[profile.release]
|
|
opt-level = 3
|
|
lto = true
|
|
strip = true
|
|
codegen-units = 1
|
|
|
|
[package.metadata.docs.rs]
|
|
features = [ "std", "ndarray", "half", "num-complex", "training", "fetch-models", "load-dynamic", "copy-dylibs" ]
|
|
targets = ["x86_64-unknown-linux-gnu"]
|
|
rustdoc-args = [ "--cfg", "docsrs" ]
|
|
|
|
[features]
|
|
default = [ "std", "ndarray", "tracing", "download-binaries", "copy-dylibs" ]
|
|
|
|
std = [ "ort-sys/std", "ndarray/std", "tracing?/std" ]
|
|
training = [ "ort-sys/training" ]
|
|
|
|
ndarray = [ "dep:ndarray" ]
|
|
half = [ "dep:half" ]
|
|
num-complex = [ "dep:num-complex" ]
|
|
tracing = [ "dep:tracing" ]
|
|
|
|
fetch-models = [ "std", "dep:ureq", "dep:sha2" ]
|
|
download-binaries = [ "ort-sys/download-binaries" ]
|
|
load-dynamic = [ "std", "libloading", "ort-sys/load-dynamic" ]
|
|
copy-dylibs = [ "ort-sys/copy-dylibs" ]
|
|
|
|
alternative-backend = [ "ort-sys/disable-linking" ]
|
|
|
|
cuda = [ "ort-sys/cuda" ]
|
|
tensorrt = [ "ort-sys/tensorrt" ]
|
|
openvino = [ "ort-sys/openvino" ]
|
|
onednn = [ "ort-sys/onednn" ]
|
|
directml = [ "ort-sys/directml" ]
|
|
nnapi = [ "ort-sys/nnapi" ]
|
|
coreml = [ "ort-sys/coreml" ]
|
|
xnnpack = [ "ort-sys/xnnpack" ]
|
|
rocm = [ "ort-sys/rocm" ]
|
|
acl = [ "ort-sys/acl" ]
|
|
armnn = [ "ort-sys/armnn" ]
|
|
tvm = [ "ort-sys/tvm" ]
|
|
migraphx = [ "ort-sys/migraphx" ]
|
|
rknpu = [ "ort-sys/rknpu" ]
|
|
vitis = [ "ort-sys/vitis" ]
|
|
cann = [ "ort-sys/cann" ]
|
|
qnn = [ "ort-sys/qnn" ]
|
|
webgpu = [ "ort-sys/webgpu" ]
|
|
azure = [ "ort-sys/azure" ]
|
|
nv = [ "ort-sys/nv" ]
|
|
|
|
[dependencies]
|
|
ort-sys = { version = "=2.0.0-rc.10", path = "ort-sys", default-features = false }
|
|
smallvec = { version = "=2.0.0-alpha.10", default-features = false }
|
|
|
|
ndarray = { version = "0.16", default-features = false, optional = true }
|
|
libloading = { version = "0.8", optional = true }
|
|
ureq = { version = "3", optional = true, default-features = false, features = [ "rustls" ] }
|
|
sha2 = { version = "0.10", optional = true }
|
|
tracing = { version = "0.1", optional = true, default-features = false }
|
|
half = { version = "2.1", default-features = false, optional = true }
|
|
num-complex = { version = "0.4", default-features = false, optional = true }
|
|
|
|
[dev-dependencies]
|
|
anyhow = "1.0"
|
|
ureq = { version = "3", default-features = false, features = [ "native-tls" ] }
|
|
image = "0.25"
|
|
tracing-subscriber = { version = "0.3", default-features = false, features = [ "env-filter", "fmt" ] }
|
|
tokio = { version = "1.36", features = [ "test-util" ] }
|