From 4e5d168d9a785b103dd7eb9ce9e13349887fd828 Mon Sep 17 00:00:00 2001 From: Han Qi Date: Wed, 24 Apr 2024 09:52:23 -0700 Subject: [PATCH] Update deps to torch 2.3 --- experimental/torch_xla2/dev-requirements.txt | 2 +- experimental/torch_xla2/test/test_ops.py | 1 + 2 files changed, 2 insertions(+), 1 deletion(-) diff --git a/experimental/torch_xla2/dev-requirements.txt b/experimental/torch_xla2/dev-requirements.txt index 004a1d71ad75..208f70d5fef8 100644 --- a/experimental/torch_xla2/dev-requirements.txt +++ b/experimental/torch_xla2/dev-requirements.txt @@ -1,3 +1,3 @@ -f https://download.pytorch.org/whl/torch -torch==2.2.1+cpu +torch==2.3.0+cpu ruff~=0.3.5 diff --git a/experimental/torch_xla2/test/test_ops.py b/experimental/torch_xla2/test/test_ops.py index ed14e636e5cb..72a39ae85824 100644 --- a/experimental/torch_xla2/test/test_ops.py +++ b/experimental/torch_xla2/test/test_ops.py @@ -570,6 +570,7 @@ "special.xlog1py", "split", "split_with_sizes", + "split_with_sizes_copy", "sqrt", "square", "stack",