diff --git a/.github/workflows/examples_llm_pytest.yml b/.github/workflows/examples_llm_pytest.yml index f0bd1c9eb..8c5554565 100644 --- 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 index a2138fd84..b644dc9d0 100644 --- 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 index dfb9296fd..af858e90c 100644 --- 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']