[BE] Fix triton windows build (pytorch#150547)

[BE] Fix triton windows build (pytorch#150512)

Fixes pytorch#150480
Pull Request resolved: pytorch#150512
Approved by: https://github.com/atalman

Co-authored-by: Andrey Talman <[email protected]>
(cherry picked from commit 8102272)

Co-authored-by: Wang, Chuanqi <[email protected]>

Author: pytorchbot

.github/scripts/windows/build_triton.bat

@@ -9,7 +9,8 @@ if "%PY_VERS%" == "3.13t" (
 ) else (
     call conda create -n %PYTHON_PREFIX% -y -c=conda-forge python=%PY_VERS%
 )
-call conda run -n %PYTHON_PREFIX% pip install wheel pybind11 certifi cython cmake setuptools==72.1.0 ninja
+:: Fix cmake version for issue https://github.com/pytorch/pytorch/issues/150480
+call conda run -n %PYTHON_PREFIX% pip install wheel pybind11 certifi cython cmake==3.31.6 setuptools==72.1.0 ninja
 
 dir "%VC_INSTALL_PATH%"
 

.github/workflows/build-triton-wheel.yml

@@ -12,13 +12,17 @@ on:
       - .github/workflows/build-triton-wheel.yml
       - .github/scripts/build_triton_wheel.py
       - .github/ci_commit_pins/triton.txt
+      - .github/scripts/windows/install_vs2022.ps1
+      - .github/scripts/windows/build_triton.bat
       - .ci/docker/ci_commit_pins/triton.txt
       - .ci/docker/ci_commit_pins/triton-xpu.txt
   pull_request:
     paths:
       - .github/workflows/build-triton-wheel.yml
       - .github/scripts/build_triton_wheel.py
       - .github/ci_commit_pins/triton.txt
+      - .github/scripts/windows/install_vs2022.ps1
+      - .github/scripts/windows/build_triton.bat
       - .ci/docker/ci_commit_pins/triton.txt
       - .ci/docker/ci_commit_pins/triton-xpu.txt
 
@@ -243,7 +247,6 @@ jobs:
           .github/scripts/windows/build_triton.bat
           mkdir -p "${RUNNER_TEMP}/artifacts/"
           mv ./*.whl "${RUNNER_TEMP}/artifacts/"
-
       - uses: actions/[email protected]
         with:
           name: pytorch-triton-wheel-${{ matrix.py_vers }}-${{ matrix.device }}