Update pytorch nightly to 2.9.0.dev20250820 (#4301)

* Builds on top of @sjarus's commit for updating to 2.9.0.dev20250820
* Updated the CI build to initialize python 3.11 version correctly.

Author: sahas3

.github/actions/setup-build/action.yml

@@ -26,7 +26,6 @@ runs:
 
   steps:
     - name: Set up Python
-      if: ${{ runner.arch == 'X64' }}
       uses: actions/setup-python@0b93645e9fea7318ecaed2b359559ac225c90a2b # v5.3.0
       with:
         python-version: '3.11'

.github/workflows/ci.yml

@@ -22,6 +22,7 @@ jobs:
       fail-fast: true
       matrix:
         torch-version: [nightly, stable]
+
     name: Build and Test (Linux, torch-${{ matrix.torch-version }}, assertions)
     runs-on: torch-mlir-cpubuilder-linux-x86-64-scale
     env:
@@ -40,25 +41,25 @@ jobs:
           restore-keys: |
             build-test-cpp-asserts-manylinux-${{ matrix.torch-version }}-v2-
 
-      - name: "Setting up Python"
-        run: |
-          sudo apt update
-          sudo apt install software-properties-common -y
-          sudo add-apt-repository ppa:deadsnakes/ppa -y
-          sudo apt install python3.11 python3-pip -y
-          sudo apt-get install python3.11-dev python3.11-venv build-essential -y
-
-      - name: Install python deps (torch-${{ matrix.torch-version }})
-        run: |
-          export cache_dir="${{ env.CACHE_DIR }}"
-          bash build_tools/ci/install_python_deps.sh ${{ matrix.torch-version }}
-
       - name: ccache
         uses: hendrikmuhs/ccache-action@63069e3931dedbf3b63792097479563182fe70d1 # v1.2.18
         with:
           key: ${{ github.job }}-${{ matrix.torch-version }}
           save: ${{ needs.setup.outputs.write-caches == 1 }}
 
+      - name: Setup workspace and python
+        uses: ./.github/actions/setup-build
+
+      - name: Display Python version and path
+        run: |
+          python -c "import sys; print(sys.version)"
+          which python
+
+      - name: Install python deps (torch-${{ matrix.torch-version }})
+        run: |
+          cd $GITHUB_WORKSPACE
+          ./build_tools/ci/install_python_deps.sh ${{ matrix.torch-version }}
+
       - name: Build project
         run: |
           export cache_dir="${{ env.CACHE_DIR }}"

lib/Dialect/Torch/Transforms/AbstractInterpLibrary.cpp

@@ -6172,7 +6172,7 @@ StringRef mlir::torch::Torch::getAbstractInterpLibrary() {
 "    %true = torch.constant.bool true\n"
 "    %none = torch.constant.none\n"
 "    %str_0 = torch.constant.str \"AssertionError: \"\n"
-"    %str_1 = torch.constant.str \"The dims of tensor b ({}) must be less than or equal tothe dims of tensor a ({}) \"\n"
+"    %str_1 = torch.constant.str \"The dims of tensor b ({}) must be less than or equal to the dims of tensor a ({}) \"\n"
 "    %int0 = torch.constant.int 0\n"
 "    %int1 = torch.constant.int 1\n"
 "    %0 = torch.aten.len.t %arg0 : !torch.list<int> -> !torch.int\n"

projects/pt1/e2e_testing/xfail_sets.py

@@ -3427,7 +3427,9 @@
         "ElementwiseBitwiseAndStaticShapeModule_basic",
     }
 
-if torch_version_for_comparison() >= version.parse("2.5.0.dev"):
+if torch_version_for_comparison() >= version.parse(
+    "2.5.0.dev"
+) and torch_version_for_comparison() < version.parse("2.9.0.dev"):
     ONNX_XFAIL_SET = ONNX_XFAIL_SET | {
         # ERROR: value (Tensor with shape=[2, 3, 8, 20], dtype=torch.float32, min=+nan, max=+nan, mean=+nan) is not close to golden value (Tensor with shape=[2, 3, 8, 20], dtype=torch.float32, min=-2.394, max=+2.454, mean=-0.02828)
         "ScaledDotProductAttentionBoolMaskModule_basic",

pytorch-hash.txt

@@ -1 +1 @@
-382411c3d8fe3b2eed522fa97c24e8caa9530bec
+7956a1d1d0dc7cdaaaa42d0863eebb1b1e75eb65

pytorch-requirements.txt

@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/nightly/cpu/torch/
 --pre
-torch==2.8.0.dev20250622
+torch==2.9.0.dev20250820

torchvision-requirements.txt

@@ -1,3 +1,3 @@
 -f https://download.pytorch.org/whl/nightly/cpu/torchvision/
 --pre
-torchvision==0.23.0.dev20250622
+torchvision==0.24.0.dev20250820