Skip to content

Commit 9f71ffc

Browse files
Enable version converter for torch>=2.6 (#2111)
Enable version converter for torch>=2.6
1 parent 57dbc70 commit 9f71ffc

File tree

1 file changed

+9
-2
lines changed

1 file changed

+9
-2
lines changed

onnxscript/_framework_apis/torch_2_6.py

+9-2
Original file line numberDiff line numberDiff line change
@@ -14,10 +14,9 @@
1414
]
1515
from typing import TYPE_CHECKING
1616

17-
from onnxscript import ir, optimizer
17+
from onnxscript import ir, optimizer, version_converter
1818
from onnxscript._framework_apis.torch_2_5 import (
1919
check_model,
20-
convert_version,
2120
get_torchlib_ops,
2221
save_model_with_external_data,
2322
)
@@ -32,6 +31,14 @@ def optimize(model: ir.Model) -> ir.Model:
3231
return model
3332

3433

34+
def convert_version(model: ir.Model, target_version: int) -> ir.Model:
35+
"""Convert the model to the specified ONNX opset version."""
36+
if target_version < 18:
37+
return model
38+
version_converter.convert_version(model, target_version)
39+
return model
40+
41+
3542
def torchlib_opset() -> Opset18:
3643
"""Return the default opset for torchlib."""
3744
import onnxscript # pylint: disable=import-outside-toplevel

0 commit comments

Comments
 (0)