From bf4a36135408035e47d00266a19a9de1bbe7d69c Mon Sep 17 00:00:00 2001
From: Pavel Chekin <pavel.chekin@intel.com>
Date: Fri, 2 Aug 2024 12:10:16 -0700
Subject: [PATCH] Install torchdata<0.8.0 when using IPEX

---
 .github/workflows/e2e-reusable.yml   | 6 ++++++
 .github/workflows/nightly-wheels.yml | 3 +++
 2 files changed, 9 insertions(+)

diff --git a/.github/workflows/e2e-reusable.yml b/.github/workflows/e2e-reusable.yml
index f475adfd73..ea33bff833 100644
--- a/.github/workflows/e2e-reusable.yml
+++ b/.github/workflows/e2e-reusable.yml
@@ -195,6 +195,12 @@ jobs:
           ref: ${{ env.TORCHVISION_COMMIT_ID }}
           extra-cache-key: ${{ env.PYTORCH_VERSION }}
 
+      # FIXME: old PyTorch does not work with torchdata=0.8.0
+      - name: Install torchdata package
+        if: ${{ inputs.suite == 'torchbench' && inputs.install_ipex }}
+        run: |
+          pip install 'torchdata<0.8.0'
+
       - name: Install torchtext package
         if: ${{ inputs.suite == 'torchbench' }}
         uses: ./.github/actions/install-dependency
diff --git a/.github/workflows/nightly-wheels.yml b/.github/workflows/nightly-wheels.yml
index 5b4e7ec9cb..8bf224cc7b 100644
--- a/.github/workflows/nightly-wheels.yml
+++ b/.github/workflows/nightly-wheels.yml
@@ -24,6 +24,7 @@ jobs:
           - "3.10"
           - "3.11"
           - "3.12"
+      fail-fast: false
     defaults:
       run:
         shell: bash -noprofile --norc -eo pipefail -c "source /home/runner/intel/oneapi/setvars.sh > /dev/null; source {0}"
@@ -131,6 +132,8 @@ jobs:
       - name: Install torchtext
         run: |
           pip install text/dist/*.whl
+          # FIXME: old PyTorch does not work with torchdata=0.8.0
+          pip install 'torchdata<0.8.0'
           python -c "import torchtext; print(torchtext.__version__)"
 
       - name: Save torchtext to a cache