Setup: bump torch version for LLM tests (#1154)

Xilinx · Jan 14, 2025 · 0f3877f · 0f3877f
1 parent c44566d
commit 0f3877f
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/.github/workflows/examples_llm_pytest.yml b/.github/workflows/examples_llm_pytest.yml
@@ -16,7 +16,7 @@ jobs:
 
       matrix:
         python_version: ['3.9', '3.10']
-        pytorch_version: ['2.2.2', '2.3.1', '2.4.0']
+        pytorch_version: ['2.2.2', '2.3.1', '2.4.1']
         platform: ['windows-latest', 'ubuntu-latest', 'macos-latest']
         jit_status: ['jit_disabled', 'jit_enabled']
 

diff --git a/.github/workflows/gen_github_actions.py b/.github/workflows/gen_github_actions.py
@@ -12,7 +12,8 @@
     '2.1.0': '0.16.0',
     '2.2.2': '0.17.2',
     '2.3.1': '0.18.1',
-    '2.4.0': '0.19.0'}
+    '2.4.0': '0.19.0',
+    '2.4.1': '0.19.1'}
 
 BASE_YML_TEMPLATE = 'base.yml.template'
 BASE_YML_REDUCED_TEMPLATE = 'base_reduced.yml.template'
@@ -38,7 +39,7 @@
 
 EXAMPLES_LLM_PYTEST_MATRIX_REDUCED = od([('python_version', list(PYTHON_VERSIONS_REDUCED)),
                                          ('pytorch_version', list(
-                                             ('2.4.0',))), ('platform', PLATFORM_LIST_REDUCED)])
+                                             ('2.4.1',))), ('platform', PLATFORM_LIST_REDUCED)])
 
 FINN_MATRIX_REDUCED = od([('python_version', list(PYTHON_VERSIONS_REDUCED)),
                           ('pytorch_version', list(PYTORCH_LIST_REDUCED)),
@@ -72,7 +73,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 = ('2.2.2', '2.3.1', '2.4.1')
 EXAMPLES_LLM_PYTEST_MATRIX = od([('python_version', list(PYTHON_VERSIONS)),
                                  ('pytorch_version', list(EXAMPLES_LLM_PYTEST_PYTORCH_VERSIONS)),
                                  ('platform', PLATFORM_LIST)])

diff --git a/.github/workflows/reduced_examples_llm_pytest.yml b/.github/workflows/reduced_examples_llm_pytest.yml
@@ -18,7 +18,7 @@ jobs:
 
       matrix:
         python_version: ['3.9']
-        pytorch_version: ['2.4.0']
+        pytorch_version: ['2.4.1']
         platform: ['ubuntu-latest']
         jit_status: ['jit_disabled']