Skip to content

Commit

Permalink
Support Pytorch v2.4 (#103)
Browse files Browse the repository at this point in the history
* Add workflows to test Pytorch v2.4.
* Removed old pytorch version tests workflows.
* Update testsuite workflows frameworks versions.

---------

Co-authored-by: Ofir Gordon <[email protected]>
  • Loading branch information
ofirgo and Ofir Gordon authored Sep 4, 2024
1 parent aaee971 commit a347912
Show file tree
Hide file tree
Showing 7 changed files with 62 additions and 18 deletions.
13 changes: 13 additions & 0 deletions .github/workflows/run_comp_test_torch24_v14.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
name: Run Backward Compatibility Test - Pytorch 2.4 MCTQ v1.4.0
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *

jobs:
run-comp-torch-2_4-v1_4:
uses: ./.github/workflows/compatibility_torch_tests.yml
with:
save_version: "v1.4.0"
python_version: "3.10"
torch_version: "2.4.*"
13 changes: 13 additions & 0 deletions .github/workflows/run_forward_comp_test_torch24_v14.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
name: Run Forward Compatibility Test - Pytorch 2.4 MCTQ v1.4.0
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *

jobs:
run-comp-torch-2_4-v1_4:
uses: ./.github/workflows/forward_compatibility_torch_tests.yml
with:
load_version: "v1.4.0"
python_version: "3.10"
torch_version: "2.4.*"
4 changes: 2 additions & 2 deletions .github/workflows/run_tests_suite_python310.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install tensorflow==2.12.*
pip install torch==1.13.* onnx onnxruntime onnxruntime-extensions
pip install tensorflow==2.15.*
pip install torch==2.4.* onnx onnxruntime onnxruntime-extensions
- name: Run unittests
run: python -m unittest discover -s tests -v
29 changes: 29 additions & 0 deletions .github/workflows/run_tests_suite_python311.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,29 @@
name: Python 3.11
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *
pull_request:
branches:
- main

jobs:
build:
runs-on: ubuntu-latest
timeout-minutes: 10
env:
COVERAGE_THRESHOlD: 80
steps:
- uses: actions/checkout@v4
- name: Install Python 3
uses: actions/setup-python@v5
with:
python-version: '3.11'
- name: Install dependencies
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install tensorflow==2.15.*
pip install torch==2.4.* onnx onnxruntime onnxruntime-extensions
- name: Run unittests
run: python -m unittest discover -s tests -v
4 changes: 2 additions & 2 deletions .github/workflows/run_tests_suite_python39.yml
Original file line number Diff line number Diff line change
Expand Up @@ -23,7 +23,7 @@ jobs:
run: |
python -m pip install --upgrade pip
pip install -r requirements.txt
pip install tensorflow==2.12.*
pip install torch==1.13.* onnx onnxruntime onnxruntime-extensions
pip install tensorflow==2.15.*
pip install torch==2.4.* onnx onnxruntime onnxruntime-extensions
- name: Run unittests
run: python -m unittest discover -s tests -v
11 changes: 0 additions & 11 deletions .github/workflows/run_tests_torch1_13.yml

This file was deleted.

Original file line number Diff line number Diff line change
@@ -1,11 +1,11 @@
name: Run Tests - PyTorch 1.12
name: Run Tests - PyTorch 2.4
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *

jobs:
run-pytorch-1_13:
run-pytorch-2_4:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
torch-version: "1.12.*"
torch-version: "2.4.*"

0 comments on commit a347912

Please sign in to comment.