mirror of
https://github.com/pykeio/ort
synced 2026-04-25 16:34:55 +02:00
94 lines
2.3 KiB
TOML
94 lines
2.3 KiB
TOML
[workspace]
|
|
members = [
|
|
'ort-sys',
|
|
'examples/gpt2',
|
|
'examples/yolov8'
|
|
]
|
|
default-members = [
|
|
'.',
|
|
'examples/gpt2',
|
|
'examples/yolov8'
|
|
]
|
|
|
|
[package]
|
|
name = "ort"
|
|
description = "A safe Rust wrapper for ONNX Runtime 1.16 - Optimize and Accelerate Machine Learning Inferencing"
|
|
version = "2.0.0-alpha.0"
|
|
edition = "2021"
|
|
rust-version = "1.70"
|
|
license = "MIT OR Apache-2.0"
|
|
repository = "https://github.com/pykeio/ort"
|
|
documentation = "https://ort.pyke.io/"
|
|
readme = "README.md"
|
|
keywords = [ "machine-learning", "ai", "ml" ]
|
|
categories = [ "algorithms", "mathematics", "science", "science" ]
|
|
authors = [
|
|
"pyke.io <contact@pyke.io>",
|
|
"Nicolas Bigaouette <nbigaouette@gmail.com>"
|
|
]
|
|
include = [ "src/", "examples/", "tests/", "LICENSE-APACHE", "LICENSE-MIT", "README.md" ]
|
|
|
|
[profile.release]
|
|
opt-level = 3
|
|
lto = true
|
|
strip = true
|
|
codegen-units = 1
|
|
|
|
[package.metadata.docs.rs]
|
|
features = [ "half", "fetch-models", "copy-dylibs" ]
|
|
|
|
[features]
|
|
default = [ "half", "download-binaries", "copy-dylibs" ]
|
|
|
|
profiling = [ "widestring" ]
|
|
custom-ops = [ "libc", "winapi" ]
|
|
|
|
fetch-models = [ "ureq" ]
|
|
download-binaries = [ "ort-sys/download-binaries" ]
|
|
load-dynamic = [ "libloading", "ort-sys/load-dynamic" ]
|
|
copy-dylibs = [ "ort-sys/copy-dylibs" ]
|
|
|
|
cuda = [ "ort-sys/cuda" ]
|
|
tensorrt = [ "ort-sys/tensorrt" ]
|
|
openvino = []
|
|
onednn = []
|
|
directml = []
|
|
nnapi = []
|
|
coreml = []
|
|
xnnpack = []
|
|
rocm = [ "ort-sys/rocm" ]
|
|
acl = []
|
|
armnn = []
|
|
tvm = []
|
|
migraphx = []
|
|
rknpu = []
|
|
vitis = []
|
|
cann = []
|
|
qnn = []
|
|
|
|
[dependencies]
|
|
ndarray = "0.15"
|
|
thiserror = "1.0"
|
|
ort-sys = { version = "2.0.0-alpha.0", path = "ort-sys" }
|
|
libloading = { version = "0.8", optional = true }
|
|
smallvec = "1.11"
|
|
|
|
ureq = { version = "2.1", optional = true, default-features = false, features = [ "tls" ] }
|
|
once_cell = "1.18"
|
|
tracing = "0.1"
|
|
half = { version = "2.1", optional = true }
|
|
|
|
[target.'cfg(unix)'.dependencies]
|
|
libc = { version = "0.2", optional = true }
|
|
|
|
[target.'cfg(windows)'.dependencies]
|
|
winapi = { version = "0.3", optional = true, features = [ "std", "libloaderapi" ] }
|
|
widestring = { version = "1.0", optional = true }
|
|
|
|
[dev-dependencies]
|
|
anyhow = "1.0"
|
|
ureq = "2.1"
|
|
image = "0.24"
|
|
test-log = { version = "0.2", default-features = false, features = [ "trace" ] }
|
|
tracing-subscriber = { version = "0.3", default-features = false, features = [ "env-filter", "fmt" ] }
|