Skip to content

Commit 8fcc0ec

Browse files
authored
Improve INC CI test torch version (#1027)
* improve inc and ipex test ci torch version Signed-off-by: changwa1 <[email protected]> * only improve test_inc.yml Signed-off-by: changwa1 <[email protected]> * recover test_ipex.yml Signed-off-by: changwa1 <[email protected]> --------- Signed-off-by: changwa1 <[email protected]>
1 parent a9c4772 commit 8fcc0ec

File tree

2 files changed

+2
-6
lines changed

2 files changed

+2
-6
lines changed

.github/workflows/test_inc.yml

Lines changed: 1 addition & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
strategy:
1919
fail-fast: false
2020
matrix:
21-
torch-version: ["2.2.0", "2.3.*", "2.4.*"]
21+
torch-version: ["2.4.*", "2.5.0"]
2222

2323
runs-on: ubuntu-22.04
2424

@@ -37,10 +37,6 @@ jobs:
3737
pip install torch==${{ matrix.torch-version }} torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu
3838
pip install .[neural-compressor,ipex,diffusers,peft,tests] transformers[testing] intel-extension-for-pytorch==${{ matrix.torch-version }}
3939
40-
- if: ${{ matrix.torch-version == '2.2.0' }}
41-
name: Downgrade Numpy
42-
run: pip install numpy==1.*
43-
4440
- name: Assert versions
4541
run: |
4642
python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))"

.github/workflows/test_ipex.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -50,4 +50,4 @@ jobs:
5050
5151
- name: Test with Pytest
5252
run: |
53-
pytest tests/ipex
53+
pytest tests/ipex

0 commit comments

Comments
 (0)