From 8fcc0ec3ee3da65a4e20363949493773d6b3851c Mon Sep 17 00:00:00 2001 From: "Wang, Chang" Date: Mon, 25 Nov 2024 17:19:57 +0800 Subject: [PATCH] Improve INC CI test torch version (#1027) * improve inc and ipex test ci torch version Signed-off-by: changwa1 * only improve test_inc.yml Signed-off-by: changwa1 * recover test_ipex.yml Signed-off-by: changwa1 --------- Signed-off-by: changwa1 --- .github/workflows/test_inc.yml | 6 +----- .github/workflows/test_ipex.yml | 2 +- 2 files changed, 2 insertions(+), 6 deletions(-) diff --git a/.github/workflows/test_inc.yml b/.github/workflows/test_inc.yml index c1a75a6e3..b31f2055f 100644 --- a/.github/workflows/test_inc.yml +++ b/.github/workflows/test_inc.yml @@ -18,7 +18,7 @@ jobs: strategy: fail-fast: false matrix: - torch-version: ["2.2.0", "2.3.*", "2.4.*"] + torch-version: ["2.4.*", "2.5.0"] runs-on: ubuntu-22.04 @@ -37,10 +37,6 @@ jobs: pip install torch==${{ matrix.torch-version }} torchaudio torchvision --index-url https://download.pytorch.org/whl/cpu pip install .[neural-compressor,ipex,diffusers,peft,tests] transformers[testing] intel-extension-for-pytorch==${{ matrix.torch-version }} - - if: ${{ matrix.torch-version == '2.2.0' }} - name: Downgrade Numpy - run: pip install numpy==1.* - - name: Assert versions run: | python -c "import torch; print(torch.__version__); assert torch.__version__.startswith('${{ matrix.torch-version }}'.replace('.*', ''))" diff --git a/.github/workflows/test_ipex.yml b/.github/workflows/test_ipex.yml index a14fc7337..59a316152 100644 --- a/.github/workflows/test_ipex.yml +++ b/.github/workflows/test_ipex.yml @@ -50,4 +50,4 @@ jobs: - name: Test with Pytest run: | - pytest tests/ipex + pytest tests/ipex \ No newline at end of file