We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c5740bd commit 0063681Copy full SHA for 0063681
onnx_diagnostic/export/api.py
@@ -91,7 +91,9 @@ def to_onnx(
91
**(exporter_kwargs or {}),
92
dispatcher=dispatcher if use_control_flow_dispatcher else None,
93
)
94
+
95
if exporter in ("dynamo", "onnx-dynamo"):
96
+ import os
97
import onnxscript.rewriter.ort_fusions as ort_fusions
98
99
assert (
@@ -106,6 +108,9 @@ def to_onnx(
106
108
opset_version=target_opset,
107
109
dynamic_shapes=dynamic_shapes,
110
dynamo=True,
111
+ verbose=verbose,
112
+ dump_exported_program=bool(save_ep),
113
+ artifacts_dir=os.path.dirname(filename) if filename else ".",
114
115
116
if optimize:
0 commit comments