ONNX Runtime: cross-platform, high performance ML inferencing and training accelerator
Find a file
Alexey Gladyshev 7dc7529ec8
[TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505)
* add support for bool type

* add TVM EP support for tests

* include TVM EP in python test pool

* fix pylint

* moved technical imports to a separate file

* clean up post build actions & move _ld_preload.py extension to CMake level

* add files for include TVM EP into CI

* implement custom logger for TVM

* replace TVM logging with ONNX RT logging

* update link for TVM EP tutorial

* clean up TVM EP cmake

* add pybind auto enabling for TVM EP

* fix blank spaces

* code review fixes

* replace print with comment

* add list of EP without TVM EP

* enable onnx tests

* disable contrib ops and ml ops

* reuse Dockerfile.ubuntu

* Move install_tvm_test_dependencies.sh out of Docker context dir, update build definition.

Co-authored-by: Edward Chen <18449977+edgchen1@users.noreply.github.com>
2022-02-24 16:24:23 +01:00
.gdn Update compliance tasks in python packaging pipeline and fix some compile warnings (#8471) 2021-07-30 17:16:37 -07:00
.github Update C/C++ API docs automation to create a PR (instead of push to publish branch) (#10093) 2022-01-07 16:16:47 -08:00
cgmanifests [TVM EP] Rename Standalone TVM (STVM) Execution Provider to TVM EP (#10260) 2022-02-15 10:21:02 +01:00
cmake [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
csharp Use IntPtr instead of int conversion for pointer in Memory.Pin() (#10485) 2022-02-16 14:49:56 -08:00
dockerfiles Merged PR 6917440: ONNX Runtime update from GitHub master 2022-02-04 10:13:38 +00:00
docs [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
include/onnxruntime/core Add restrictions for hybrid cpus for thread pool task distribution (#10393) 2022-02-17 14:34:09 -08:00
java [TVM EP] Rename Standalone TVM (STVM) Execution Provider to TVM EP (#10260) 2022-02-15 10:21:02 +01:00
js [js/web] fix lint error when run without ort-web TS types (#10429) 2022-02-17 22:34:38 -08:00
objectivec [Objective-C API] WIgnore clang documentation warnings from C/C++ header usage. (#9057) 2021-09-14 13:03:48 -07:00
onnxruntime [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
orttraining [python] [orttraining] Add utility to export a graph to compute gradients (#8125) 2022-02-18 14:00:49 -08:00
package/rpm Bump master version to 1.11 (#9957) 2021-12-14 23:32:06 -08:00
samples Add Python checks pipeline (#7032) 2021-08-09 10:37:05 -07:00
server [TVM EP] Rename Standalone TVM (STVM) Execution Provider to TVM EP (#10260) 2022-02-15 10:21:02 +01:00
tools [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
winml Merge pull request #10619 from microsoft/user/dwayner/DmlDev20220221 2022-02-23 01:09:26 -08:00
.clang-format
.clang-tidy
.dockerignore
.flake8 Add Python checks pipeline (#7032) 2021-08-09 10:37:05 -07:00
.gitattributes
.gitignore Merged PR 6917440: ONNX Runtime update from GitHub master 2022-02-04 10:13:38 +00:00
.gitmodules Merged PR 6917440: ONNX Runtime update from GitHub master 2022-02-04 10:13:38 +00:00
build.amd64.1411.bat
build.bat
build.sh
CITATION.cff Add citation file (#10061) 2021-12-16 19:56:21 -08:00
CODEOWNERS Merge two helpers involving the kernel def hashes into one file (#10609) 2022-02-23 20:46:09 +10:00
CONTRIBUTING.md fixed the link (#8757) 2021-08-18 11:45:42 -07:00
LICENSE
NuGet.config
ort.wprp
packages.config Bump winrt version (#10243) 2022-01-12 10:52:27 -08:00
README.md Fix typo 2021-08-12 15:57:15 -07:00
requirements-dev.txt Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00
requirements-doc.txt
requirements-training.txt Add post-install command to build PyTorch CPP extensions from within onnxruntime package (#8027) 2021-06-28 18:11:58 -07:00
requirements.txt.in Chang how numpy version is handled. (#8130) 2021-06-23 14:08:37 -07:00
setup.py [TVM EP] Integrate tests for TVM EP into public onnxruntime CI (#10505) 2022-02-24 16:24:23 +01:00
ThirdPartyNotices.txt add copyright (#9943) (#9970) 2021-12-08 14:34:53 -08:00
VERSION_NUMBER Bump master version to 1.11 (#9957) 2021-12-14 23:32:06 -08:00

ONNX Runtime is a cross-platform inference and training machine-learning accelerator.

ONNX Runtime inference can enable faster customer experiences and lower costs, supporting models from deep learning frameworks such as PyTorch and TensorFlow/Keras as well as classical machine learning libraries such as scikit-learn, LightGBM, XGBoost, etc. ONNX Runtime is compatible with different hardware, drivers, and operating systems, and provides optimal performance by leveraging hardware accelerators where applicable alongside graph optimizations and transforms. Learn more →

ONNX Runtime training can accelerate the model training time on multi-node NVIDIA GPUs for transformer models with a one-line addition for existing PyTorch training scripts. Learn more →

Get Started

General Information: onnxruntime.ai

Usage documention and tutorials: onnxruntime.ai/docs

Companion sample repositories:

Build Pipeline Status

System CPU GPU EPs
Windows Build Status Build Status Build Status
Linux Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Build Status
Mac Build Status
Build Status
Android Build Status
iOS Build Status
WebAssembly Build Status

Data/Telemetry

Windows distributions of this project may collect usage data and send it to Microsoft to help improve our products and services. See the privacy statement for more details.

Contributions and Feedback

We welcome contributions! Please see the contribution guidelines.

For feature requests or bug reports, please file a GitHub Issue.

For general discussion or questions, please use GitHub Discussions.

Code of Conduct

This project has adopted the Microsoft Open Source Code of Conduct. For more information see the Code of Conduct FAQ or contact opencode@microsoft.com with any additional questions or comments.

License

This project is licensed under the MIT License.