Update torch version

Xilinx · Oct 31, 2024 · 7a48aa1 · 7a48aa1
1 parent 57fc394
commit 7a48aa1
Show file tree

Hide file tree

Showing 17 changed files with 22 additions and 97 deletions.
diff --git a/.github/workflows/develop_install.yml b/.github/workflows/develop_install.yml
@@ -16,14 +16,10 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/end_to_end.yml b/.github/workflows/end_to_end.yml
@@ -16,14 +16,11 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
 
 
         exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
           - platform: 'windows-latest'
 
 

diff --git a/.github/workflows/examples_llm_pytest.yml b/.github/workflows/examples_llm_pytest.yml
@@ -16,18 +16,11 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['2.2.2', '2.3.1', '2.4.0']
+        pytorch_version: ['2.2.2', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
         jit_status: ['jit_disabled', 'jit_enabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
-          - pytorch_version: '1.9.1'
-            jit_status: 'jit_enabled'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/examples_pytest.yml b/.github/workflows/examples_pytest.yml
@@ -16,18 +16,11 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
         jit_status: ['jit_disabled', 'jit_enabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
-          - pytorch_version: '1.9.1'
-            jit_status: 'jit_enabled'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/finn_integration.yml b/.github/workflows/finn_integration.yml
@@ -16,14 +16,10 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/gen_github_actions.py b/.github/workflows/gen_github_actions.py
@@ -18,7 +18,7 @@
 # Reduced Test for PRs, except when a review is requested
 PYTHON_VERSIONS_REDUCED = ('3.8',)
 
-PYTORCH_LIST_REDUCED = ["1.9.1", "1.13.0", "2.1.0"]
+PYTORCH_LIST_REDUCED = ["2.0.1", "2.2.2", "2.5"]
 
 PLATFORM_LIST_REDUCED = ['ubuntu-latest']
 
@@ -40,7 +40,7 @@
 # Data shared betwen Nox sessions and Github Actions, formatted as tuples
 PYTHON_VERSIONS = ('3.8', '3.9')
 
-PYTORCH_VERSIONS = ('1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0')
+PYTORCH_VERSIONS = ('2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5')
 JIT_STATUSES = ('jit_disabled', 'jit_enabled')
 
 # Data used only by Github Actions, formatted as lists or lists of ordered dicts
@@ -49,13 +49,7 @@
 
 STRATEGY = od([('fail-fast', 'false')])
 
-EXCLUDE_LIST = generate_exclusion_list([[
-    ['pytorch_version', ['1.9.1']],
-    ['platform', ['macos-latest']],]])
-
-JIT_EXCLUDE_LIST = generate_exclusion_list([[['pytorch_version', ['1.9.1']],
-                                             ['jit_status', [
-                                                 'jit_enabled',]]]])
+EXCLUDE_LIST = []
 
 NOTEBOOK_EXCLUDE_LIST = generate_exclusion_list([[['platform', [
     'macos-latest',]]]])
@@ -66,7 +60,7 @@
 MATRIX = od([('python_version', list(PYTHON_VERSIONS)), ('pytorch_version', list(PYTORCH_VERSIONS)),
              ('platform', PLATFORM_LIST)])
 
-EXAMPLES_LLM_PYTEST_PYTORCH_VERSIONS = ('2.2.2', '2.3.1', '2.4.0')
+EXAMPLES_LLM_PYTEST_PYTORCH_VERSIONS = tuple(PYTORCH_LIST_REDUCED[1:])
 EXAMPLES_LLM_PYTEST_MATRIX = od([('python_version', list(PYTHON_VERSIONS)),
                                  ('pytorch_version', list(EXAMPLES_LLM_PYTEST_PYTORCH_VERSIONS)),
                                  ('platform', PLATFORM_LIST)])
@@ -153,7 +147,7 @@
 def gen_pytest_yml():
     pytest = Action(
         'Pytest',
-        EXCLUDE_LIST + JIT_EXCLUDE_LIST,
+        EXCLUDE_LIST,
         combine_od_list([MATRIX, PYTEST_MATRIX_EXTRA]),
         PYTEST_STEP_LIST,
         STRATEGY)
@@ -170,7 +164,7 @@ def gen_pytest_yml():
 def gen_examples_pytest_yml():
     pytest = Action(
         'Examples Pytest',
-        EXCLUDE_LIST + JIT_EXCLUDE_LIST,
+        EXCLUDE_LIST,
         combine_od_list([MATRIX, PYTEST_MATRIX_EXTRA]),
         EXAMPLES_PYTEST_STEP_LIST,
         STRATEGY)
@@ -187,7 +181,7 @@ def gen_examples_pytest_yml():
 def gen_examples_llm_pytest_yml():
     pytest = Action(
         'Examples LLM Pytest',
-        EXCLUDE_LIST + JIT_EXCLUDE_LIST,
+        EXCLUDE_LIST,
         combine_od_list([EXAMPLES_LLM_PYTEST_MATRIX, PYTEST_MATRIX_EXTRA]),
         EXAMPLES_LLM_PYTEST_STEP_LIST,
         STRATEGY)

diff --git a/.github/workflows/notebook.yml b/.github/workflows/notebook.yml
@@ -16,14 +16,11 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
 
 
         exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
           - platform: 'macos-latest'
 
 

diff --git a/.github/workflows/ort_integration.yml b/.github/workflows/ort_integration.yml
@@ -16,14 +16,10 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/pytest.yml b/.github/workflows/pytest.yml
@@ -16,18 +16,11 @@ jobs:
 
       matrix:
         python_version: ['3.8', '3.9']
-        pytorch_version: ['1.9.1', '1.10.1', '1.11.0', '1.12.1', '1.13.0', '2.0.1', '2.1.0']
+        pytorch_version: ['2.0.1', '2.1.2', '2.2.2', '2.3.1', '2.4.1', '2.5']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
         jit_status: ['jit_disabled', 'jit_enabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
-          - pytorch_version: '1.9.1'
-            jit_status: 'jit_enabled'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_develop_install.yml b/.github/workflows/reduced_develop_install.yml
@@ -18,14 +18,10 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_end_to_end.yml b/.github/workflows/reduced_end_to_end.yml
@@ -18,14 +18,11 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
 
 
         exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
           - platform: 'windows-latest'
 
 

diff --git a/.github/workflows/reduced_examples_llm_pytest.yml b/.github/workflows/reduced_examples_llm_pytest.yml
@@ -23,10 +23,6 @@ jobs:
         jit_status: ['jit_disabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_examples_pytest.yml b/.github/workflows/reduced_examples_pytest.yml
@@ -18,15 +18,11 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
         jit_status: ['jit_disabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_finn_integration.yml b/.github/workflows/reduced_finn_integration.yml
@@ -18,14 +18,10 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_notebook.yml b/.github/workflows/reduced_notebook.yml
@@ -18,14 +18,11 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
 
 
         exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
           - platform: 'macos-latest'
 
 

diff --git a/.github/workflows/reduced_ort_integration.yml b/.github/workflows/reduced_ort_integration.yml
@@ -18,14 +18,10 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}

diff --git a/.github/workflows/reduced_pytest.yml b/.github/workflows/reduced_pytest.yml
@@ -18,15 +18,11 @@ jobs:
 
       matrix:
         python_version: ['3.8']
-        pytorch_version: ['1.9.1', '1.13.0', '2.1.0']
+        pytorch_version: ['2.0.1', '2.2.2', '2.5']
         platform: ['ubuntu-latest']
         jit_status: ['jit_disabled']
 
 
-        exclude:
-          - pytorch_version: '1.9.1'
-            platform: 'macos-latest'
-
 
 
     if: ${{ !github.event.pull_request.draft }}