Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Support RegisterCustomOpsLibrary via the Python API #4764

Merged
merged 37 commits into from
Aug 28, 2020
Merged
Changes from 1 commit
Commits
Show all changes
37 commits
Select commit Hold shift + click to select a range
306a114
Initial commit
hariharans29 Aug 10, 2020
8bcc970
More changes
hariharans29 Aug 12, 2020
b033785
Resolve conflicts
hariharans29 Aug 12, 2020
ebb6987
Nit
hariharans29 Aug 12, 2020
a72aa6d
Nit
hariharans29 Aug 12, 2020
5a735ed
Fix warning
hariharans29 Aug 12, 2020
f65e504
More changes
hariharans29 Aug 12, 2020
cf70867
Debug linux failure
hariharans29 Aug 12, 2020
08ebd49
debug
hariharans29 Aug 12, 2020
a1c53c9
debug 2
hariharans29 Aug 12, 2020
62064b8
Debug
hariharans29 Aug 13, 2020
6514806
More changes
hariharans29 Aug 13, 2020
aecbc45
More changes
hariharans29 Aug 14, 2020
14b0b36
Changes
hariharans29 Aug 14, 2020
03569a0
Fix link
hariharans29 Aug 14, 2020
bb4bf92
Fix training build
hariharans29 Aug 19, 2020
29bb93f
Merge master and resolve conflicts
hariharans29 Aug 19, 2020
0f08762
Fix build
hariharans29 Aug 19, 2020
fa64d83
Fix build
hariharans29 Aug 19, 2020
d46b900
Merge remote-tracking branch 'origin/master' into CustomOpPythonApi
hariharans29 Aug 24, 2020
537337c
PR feednack
hariharans29 Aug 25, 2020
dcc1963
Fix training build
hariharans29 Aug 25, 2020
3c554ad
Fix training build
hariharans29 Aug 25, 2020
adfa4e3
Debug training build
hariharans29 Aug 25, 2020
7f562be
a
hariharans29 Aug 25, 2020
bc1e4d3
Fix training build
hariharans29 Aug 25, 2020
ad8ff15
Revert some changes
hariharans29 Aug 25, 2020
61450d4
More changes
hariharans29 Aug 27, 2020
51d8f43
Fix mac build
hariharans29 Aug 27, 2020
e0977b4
Fix build
hariharans29 Aug 27, 2020
bb8e0b1
More changes
hariharans29 Aug 27, 2020
ac4f920
Log
hariharans29 Aug 27, 2020
feb8501
Nit
hariharans29 Aug 27, 2020
82ef63d
Nits
hariharans29 Aug 27, 2020
1d0d7ed
refinement
hariharans29 Aug 27, 2020
57b4490
Nit
hariharans29 Aug 28, 2020
a841e6d
PR feedback
hariharans29 Aug 28, 2020
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
Fix link
  • Loading branch information
hariharans29 committed Aug 14, 2020
commit 03569a0c180bdfa1a9523d1b2bb9c3683072a697
2 changes: 1 addition & 1 deletion docs/AddingCustomOp.md
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,7 @@ Note: These APIs are experimental and will change in the next release. They're r
* Call OrtAddCustomOpDomain to add the custom domain of ops to the session options
See [this](../onnxruntime/test/shared_lib/test_inference.cc) for an example called MyCustomOp that uses the C++ helper API (onnxruntime_cxx_api.h).
You can also compile the custom ops into a shared library and use that to run a model via the C++ API. The same test file contains an example.
hariharans29 marked this conversation as resolved.
Show resolved Hide resolved
The source code for a sample custom op shared library containing two custom kernels is [here](..onnxruntime/test/testdata/custom_op_library/custom_op_library.cc).
The source code for a sample custom op shared library containing two custom kernels is [here](../onnxruntime/test/testdata/custom_op_library/custom_op_library.cc).
See [this](../onnxruntime/test/python/onnxruntime_test_python.py) for an example called testRegisterCustomOpsLibrary that uses the Python API
to register a shared library that contains custom op kernels.
Currently, the only supported Execution Providers (EPs) for custom ops registered via this approach are the `CUDA` and the `CPU` EPs.
Expand Down