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

Update converter registry usage in orttraining_test_dort_custom_ops.py (#16663)

Browse source 
Fix Orttraining Linux Lazy Tensor CI       

Orttraining Linux Lazy Tensor CI is broken.
The error message is
AttributeError: 'OnnxRegistry' object has no attribute 'register'
This commit is contained in:
Ti-Tai Wang 2023-07-11 12:03:12 -07:00 committed by GitHub
parent d41bbac7b9
commit 72076e5320
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 10 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

19
orttraining/orttraining/test/python/orttraining_test_dort_custom_ops.py
View file

@ -39,11 +39,11 @@ def custom_exporter_for_aten_add_Tensor(x, y):
# exporter.
# Use custom_exporter_for_aten_add_Tensor.to_function_proto() to investigate
# function representing "aten::mul.Tensor".
DEFAULT_ONNX_EXPORTER_OPTIONS.onnxfunction_dispatcher.onnx_registry.register(
"aten::mul.Tensor",
DEFAULT_ONNX_EXPORTER_OPTIONS.opset_version,
custom_exporter_for_aten_add_Tensor,
True,
DEFAULT_ONNX_EXPORTER_OPTIONS.onnxfunction_dispatcher.onnx_registry.register_custom_op(
function=custom_exporter_for_aten_add_Tensor,
namespace="aten",
op_name="mul",
overload="Tensor",
)
@ -57,11 +57,10 @@ def custom_exporter_for_foo_bar_default(x):
# Ask exporter to map "torch.ops.foo.bar" to
# custom_exporter_for_foo_bar_default.
DEFAULT_ONNX_EXPORTER_OPTIONS.onnxfunction_dispatcher.onnx_registry.register(
"foo::bar",
DEFAULT_ONNX_EXPORTER_OPTIONS.opset_version,
custom_exporter_for_foo_bar_default,
True,
DEFAULT_ONNX_EXPORTER_OPTIONS.onnxfunction_dispatcher.onnx_registry.register_custom_op(
function=custom_exporter_for_foo_bar_default,
namespace="foo",
op_name="bar",
)