reuse cibw workflow in maint

Author: oraluben

maint/scripts/docker_build_all.sh

@@ -1,70 +1,5 @@
 #!/usr/bin/env bash
 set -euxo pipefail
 
-IMAGE="tilelang-builder:manylinux"
-
-HOST_UNAME=$(uname -m)
-case "$HOST_UNAME" in
-  x86_64) TARGETARCH=amd64 ;;
-  aarch64|arm64) TARGETARCH=arm64 ;;
-  *) echo "Unsupported architecture: $HOST_UNAME" >&2; exit 1 ;;
-esac
-
-if docker buildx version >/dev/null 2>&1; then
-  if docker info >/dev/null 2>&1; then
-    docker run --rm --privileged tonistiigi/binfmt --install amd64,arm64 >/dev/null 2>&1 || true
-  fi
-
-  if ! docker buildx inspect multi >/dev/null 2>&1; then
-    docker buildx create --name multi --driver docker-container --use >/dev/null 2>&1 || true
-  else
-    docker buildx use multi >/dev/null 2>&1 || true
-  fi
-  docker buildx inspect --bootstrap >/dev/null 2>&1 || true
-
-  for ARCH in amd64 arm64; do
-    TAG_PLATFORM="linux/${ARCH}"
-    TAG_IMAGE="${IMAGE}-${ARCH}"
-
-    docker buildx build \
-      --platform "${TAG_PLATFORM}" \
-      --build-arg TARGETARCH="${ARCH}" \
-      -f "$(dirname "${BASH_SOURCE[0]}")/pypi.manylinux.Dockerfile" \
-      -t "${TAG_IMAGE}" \
-      --load \
-      .
-
-    script="sh maint/scripts/local_distribution.sh"
-    docker run --rm \
-      --platform "${TAG_PLATFORM}" \
-      -v "$(pwd):/tilelang" \
-      "${TAG_IMAGE}" \
-      /bin/bash -lc "$script"
-
-    if [ -d dist ]; then
-      mv -f dist "dist-local-${ARCH}"
-    fi
-  done
-
-else
-  echo "docker buildx not found; building only host arch: ${TARGETARCH}" >&2
-  TAG_IMAGE="${IMAGE}-${TARGETARCH}"
-  TAG_PLATFORM="linux/${TARGETARCH}"
-
-  docker build \
-    --build-arg TARGETARCH="$TARGETARCH" \
-    -f "$(dirname "${BASH_SOURCE[0]}")/pypi.manylinux.Dockerfile" \
-    -t "${TAG_IMAGE}" \
-    .
-
-  script="sh maint/scripts/local_distribution.sh"
-  docker run --rm \
-    --platform "${TAG_PLATFORM}" \
-    -v "$(pwd):/tilelang" \
-    "${TAG_IMAGE}" \
-    /bin/bash -lc "$script"
-
-  if [ -d dist ]; then
-    mv -f dist "dist-local-${TARGETARCH}"
-  fi
-fi
+# Build for local architecture
+CIBW_BUILD='cp38-*' cibuildwheel .

maint/scripts/docker_local_distribute.sh

@@ -1,15 +1,6 @@
 #!/usr/bin/env bash
 set -euxo pipefail
 
-IMAGE="tilelang-builder:manylinux"
-
-HOST_UNAME=$(uname -m)
-case "$HOST_UNAME" in
-  x86_64) TARGETARCH=amd64 ;;
-  aarch64|arm64) TARGETARCH=arm64 ;;
-  *) echo "Unsupported architecture: $HOST_UNAME" >&2; exit 1 ;;
-esac
-
 if docker buildx version >/dev/null 2>&1; then
   if docker info >/dev/null 2>&1; then
     docker run --rm --privileged tonistiigi/binfmt --install amd64,arm64 >/dev/null 2>&1 || true
@@ -21,50 +12,9 @@ if docker buildx version >/dev/null 2>&1; then
     docker buildx use multi >/dev/null 2>&1 || true
   fi
   docker buildx inspect --bootstrap >/dev/null 2>&1 || true
