Skip to content

Commit

Permalink
test multiple ort
Browse files Browse the repository at this point in the history
  • Loading branch information
adityagoel4512 committed Sep 26, 2024
1 parent 7c5be95 commit 29cb451
Show file tree
Hide file tree
Showing 5 changed files with 758 additions and 123 deletions.
2 changes: 2 additions & 0 deletions .github/workflows/ci.yml
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ jobs:
- py311
- py312
- np1x
- ort117
- ort118
steps:
- name: Checkout branch
uses: actions/checkout@v4
Expand Down
5 changes: 5 additions & 0 deletions docs/intros/modelconversion.rst
Original file line number Diff line number Diff line change
Expand Up @@ -110,3 +110,8 @@ We can use an ONNX backend like onnxruntime to run our model. Here we use onnxru
print(out)
# array([0, 0, 0, 1, 2, 2, 0, 1, 2, 1, 0, 2, 1, 1, 1, 2, 2, 0, 0, 1, 0, 1, 1, 0, 0, 2, 2, 2, 2, 2])
.. note::

ONNX backends may not always support the entire ONNX specification and can sometimes miss kernel implementations for specific data types.
Since onnxruntime is such a common backend, ndonnx ensures proactively that any model generated using :func:`ndonnx.build` is compatible with at least the two latest onnxruntime releases on conda-forge.
2 changes: 2 additions & 0 deletions ndonnx/_build.py
Original file line number Diff line number Diff line change
Expand Up @@ -36,6 +36,8 @@ def build(
-------
out: onnx.ModelProto
ONNX model
Note: the generated ONNX model should be runnable on the latest two versions of onnxruntime documented in an ndonnx release.
"""

def collect_vars(name: str, arr: _CoreArray | Array):
Expand Down
Loading

0 comments on commit 29cb451

Please sign in to comment.