Skip to content

Commit 6c089f5

Browse files
chuanqi129pytorchmergebot
authored andcommitted
ci: move xpu triton build to manylinux 2.28 (pytorch#148195)
Follow PR pytorch#148129 to remove manylinux builds for triton xpu Pull Request resolved: pytorch#148195 Approved by: https://github.com/seemethere
1 parent 165e335 commit 6c089f5

File tree

1 file changed

+4
-17
lines changed

1 file changed

+4
-17
lines changed

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

+4-17
Original file line numberDiff line numberDiff line change
@@ -46,24 +46,12 @@ jobs:
4646
matrix:
4747
py_vers: [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.13t" ]
4848
device: ["cuda", "rocm", "xpu"]
49-
docker-image: ["pytorch/manylinux-builder:cpu", "pytorch/manylinux2_28-builder:cpu"]
50-
exclude:
51-
# TODO: Remove this for rocm when manylinux2_28 migration for xpu is done
52-
- device: "rocm"
53-
docker-image: "pytorch/manylinux-builder:cpu"
54-
# TODO: Remove this for cuda when manylinux2_28 migration for xpu is done
55-
- device: "cuda"
56-
docker-image: "pytorch/manylinux-builder:cpu"
57-
- device: "xpu"
58-
docker-image: "pytorch/manylinux2_28-builder:cpu"
49+
docker-image: ["pytorch/manylinux2_28-builder:cpu"]
5950
include:
6051
- device: "rocm"
6152
rocm_version: "6.3"
6253
- device: "cuda"
6354
rocm_version: ""
64-
# TODO: Remove this for xpu when manylinux2_28 migration for xpu is done
65-
- device: "xpu"
66-
docker-image: "pytorch/manylinux-builder:cpu"
6755
timeout-minutes: 40
6856
env:
6957
DOCKER_IMAGE: ${{ matrix.device == 'rocm' && format('pytorch/manylinux2_28-builder:rocm{0}', matrix.rocm_version) || matrix.docker-image }}
@@ -143,14 +131,13 @@ jobs:
143131
docker exec -t "${container_name}" dnf install clang lld -y
144132
WITH_CLANG_LDD="--with-clang-ldd"
145133
fi
146-
# TODO: Remove this conditional block when manylinux2_28 migration for xpu is done
147134
if [[ "${BUILD_DEVICE}" == xpu ]]; then
148-
docker exec -t "${container_name}" yum install -y devtoolset-11-gcc-c++
149-
docker exec -t "${container_name}" bash -c "source /opt/rh/devtoolset-11/enable && ${PYTHON_EXECUTABLE} /pytorch/.github/scripts/build_triton_wheel.py --device=$BUILD_DEVICE $RELEASE"
135+
docker exec -t "${container_name}" bash -c "dnf install -y gcc-toolset-13-gcc-c++"
136+
docker exec -t "${container_name}" bash -c "source /opt/rh/gcc-toolset-13/enable && ${PYTHON_EXECUTABLE} /pytorch/.github/scripts/build_triton_wheel.py --device=$BUILD_DEVICE $RELEASE"
150137
else
151138
docker exec -t "${container_name}" bash -c "${PYTHON_EXECUTABLE} /pytorch/.github/scripts/build_triton_wheel.py --device=$BUILD_DEVICE $RELEASE $WITH_CLANG_LDD"
152139
fi
153-
if [[ "${{ matrix.device }}" == "cuda" ]]; then
140+
if [[ ("${{ matrix.device }}" == "cuda" || "${{ matrix.device }}" == "xpu") ]]; then
154141
docker exec -t "${container_name}" bash -c "auditwheel repair --plat ${PLATFORM} //artifacts/*.whl"
155142
else
156143
docker exec -t "${container_name}" bash -c "mkdir //artifacts/wheelhouse"

0 commit comments

Comments
 (0)