-
-  for ARCH in amd64 arm64; do
-    TAG_PLATFORM="linux/${ARCH}"
-    TAG_IMAGE="${IMAGE}-${ARCH}"
-
-    docker buildx build \
-      --platform "${TAG_PLATFORM}" \
-      --build-arg TARGETARCH="${ARCH}" \
-      -f "$(dirname "${BASH_SOURCE[0]}")/pypi.manylinux.Dockerfile" \
-      -t "${TAG_IMAGE}" \
-      --load \
-      .
-
-    script="sh maint/scripts/pypi_distribution.sh"
-    docker run --rm \
-      --platform "${TAG_PLATFORM}" \
-      -v "$(pwd):/tilelang" \
-      "${TAG_IMAGE}" \
-      /bin/bash -lc "$script"
-
-    if [ -d dist ]; then
-      mv -f dist "dist-pypi-${ARCH}"
-    fi
   done
 
-else
-  echo "docker buildx not found; building only host arch: ${TARGETARCH}" >&2
-  TAG_IMAGE="${IMAGE}-${TARGETARCH}"
-  TAG_PLATFORM="linux/${TARGETARCH}"
-
-  docker build \
-    --build-arg TARGETARCH="$TARGETARCH" \
-    -f "$(dirname "${BASH_SOURCE[0]}")/pypi.manylinux.Dockerfile" \
-    -t "${TAG_IMAGE}" \
-    .
-
-  script="sh maint/scripts/pypi_distribution.sh"
-  docker run --rm \
-    --platform "${TAG_PLATFORM}" \
-    -v "$(pwd):/tilelang" \
-    "${TAG_IMAGE}" \
-    /bin/bash -lc "$script"
-
-  if [ -d dist ]; then
-    mv -f dist "dist-pypi-${TARGETARCH}"
-  fi
+  export CIBW_ARCHS='x86_64 aarch64'
 fi
+
+NO_VERSION_LABEL=ON CIBW_BUILD='cp38-*' cibuildwheel .

maint/scripts/docker_pypi_distribute.sh

@@ -1,14 +1,18 @@
-ARG TARGETARCH
-FROM pytorch/manylinux2_28-builder:cuda12.1 AS builder_amd64
-ENV CUDA_VERSION=12.1 \
-    AUDITWHEEL_PLAT=manylinux_2_28_x86_64
-RUN pip3 install uv
+FROM quay.io/pypa/manylinux2014_x86_64 AS builder_amd64
+
+RUN yum-config-manager --add-repo https://developer.download.nvidia.cn/compute/cuda/repos/rhel7/x86_64/cuda-rhel7.repo
+
+ARG CUDA_VERSION=12.1
+ENV CUDA_VERSION=${CUDA_VERSION}
+
+FROM quay.io/pypa/manylinux_2_28_aarch64 AS builder_arm64
 
-FROM pytorch/manylinuxaarch64-builder:cuda12.8 AS builder_arm64
-ENV CUDA_VERSION=12.8 \
-    AUDITWHEEL_PLAT=manylinux_2_28_aarch64
-RUN /opt/python/cp312-cp312/bin/pip install uv
+RUN dnf config-manager --add-repo https://developer.download.nvidia.com/compute/cuda/repos/rhel8/sbsa/cuda-rhel8.repo
 
+ARG CUDA_VERSION=12.8
+ENV CUDA_VERSION=${CUDA_VERSION}
+
+ARG TARGETARCH
 FROM builder_${TARGETARCH}
 
 ENV DEBIAN_FRONTEND=noninteractive \
@@ -19,12 +23,7 @@ ENV PATH="/usr/local/cuda/bin:${PATH}"
 ENV LD_LIBRARY_PATH="/usr/local/cuda/lib64:${LD_LIBRARY_PATH}"
 
 RUN set -eux; \
-    uv venv -p 3.12 --seed /venv; \
+    pipx install cibuildwheel; \
     git config --global --add safe.directory '/tilelang'
 
-ENV PATH="/venv/bin:$PATH" \
-    VIRTUAL_ENV=/venv
-
-RUN uv pip install build wheel
-
 WORKDIR /tilelang