From 222e1cdb9b1d9a40457ca6feaa5dd91b9a825910 Mon Sep 17 00:00:00 2001 From: Justin Chu Date: Tue, 18 Feb 2025 12:22:26 -0800 Subject: [PATCH] Create stable apis for torch 2.7 --- onnxscript/_framework_apis/torch_2_7.py | 21 +++++++++++++++++++++ 1 file changed, 21 insertions(+) create mode 100644 onnxscript/_framework_apis/torch_2_7.py diff --git a/onnxscript/_framework_apis/torch_2_7.py b/onnxscript/_framework_apis/torch_2_7.py new file mode 100644 index 0000000000..ee5e6089e5 --- /dev/null +++ b/onnxscript/_framework_apis/torch_2_7.py @@ -0,0 +1,21 @@ +# Copyright (c) Microsoft Corporation. +# Licensed under the MIT License. +"""Stable APIs for PyTorch 2.7.""" + +from __future__ import annotations + +__all__ = [ + "check_model", + "convert_version", + "get_torchlib_ops", + "optimize", + "save_model_with_external_data", +] + +from onnxscript._framework_apis.torch_2_6 import ( + check_model, + convert_version, + get_torchlib_ops, + optimize, + save_model_with_external_data, +)