-
-
Notifications
You must be signed in to change notification settings - Fork 106
/
Cargo.toml
107 lines (95 loc) · 2.96 KB
/
Cargo.toml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
[workspace]
members = [
'ort-sys',
'examples/async-gpt2-api',
'examples/custom-ops',
'examples/gpt2',
'examples/model-info',
'examples/yolov8',
'examples/phi-3-vision',
'examples/modnet',
'examples/sentence-transformers',
'examples/training'
]
default-members = [
'.',
'examples/async-gpt2-api',
'examples/custom-ops',
'examples/gpt2',
'examples/model-info',
'examples/yolov8',
'examples/modnet',
'examples/sentence-transformers'
]
exclude = [ 'examples/cudarc' ]
[package]
name = "ort"
description = "A safe Rust wrapper for ONNX Runtime 1.20 - Optimize and accelerate machine learning inference & training"
version = "2.0.0-rc.9"
edition = "2021"
rust-version = "1.70"
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" ]
categories = [ "algorithms", "mathematics", "science", "science" ]
authors = [
"pyke.io <[email protected]>",
"Nicolas Bigaouette <[email protected]>"
]
include = [ "src/", "benches/", "LICENSE-APACHE", "LICENSE-MIT", "README.md" ]
[profile.release]
opt-level = 3
lto = true
strip = true
codegen-units = 1
[package.metadata.docs.rs]
features = [ "ndarray", "half", "training", "fetch-models", "load-dynamic", "copy-dylibs" ]
targets = ["x86_64-unknown-linux-gnu"]
rustdoc-args = [ "--cfg", "docsrs" ]
[features]
default = [ "ndarray", "half", "download-binaries", "copy-dylibs" ]
training = [ "ort-sys/training" ]
fetch-models = [ "ureq", "sha2" ]
download-binaries = [ "ort-sys/download-binaries" ]
load-dynamic = [ "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" ]
[dependencies]
ndarray = { version = "0.16", optional = true }
ort-sys = { version = "=2.0.0-rc.9", path = "ort-sys" }
libloading = { version = "0.8", optional = true }
ureq = { version = "2.1", optional = true, default-features = false, features = [ "tls" ] }
sha2 = { version = "0.10", optional = true }
tracing = { version = "0.1", default-features = false, features = [ "std" ] }
half = { version = "2.1", optional = true }
[dev-dependencies]
anyhow = "1.0"
ureq = "2.1"
image = "0.25"
test-log = { version = "0.2", default-features = false, features = [ "trace" ] }
tracing-subscriber = { version = "0.3", default-features = false, features = [ "env-filter", "fmt" ] }
glassbench = "0.4"
tokio = { version = "1.36", features = [ "test-util" ] }
tokio-test = "0.4.3"
[[bench]]
name = "squeezenet"
harness = false