[CI][Packaging] Cibuildwheel-based multi-platform wheel publishing

.github/actions/build-cuda/action.yml

@@ -0,0 +1,117 @@
+# Licensed to the Apache Software Foundation (ASF) under one
+# or more contributor license agreements.  See the NOTICE file
+# distributed with this work for additional information
+# regarding copyright ownership.  The ASF licenses this file
+# to you under the Apache License, Version 2.0 (the
+# "License"); you may not use this file except in compliance
+# with the License.  You may obtain a copy of the License at
+#
+#   http://www.apache.org/licenses/LICENSE-2.0
+#
+# Unless required by applicable law or agreed to in writing,
+# software distributed under the License is distributed on an
+# "AS IS" BASIS, WITHOUT WARRANTIES OR CONDITIONS OF ANY
+# KIND, either express or implied.  See the License for the
+# specific language governing permissions and limitations
+# under the License.
+
+name: Build CUDA Runtime
+description: Build libtvm_runtime_cuda for the TVM wheel packaging flow.
+
+inputs:
+  arch:
+    description: "Target Linux architecture for CUDA builds (x86_64 or aarch64)"
+    required: true
+  linux_image:
+    description: "Manylinux image tag to use on Linux runners"
+    required: false
+    default: ""
+  linux_image_tag:
+    description: "Pinned manylinux container tag shared with cibuildwheel"
+    required: false
+    default: ""
+  cuda_architectures:
+    description: "CMake CUDA architectures for libtvm_runtime_cuda.so"
+    required: false
+    default: "75"
+  include_cuda_runtime:
+    description: "Set to true to build the CUDA runtime library"
+    required: false
+    default: "false"
+
+outputs:
+  cuda_runtime_path:
+    description: "Absolute path to the built libtvm_runtime_cuda.so, or empty for CPU-only wheels"
+    value: ${{ steps.cuda_runtime.outputs.path }}
+
+runs:
+  using: "composite"
+  steps:
+    - uses: ./.github/actions/detect-env-vars
+      id: env_vars
+
+    - name: Detect CUDA inputs
+      id: cuda_inputs
+      shell: bash -l {0}
+      env:
+        INPUT_INCLUDE_CUDA_RUNTIME: ${{ inputs.include_cuda_runtime }}
+      run: |
+        set -eux
+        include_cuda_runtime="$(printf '%s' "${INPUT_INCLUDE_CUDA_RUNTIME}" | tr '[:upper:]' '[:lower:]')"
+        case "${include_cuda_runtime}" in
+          1|true|yes|on) include_cuda_runtime=1 ;;
+          0|false|no|off) include_cuda_runtime=0 ;;
+          *)
+            echo "include_cuda_runtime must be a boolean value" >&2
+            exit 1
+            ;;
+        esac
+        echo "include_cuda_runtime=${include_cuda_runtime}" >> "${GITHUB_OUTPUT}"
+
+    - name: Build CUDA runtime in manylinux
+      if: runner.os == 'Linux' && steps.cuda_inputs.outputs.include_cuda_runtime == '1'
+      shell: bash -l {0}
+      env:
+        TVM_MANYLINUX_IMAGE: ${{ inputs.linux_image }}
+        TVM_MANYLINUX_IMAGE_TAG: ${{ inputs.linux_image_tag }}
+        TVM_ARCH: ${{ inputs.arch }}
+        TVM_CUDA_ARCHITECTURES: ${{ inputs.cuda_architectures }}
+        TVM_CUDA_BUILD_DIR: ${{ runner.temp }}/tvm-wheel-cuda
+        TVM_INCLUDE_CUDA_RUNTIME: "1"
+        TVM_BUILD_PARALLEL_LEVEL: ${{ steps.env_vars.outputs.cpu_count }}
+        CMAKE_BUILD_PARALLEL_LEVEL: ${{ steps.env_vars.outputs.cpu_count }}
+      run: ci/scripts/package/tvm_wheel_helper.sh manylinux-cuda
+
+    - name: Reject non-Linux CUDA runtime builds
+      if: runner.os != 'Linux' && steps.cuda_inputs.outputs.include_cuda_runtime == '1'
+      shell: bash -l {0}
+      run: |
+        echo "CUDA runtime wheels are only enabled on Linux in this workflow" >&2
+        exit 1
+
+    - name: Report CUDA runtime output
+      id: cuda_runtime
+      shell: bash -l {0}
+      env:
+        INCLUDE_CUDA_RUNTIME: ${{ steps.cuda_inputs.outputs.include_cuda_runtime }}
+        TVM_CUDA_BUILD_DIR: ${{ runner.temp }}/tvm-wheel-cuda
+        TVM_CUDA_RUNTIME_PATH: ""
+      run: |
+        set -eux
+        if [[ "${INCLUDE_CUDA_RUNTIME}" != "1" ]]; then
+          echo "path=" >> "${GITHUB_OUTPUT}"
+          exit 0
+        fi
+        cuda_runtime="$(ci/scripts/package/tvm_wheel_helper.sh cuda-path)"
+        if [[ -z "${cuda_runtime}" ]]; then
+          echo "CUDA runtime build did not produce libtvm_runtime_cuda.so" >&2
+          exit 1
+        fi
+        case "${cuda_runtime}" in
+          "${TVM_CUDA_BUILD_DIR}"/*) ;;
+          *)
+            echo "CUDA runtime path is outside the expected build directory: ${cuda_runtime}" >&2
+            exit 1
+            ;;
+        esac
+        echo "path=${cuda_runtime}" >> "${GITHUB_OUTPUT}"