Skip to content

Commit

Permalink
Add workflows for torch versions (#1112)
Browse files Browse the repository at this point in the history
Add workflows for torch versions 2.2 and 2.3
Update README versions table.

---------

Co-authored-by: reuvenp <reuvenp@altair-semi.com>
  • Loading branch information
reuvenperetz and reuvenp authored Jun 24, 2024
1 parent 3307fd0 commit 04bc6eb
Show file tree
Hide file tree
Showing 7 changed files with 46 additions and 12 deletions.
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Python 3.10, Pytorch 1.12
name: Python 3.10, Pytorch 2.2
on:
workflow_dispatch: # Allow manual triggers
schedule:
Expand All @@ -13,4 +13,4 @@ jobs:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.10"
torch-version: "1.12.*"
torch-version: "2.2.*"
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Python 3.10, Pytorch 1.13
name: Python 3.10, Pytorch 2.3
on:
workflow_dispatch: # Allow manual triggers
schedule:
Expand All @@ -13,4 +13,4 @@ jobs:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.10"
torch-version: "1.13.*"
torch-version: "2.3.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python311_pytorch22.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.11, Pytorch 2.2
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *
pull_request:
branches:
- main


jobs:
run-tests:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.11"
torch-version: "2.2.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python311_pytorch23.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.11, Pytorch 2.3
on:
workflow_dispatch: # Allow manual triggers
schedule:
- cron: 0 0 * * *
pull_request:
branches:
- main


jobs:
run-tests:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.11"
torch-version: "2.3.*"
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Python 3.9, Pytorch 1.12
name: Python 3.9, Pytorch 2.2
on:
workflow_dispatch: # Allow manual triggers
schedule:
Expand All @@ -13,4 +13,4 @@ jobs:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.9"
torch-version: "1.12.*"
torch-version: "2.2.*"
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
name: Python 3.9, Pytorch 1.13
name: Python 3.9, Pytorch 2.3
on:
workflow_dispatch: # Allow manual triggers
schedule:
Expand All @@ -13,4 +13,4 @@ jobs:
uses: ./.github/workflows/run_pytorch_tests.yml
with:
python-version: "3.9"
torch-version: "1.13.*"
torch-version: "2.3.*"
10 changes: 6 additions & 4 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -50,11 +50,13 @@ for hands-on learning. For example:

Currently, MCT is being tested on various Python, Pytorch and TensorFlow versions:

| | PyTorch 1.13 | PyTorch 2.0 | PyTorch 2.1 |

| | PyTorch 2.1 | PyTorch 2.2 | PyTorch 2.3 |
|-------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| Python 3.9 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch113.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch113.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch20.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch20.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch21.yml) |
| Python 3.10 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch112.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch112.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch113.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch113.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch20.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch20.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch21.yml) |
| Python 3.11 | | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch20.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch20.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch21.yml) |
| Python 3.9 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch21.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch22.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch22.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch23.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch23.yml) |
| Python 3.10 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch21.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch22.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch22.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch23.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python310_pytorch23.yml) |
| Python 3.11 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch21.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch22.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch22.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch23.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch23.yml) |



| | TensorFlow 2.12 | TensorFlow 2.13 | TensorFlow 2.14 | TensorFlow 2.15 |
Expand Down

0 comments on commit 04bc6eb

Please sign in to comment.