Use latest cmake and revert #2080.

intel · Aug 31, 2024 · 4d44cfe · 4d44cfe
1 parent cb994ee
commit 4d44cfe
Showing 1 changed file with 38 additions and 1 deletion.
diff --git a/.github/workflows/nightly-wheels.yml b/.github/workflows/nightly-wheels.yml
@@ -40,13 +40,20 @@ jobs:
 
       - name: Install Python build dependencies
         run: |
-          pip install wheel
+          pip install wheel cmake
 
       - name: Setup PyTorch
         uses: ./.github/actions/setup-pytorch
         with:
           repository: pytorch/pytorch
 
+      - name: Identify pinned versions
+        run: |
+          cd pytorch
+          echo "TORCHVISION_COMMIT_ID=$(<.github/ci_commit_pins/vision.txt)" >> $GITHUB_ENV
+          echo "TORCHTEXT_COMMIT_ID=$(<.github/ci_commit_pins/text.txt)" >> $GITHUB_ENV
+          echo "TORCHAUDIO_COMMIT_ID=$(<.github/ci_commit_pins/audio.txt)" >> $GITHUB_ENV
+
       - name: Identify Triton commit id
         run: |
           echo "TRITON_COMMIT_ID=$(git rev-parse HEAD)" >> $GITHUB_ENV
@@ -80,11 +87,38 @@ jobs:
           path: ${{ steps.triton-cache.outputs.path }}
           dest: ${{ steps.triton-cache.outputs.dest }}
 
+      - name: Install torchvision package
+        uses: ./.github/actions/install-dependency
+        with:
+          package: torchvision
+          repository: pytorch/vision
+          ref: ${{ env.TORCHVISION_COMMIT_ID }}
+          extra-cache-key: ${{ env.PYTORCH_VERSION }}
+
+      - name: Install torchtext package
+        uses: ./.github/actions/install-dependency
+        with:
+          package: torchtext
+          repository: pytorch/text
+          ref: ${{ env.TORCHTEXT_COMMIT_ID }}
+          extra-cache-key: ${{ env.PYTORCH_VERSION }}
+
+      - name: Install torchaudio package
+        uses: ./.github/actions/install-dependency
+        with:
+          package: torchaudio
+          repository: pytorch/audio
+          ref: ${{ env.TORCHAUDIO_COMMIT_ID }}
+          extra-cache-key: ${{ env.PYTORCH_VERSION }}
+
       - name: Prepare wheels for upload
         run: |
           mkdir -p wheels
           cp -L pytorch/dist/*.whl wheels/
           cp -L python/dist/*.whl wheels/
+          cp -L torchvision*/dist/*.whl wheels/
+          cp -L torchtext*/dist/*.whl wheels/
+          cp -L torchaudio*/dist/*.whl wheels/
 
       - name: Report environment details
         run: |
@@ -103,6 +137,9 @@ jobs:
           LLVM_COMMIT_ID=$LLVM_COMMIT_ID
           TRITON_REPO=intel/intel-xpu-backend-for-triton
           TRITON_COMMIT_ID=$TRITON_COMMIT_ID
+          TORCHVISION_COMMIT_ID=$TORCHVISION_COMMIT_ID
+          TORCHTEXT_COMMIT_ID=$TORCHTEXT_COMMIT_ID
+          TORCHAUDIO_COMMIT_ID=$TORCHAUDIO_COMMIT_ID
           EOF
 
       - name: Upload PyTorch wheels to artifacts