test (llm/nox/gha): Updated env setup

Xilinx · Aug 23, 2024 · 452016d · 452016d
1 parent 6352adf
commit 452016d
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 4 deletions.
diff --git a/.github/workflows/gen_github_actions.py b/.github/workflows/gen_github_actions.py
@@ -28,7 +28,7 @@
 
 EXAMPLES_LLM_PYTEST_MATRIX_REDUCED = od([('python_version', list(PYTHON_VERSIONS_REDUCED)),
                                          ('pytorch_version', list(
-                                             ('2.2.2',))), ('platform', PLATFORM_LIST_REDUCED)])
+                                             ('2.4.0',))), ('platform', PLATFORM_LIST_REDUCED)])
 
 FINN_MATRIX_REDUCED = od([('python_version', list(PYTHON_VERSIONS_REDUCED)),
                           ('pytorch_version', list(PYTORCH_LIST_REDUCED)),

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.8']
-        pytorch_version: ['2.2.2']
+        pytorch_version: ['2.4.0']
         platform: ['ubuntu-latest']
         jit_status: ['jit_disabled']
 

diff --git a/noxfile.py b/noxfile.py
@@ -141,8 +141,7 @@ def tests_brevitas_examples_llm(session, pytorch, jit_status):
     session.env['BREVITAS_JIT'] = '{}'.format(int(jit_status == 'jit_enabled'))
     install_pytorch(pytorch, session)
     install_torchvision(pytorch, session)  # Optimum seems to require torchvision
-    session.install(
-        '-e', '.[test, llm, export]', f'torch=={pytorch}' if IS_OSX else f'torch=={pytorch}+cpu')
+    session.install('-e', '.[test, llm, export]')
     session.run('pytest', '-n', 'logical', '-k', 'llm', 'tests/brevitas_examples/test_llm.py')