Pin torchdata version (#2986)

https://github.com/pytorch/data/blob/e316c5ca1ab2a4f69dd6d48e8fc9c6f8d0c7c468/README.md?plain=1#L6-L15.
intel · Dec 11, 2024 · 0e3a9c0 · 0e3a9c0
1 parent 718a7fd
commit 0e3a9c0
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/.github/workflows/e2e-reusable.yml b/.github/workflows/e2e-reusable.yml
@@ -87,6 +87,11 @@ jobs:
           # cmake 3.22.1 does not work with the recent torchaudio: https://github.com/intel/intel-xpu-backend-for-triton/issues/2079
           pip install wheel cmake
 
+      # https://github.com/pytorch/data/blob/e316c5ca1ab2a4f69dd6d48e8fc9c6f8d0c7c468/README.md?plain=1#L6-L15
+      - name: Install pinned torchdata
+        run: |
+          pip install torchdata==0.9.0
+
       - name: Setup PyTorch
         uses: ./.github/actions/setup-pytorch
         with:

diff --git a/.github/workflows/nightly-wheels.yml b/.github/workflows/nightly-wheels.yml
@@ -49,6 +49,11 @@ jobs:
           # cmake 3.22.1 does not work with the recent torchaudio: https://github.com/intel/intel-xpu-backend-for-triton/issues/2079
           pip install wheel cmake
 
+      # https://github.com/pytorch/data/blob/e316c5ca1ab2a4f69dd6d48e8fc9c6f8d0c7c468/README.md?plain=1#L6-L15
+      - name: Install pinned torchdata
+        run: |
+          pip install torchdata==0.9.0
+
       - name: Setup PyTorch
         uses: ./.github/actions/setup-pytorch
         with: