Skip to content

Commit

Permalink
feat: tract backend
Browse files Browse the repository at this point in the history
  • Loading branch information
decahedron1 committed Jan 8, 2025
1 parent 934929f commit f6f1f83
Show file tree
Hide file tree
Showing 10 changed files with 2,458 additions and 0 deletions.
1 change: 1 addition & 0 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@
members = [
'ort-sys',
'backends/candle',
'backends/tract',
'examples/async-gpt2-api',
'examples/custom-ops',
'examples/gpt2',
Expand Down
35 changes: 35 additions & 0 deletions backends/tract/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,35 @@
[package]
name = "ort-tract"
description = "ort + tract = 🦀 - An alternative backend for ort, powered by tract."
version = "0.1.0+0.8.1"
edition = "2021"
rust-version = "1.70"
license = "MIT OR Apache-2.0"
repository = "https://github.com/pykeio/ort"
homepage = "https://ort.pyke.io/"
keywords = [ "machine-learning", "ai", "ml" , "sys"]
categories = [ "algorithms", "mathematics", "science" ]
authors = [
"pyke.io <[email protected]>"
]

[lib]
name = "ort_tract"
path = "lib.rs"

[features]

[dependencies]
ort-sys = { version = "=2.0.0-rc.9", path = "../../ort-sys", default-features = false }
tract-onnx = "0.21"
parking_lot = "0.12"

[dev-dependencies]
ort = { version = "=2.0.0-rc.9", path = "../../", default-features = false, features = [ "alternative-backend", "fetch-models" ] }

[[test]]
name = "memory"
path = "tests/memory.rs"
[[test]]
name = "tensor"
path = "tests/tensor.rs"
Loading

0 comments on commit f6f1f83

Please sign in to comment.