ci: select 2_28 manylinux builder for new torch+cuda versions (#1000)

torch 2.6 + cu126 and later torch versions uses manylinux_2_28 instead
of manylinux_2014 and we have to select manylinux_2_28 docker images for
new versions.

Ref:
https://dev-discuss.pytorch.org/t/pytorch-linux-wheels-switching-to-new-wheel-build-platform-manylinux-2-28-on-november-12-2024/2581

cc @abcdabcd987 @zhyncs

Author: yzh119

.github/workflows/release_wheel.yml

@@ -46,8 +46,25 @@ jobs:
         with:
           submodules: true
 
+      - name: Set torch and cuda version value
+        id: set_torch_and_cuda_version
+        run: |
+          IFS='.' read -r major minor <<< "${{ matrix.torch }}"
+          version_value=$((major * 100 + minor))
+          echo "TORCH_VERSION=$version_value" >> $GITHUB_OUTPUT
+          IFS='.' read -r major minor <<< "${{ matrix.cuda }}"
+          version_value=$((major * 100 + minor))
+          echo "CUDA_VERSION=$version_value" >> $GITHUB_OUTPUT
+
       - name: Build wheel
+        env:
+          TORCH_VERSION: ${{ steps.set_torch_and_cuda_version.outputs.TORCH_VERSION }}
+          CUDA_VERSION: ${{ steps.set_torch_and_cuda_version.outputs.CUDA_VERSION }}
+          BUILDER_IMAGE: ${{ ((steps.set_torch_and_cuda_version.outputs.TORCH_VERSION == 206 && steps.set_torch_and_cuda_version.outputs.CUDA_VERSION >= 1206) || steps.set_torch_and_cuda_version.outputs.TORCH_VERSION > 206) && 'pytorch/manylinux2_28-builder:cuda' || 'pytorch/manylinux-builder:cuda' }}${{ matrix.cuda }}
         run: |
+          echo "TORCH_VERSION: $TORCH_VERSION"
+          echo "CUDA_VERSION: $CUDA_VERSION"
+          echo "BUILDER_IMAGE: $BUILDER_IMAGE"
           chown -R $CI_UID:$CI_GID "$GITHUB_WORKSPACE"
           docker run --rm -t \
               -v "$CI_RUNNER_CACHE_DIR":/ci-cache \
@@ -59,7 +76,7 @@ jobs:
               -e TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST" \
               -e MAX_JOBS=128 \
               --user $CI_UID:$CI_GID \
-              pytorch/manylinux-builder:cuda${{ matrix.cuda }} \
+              $BUILDER_IMAGE \
               bash /app/scripts/run-ci-build-wheel.sh
         timeout-minutes: 120
       - run: du -h dist/*

.github/workflows/release_wheel_sglang.yml

@@ -31,8 +31,25 @@ jobs:
         with:
           submodules: true
 
+      - name: Set torch and cuda version value
+        id: set_torch_and_cuda_version
+        run: |
+          IFS='.' read -r major minor <<< "${{ matrix.torch }}"
+          version_value=$((major * 100 + minor))
+          echo "TORCH_VERSION=$version_value" >> $GITHUB_OUTPUT
+          IFS='.' read -r major minor <<< "${{ matrix.cuda }}"
+          version_value=$((major * 100 + minor))
+          echo "CUDA_VERSION=$version_value" >> $GITHUB_OUTPUT
+
       - name: Build wheel
+        env:
+          TORCH_VERSION: ${{ steps.set_torch_and_cuda_version.outputs.TORCH_VERSION }}
+          CUDA_VERSION: ${{ steps.set_torch_and_cuda_version.outputs.CUDA_VERSION }}
+          BUILDER_IMAGE: ${{ ((steps.set_torch_and_cuda_version.outputs.TORCH_VERSION == 206 && steps.set_torch_and_cuda_version.outputs.CUDA_VERSION >= 1206) || steps.set_torch_and_cuda_version.outputs.TORCH_VERSION > 206) && 'pytorch/manylinux2_28-builder:cuda' || 'pytorch/manylinux-builder:cuda' }}${{ matrix.cuda }}
         run: |
+          echo "TORCH_VERSION: $TORCH_VERSION"
+          echo "CUDA_VERSION: $CUDA_VERSION"
+          echo "BUILDER_IMAGE: $BUILDER_IMAGE"
           chown -R $CI_UID:$CI_GID "$GITHUB_WORKSPACE"
           docker run --rm -t \
               -v "$CI_RUNNER_CACHE_DIR":/ci-cache \
@@ -45,7 +62,7 @@ jobs:
               -e TORCH_CUDA_ARCH_LIST="$TORCH_CUDA_ARCH_LIST" \
               -e MAX_JOBS=128 \
               --user $CI_UID:$CI_GID \
-              pytorch/manylinux-builder:cuda${{ matrix.cuda }} \
+              $BUILDER_IMAGE \
               bash /app/scripts/run-ci-build-wheel.sh
         timeout-minutes: 120
       - run: du -h dist/*