Skip to content

Commit 9ab7527

Browse files
authored
Consolidate overloads in torchlib (#2604)
The goal is to have no overloads and remove the PyTorch dispatcher. Right now there are still the following ops that need to be addressed: ``` Registering private function: aten::as_strided Registering private function: aten::embedding_bag Registering private function: aten::embedding_bag.padding_idx Registering overload for function: aten::index.Tensor Registering overload for function: aten::_unsafe_index.Tensor Registering overload for function: aten::index_put ``` I did a bit of cleaning up in tests and torchlib as well. #2580 --------- Signed-off-by: Justin Chu <justinchuby@users.noreply.github.com>
1 parent 075fc4d commit 9ab7527

File tree

7 files changed

+308
-1092
lines changed

7 files changed

+308
-1092
lines changed

noxfile.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -29,9 +29,9 @@
2929
"ml-dtypes",
3030
)
3131
ONNX = "onnx==1.17"
32-
ONNX_RUNTIME = "onnxruntime==1.20.1"
33-
PYTORCH = "torch==2.5.1"
34-
TORCHVISON = "torchvision==0.20.1"
32+
ONNX_RUNTIME = "onnxruntime==1.23.0"
33+
PYTORCH = "torch==2.7.1"
34+
TORCHVISON = "torchvision==0.22.1"
3535
TRANSFORMERS = "transformers==4.37.2"
3636
ONNX_RUNTIME_NIGHTLY_DEPENDENCIES = (
3737
"flatbuffers",

onnxscript/backend/onnx_export_test.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -84,6 +84,7 @@ def skip(pattern: str | Pattern, reason: str, *, condition: bool = True):
8484
),
8585
skip(r"^test_ai_onnx_ml_label_encoder", "ONNX Runtime does not support Opset 21 at 1.17"),
8686
skip(r"^test_ai_onnx_ml_tree_ensemble", "Opset 23 is not supported"),
87+
skip(r"^test_attention", "ONNX Runtime 1.23 fails on these tests"),
8788
)
8889

8990
if sys.platform == "win32":

0 commit comments

Comments
 (0)