5959 if : ${{ ! endsWith(inputs.pytorch, '_wheel') }}
6060 runs-on : ${{ needs.runner.outputs.runner_id }}
6161 container :
62- image : ' pytorch/manylinux2_28-builder:xpu-2.9 '
62+ image : ' intelgpu/ubuntu-22.04-lts2:2523.31 '
6363 volumes :
6464 - ${{ github.workspace }}:${{ github.workspace }}
6565 env :
@@ -72,21 +72,30 @@ jobs:
7272 steps :
7373 - name : Install gh-cli
7474 run : |
75+ rm -rf ./*.whl ./*.log
7576 cat /etc/os-release
7677 hostname && id
7778 # install gh
78- dnf install -y 'dnf-command(config-manager)'
79- dnf config-manager --add-repo https://cli.github.com/packages/rpm/gh-cli.repo
80- dnf install -y gh --repo gh-cli
81- gh --version
79+ sudo apt update
80+ sudo apt install -y gpg-agent wget curl cmake git unzip zip libgl1 zlib1g-dev numactl \
81+ libglib2.0-dev rsync jq gcc-11 g++-11 python3-dev python3-venv gh
8282 - name : Setup python-${{ inputs.python }}
8383 run : |
8484 rm -rf /tmp/xpu-tool/myvenv
85- local_python=$(echo ${{ inputs.python }} |awk -F. '{printf("cp%s%s-cp%s%s", $1, $2, $1, $2)}')
86- /opt/python/${local_python}/bin/python -m venv /tmp/xpu-tool/myvenv
85+ curl -LsSf https://astral.sh/uv/install.sh | sh
86+ source $HOME/.local/bin/env
87+ uv venv /tmp/xpu-tool/myvenv --python 3.10 --clear
88+ source /tmp/xpu-tool/myvenv/bin/activate
8789 which python && python -V
8890 which pip && pip list
89- pip install -U pip wheel setuptools
91+ uv pip install -U pip wheel setuptools
92+ - name : Install oneapi
93+ run : |
94+ rm -rf /opt/intel/oneapi
95+ wget https://registrationcenter-download.intel.com/akdlm/IRC_NAS/aa5447b5-3644-43c8-8ec4-72d53f6ecc19/intel-deep-learning-essentials-2025.3.0.338_offline.sh
96+ sudo bash intel-deep-learning-essentials-2025.3.0.338_offline.sh -a -s --eula accept
97+ source /opt/intel/oneapi/setvars.sh
98+ icpx --version
9099 - name : Checkout torch-xpu-ops
91100 uses : actions/checkout@v4
92101 with :
@@ -113,8 +122,6 @@ jobs:
113122 TORCH_XPU_OPS_REPO="https://github.com/intel/torch-xpu-ops.git"
114123 TORCH_XPU_OPS_COMMIT="${{ inputs.torch_xpu_ops }}"
115124 fi
116- # gcc 11
117- source /opt/rh/gcc-toolset-11/enable
118125 source ${{ github.workspace }}/torch-xpu-ops/.github/scripts/env.sh
119126 ${{ github.workspace }}/torch-xpu-ops/.github/scripts/build.sh \
120127 --WORKSPACE="${{ github.workspace }}" \
@@ -129,8 +136,6 @@ jobs:
129136 fi
130137 - name : Build Torchvision and Torchaudio
131138 run : |
132- # gcc 11
133- source /opt/rh/gcc-toolset-11/enable
134139 cd ./pytorch
135140 TORCHVISION_COMMIT_ID="$(cat .github/ci_commit_pins/vision.txt)"
136141 TORCHAUDIO_COMMIT_ID="$(cat .github/ci_commit_pins/audio.txt)"
@@ -177,9 +182,6 @@ jobs:
177182 curl -sSL https://raw.githubusercontent.com/intel/intel-xpu-backend-for-triton/${TRITON_COMMIT_ID}/python/triton/__init__.py 2>&1 |\
178183 grep '__version__' |head -n 1 |awk -F "'" '{print $2}'
179184 )"
180- # gcc 13
181- dnf install -y gcc-toolset-13-gcc-c++ zlib-devel
182- source /opt/rh/gcc-toolset-13/enable
183185 pip install cmake ninja pybind11
184186 python .github/scripts/build_triton_wheel.py --device xpu --commit-hash ${TRITON_COMMIT_ID} --triton-version ${TRITON_VERSION_NAME} \
185187 2>&1 |tee ${{ github.workspace }}/build_triton_${TRITON_COMMIT_ID}.log
0 commit comments