Temporary adding checkout branch

checkout branch

Author: sshonTT

.github/workflows/_build_torch_xla_3.11.yml

@@ -27,6 +27,7 @@ jobs:
       - name: "Build Torch/XLA wheel"
         shell: bash
         id: build_wheels
+        timeout-minutes: 540 # 9 hours
         run: |
           cmake --version
           apt-get update && apt-get install -y curl git build-essential
@@ -50,9 +51,12 @@ jobs:
           cd pytorch/
           git clone --recursive https://github.com/tenstorrent/pytorch-xla.git xla
 
+
           (
             # Build PyTorch
             # From https://docs.pytorch.org/FBGEMM/fbgemm/development/BuildInstructions.html, section "Build Issues with GCC 12+"
+            export CFLAGS="${CFLAGS:+$CFLAGS }-w"
+            export CXXFLAGS="${CXXFLAGS:+$CXXFLAGS }-w"
             export CFLAGS+=" -Wno-error=maybe-uninitialized -Wno-error=uninitialized -Wno-error=restrict"
             export CXXFLAGS+=" -Wno-error=maybe-uninitialized -Wno-error=uninitialized -Wno-error=restrict"
             # copy pre-built wheels from cache
@@ -62,6 +66,7 @@ jobs:
 
           # Build PyTorch/XLA
           cd xla/
+          git checkout sshon/rebase-to-upstream
           python3.11 setup.py bdist_wheel
 
           # Collect wheels