Skip to content

Commit

Permalink
Support PyTorch 2.1 and add PyTorch tests for python 3.11
Browse files Browse the repository at this point in the history
  • Loading branch information
elad cohen committed Nov 8, 2023
1 parent 7e665f5 commit a0294fd
Show file tree
Hide file tree
Showing 7 changed files with 102 additions and 5 deletions.
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python310_pytorch21.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.10, Pytorch 2.1
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.10"
torch-version: "2.1.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python311_pytorch113.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.11, Pytorch 1.13
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: "1.13.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python311_pytorch20.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.11, Pytorch 2.0
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.0.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python311_pytorch21.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.11, Pytorch 2.1
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.1.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python38_pytorch21.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.8, Pytorch 2.1
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.8"
torch-version: "2.1.*"
16 changes: 16 additions & 0 deletions .github/workflows/run_tests_python39_pytorch21.yml
Original file line number Diff line number Diff line change
@@ -0,0 +1,16 @@
name: Python 3.9, Pytorch 2.1
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.9"
torch-version: "2.1.*"
11 changes: 6 additions & 5 deletions README.md
Original file line number Diff line number Diff line change
Expand Up @@ -51,11 +51,12 @@ In addition, a set of [notebooks](tutorials/notebooks) are provided for an easy

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

| | PyTorch 1.12 | PyTorch 1.13 | PyTorch 2.0 |
|-------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| Python 3.8 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch112.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch112.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch113.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch113.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch20.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch20.yml) |
| Python 3.9 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch112.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch112.yml) | [![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) |
| 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) |
| | PyTorch 1.12 | PyTorch 1.13 | PyTorch 2.0 | PyTorch 2.1 |
|-------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|----------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|--------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------------|
| Python 3.8 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch112.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch112.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch113.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch113.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch20.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch20.yml) | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch21.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python38_pytorch21.yml) |
| Python 3.9 | [![Run Tests](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch112.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python39_pytorch112.yml) | [![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_pytorch113.yml/badge.svg)](https://github.com/sony/model_optimization/actions/workflows/run_tests_python311_pytorch113.yml) | [![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) |


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

0 comments on commit a0294fd

Please sign in to comment.