saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-07 04:39:07 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Use unregister_custom_op_symbolic to unregister torch symbolics (#12146)

Browse source 
Description: Use unregister_custom_op_symbolic to unregister torch symbolics

Motivation and Context

Fixes #11305
This commit is contained in:
Justin Chu 2022-07-21 10:47:53 -07:00 committed by GitHub
parent 496618594f
commit 3d2bcb3386
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 8 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
onnxruntime/python/tools/pytorch_export_contrib_ops.py
View file

@ -94,10 +94,12 @@ def register():
def unregister():
"""Unregister ONNX Runtime's built-in contrib ops."""
# TODO: replace this once PyTorch supports unregister natively.
# https://msdata.visualstudio.com/Vienna/_workitems/edit/1342343
for name in _registered_ops:
ns, kind = name.split("::")
for version in sym_help._onnx_stable_opsets:
if version >= _OPSET_VERSION and sym_registry.is_registered_op(kind, ns, version):
del sym_registry._registry[(ns, version)][kind]
try:
torch.onnx.unregister_custom_op_symbolic(name, _OPSET_VERSION)
except AttributeError:
# unregister_custom_op_symbolic is not available before PyTorch 1.12
namespace, kind = name.split("::")
for version in sym_help._onnx_stable_opsets:
if version >= _OPSET_VERSION and sym_registry.is_registered_op(kind, namespace, version):
del sym_registry._registry[(namespace, version)][kind]