[CI/Build] Pin Cython below 3.3 for arm64 tilelang sdist (#53358)
Signed-off-by: Kevin Luu <51931015+khluu@users.noreply.github.com> Co-authored-by: OpenAI Codex <codex@openai.com>
Kevin H. Luu committed
Aug 23, 2026 at 01:09 UTC
e25c586b9030a10702d78856b43ccae9481cc28c
1 file changed
+7
docker/Dockerfile
+7
@@ -209,6 +209,13 @@ RUN --mount=type=cache,target=/opt/uv/cache \
209
sed -i 's/^nvidia-cutlass-dsl\[cu13\]/nvidia-cutlass-dsl/' requirements/cuda.txt; \
210
sed -i 's/^humming-kernels\[cu13\]/humming-kernels[cu12]/' requirements/cuda.txt; \
211
fi \
212
+ && if [ "$(uname -m)" = "aarch64" ]; then \
213
+ # The manylinux 2_28 (glibc 2.28) aarch64 builder cannot use
214
+ # tilelang's manylinux_2_34 wheel, so tilelang builds from sdist.
215
+ # Cython 3.3 rejects tilelang's hardcoded Py_LIMITED_API=3.8.
216
+ printf 'cython<3.3\n' > /tmp/build-constraints.txt \
217
+ && export UV_BUILD_CONSTRAINT=/tmp/build-constraints.txt; \
218
+ fi \
219
&& if [ "${PYTORCH_NIGHTLY}" = "1" ]; then \
220
echo "Installing torch nightly..." \
221
&& uv pip install --python /opt/venv/bin/python3 torch torchaudio torchvision --pre \