@@ -46,24 +46,12 @@ jobs:
46
46
matrix :
47
47
py_vers : [ "3.9", "3.10", "3.11", "3.12", "3.13", "3.13t" ]
48
48
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"]
59
50
include :
60
51
- device : " rocm"
61
52
rocm_version : " 6.3"
62
53
- device : " cuda"
63
54
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"
67
55
timeout-minutes : 40
68
56
env :
69
57
DOCKER_IMAGE : ${{ matrix.device == 'rocm' && format('pytorch/manylinux2_28-builder:rocm{0}', matrix.rocm_version) || matrix.docker-image }}
@@ -143,14 +131,13 @@ jobs:
143
131
docker exec -t "${container_name}" dnf install clang lld -y
144
132
WITH_CLANG_LDD="--with-clang-ldd"
145
133
fi
146
- # TODO: Remove this conditional block when manylinux2_28 migration for xpu is done
147
134
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"
150
137
else
151
138
docker exec -t "${container_name}" bash -c "${PYTHON_EXECUTABLE} /pytorch/.github/scripts/build_triton_wheel.py --device=$BUILD_DEVICE $RELEASE $WITH_CLANG_LDD"
152
139
fi
153
- if [[ "${{ matrix.device }}" == "cuda" ]]; then
140
+ if [[ ( "${{ matrix.device }}" == "cuda" || "${{ matrix.device }}" == "xpu") ]]; then
154
141
docker exec -t "${container_name}" bash -c "auditwheel repair --plat ${PLATFORM} //artifacts/*.whl"
155
142
else
156
143
docker exec -t "${container_name}" bash -c "mkdir //artifacts/wheelhouse"
0 commit comments