From 5bed1e9acacfaacaff1f712a871a2022bc0d479e Mon Sep 17 00:00:00 2001
From: gmeanti <giacomo.meanti@gmail.com>
Date: Mon, 12 Feb 2024 12:14:34 +0100
Subject: [PATCH] Update pytorch versions

---
 .github/workflows/build_docs.yml   |  2 +-
 .github/workflows/build_wheels.yml | 28 +++++-----------------------
 .github/workflows/testing.yml      |  2 +-
 3 files changed, 7 insertions(+), 25 deletions(-)

diff --git a/.github/workflows/build_docs.yml b/.github/workflows/build_docs.yml
index 4d9b29f..0a23418 100644
--- a/.github/workflows/build_docs.yml
+++ b/.github/workflows/build_docs.yml
@@ -11,7 +11,7 @@ jobs:
       matrix:
         os: [ubuntu-latest]
         python-version: ['3.8']
-        torch-version: ['2.1.0']
+        torch-version: ['2.2.0']
 
     steps:
       - uses: actions/checkout@v3
diff --git a/.github/workflows/build_wheels.yml b/.github/workflows/build_wheels.yml
index 06358ed..ba376aa 100644
--- a/.github/workflows/build_wheels.yml
+++ b/.github/workflows/build_wheels.yml
@@ -16,34 +16,16 @@ jobs:
         os: [ubuntu-20.04]
         # support version based on: https://download.pytorch.org/whl/torch/
         python-version: ['3.8', '3.9', '3.10', '3.11']
-        torch-version: [1.13.0, 2.0.0, 2.1.0]
-        cuda-version: ['cpu', 'cu113', 'cu115', 'cu116', 'cu117', 'cu118', 'cu121']
+        torch-version: [2.0.0, 2.1.0, 2.2.0]
+        cuda-version: ['cpu', 'cu117', 'cu118', 'cu121']
         exclude:
-          - torch-version: 1.13.0
-            cuda-version: 'cu113'
-          - torch-version: 1.13.0
-            cuda-version: 'cu115'
-          - torch-version: 1.13.0
-            cuda-version: 'cu118'
-          - torch-version: 1.13.0
-            cuda-version: 'cu121'
-
-          - torch-version: 2.0.0
-            cuda-version: 'cu113'
-          - torch-version: 2.0.0
-            cuda-version: 'cu115'
-          - torch-version: 2.0.0
-            cuda-version: 'cu116'
           - torch-version: 2.0.0
             cuda-version: 'cu121'
 
           - torch-version: 2.1.0
-            cuda-version: 'cu113'
-          - torch-version: 2.1.0
-            cuda-version: 'cu115'
-          - torch-version: 2.1.0
-            cuda-version: 'cu116'
-          - torch-version: 2.1.0
+            cuda-version: 'cu117'
+
+          - torch-version: 2.2.0
             cuda-version: 'cu117'
             
     steps:
diff --git a/.github/workflows/testing.yml b/.github/workflows/testing.yml
index 3b2662d..16c7304 100644
--- a/.github/workflows/testing.yml
+++ b/.github/workflows/testing.yml
@@ -10,7 +10,7 @@ jobs:
       matrix:
         os: [ubuntu-20.04]
         python-version: ['3.8']
-        torch-version: [2.0.0, 2.1.0]
+        torch-version: [2.0.0, 2.2.0]
 
     steps:
       - name: Checkout with submodules