diff --git a/.github/workflows/release.yaml b/.github/workflows/release.yaml index 6548bed..842145c 100644 --- a/.github/workflows/release.yaml +++ b/.github/workflows/release.yaml @@ -22,13 +22,13 @@ jobs: package_name: 'apache-tvm' - cuda: '10.2' image: 'tlcpack/package-cu102:1d98bf9' - package_name: 'apache-tvm' + package_name: 'apache-tvm-cu102' - cuda: '11.3' image: 'tlcpack/package-cu113:1d98bf9' - package_name: 'apache-tvm' + package_name: 'apache-tvm-cu113' - cuda: '11.6' image: 'tlcpack/package-cu116:1d98bf9' - package_name: 'apache-tvm' + package_name: 'apache-tvm-cu116' runs-on: ubuntu-latest steps: diff --git a/docker/Dockerfile.package-cu116 b/docker/Dockerfile.package-cu116 index e5b52a0..e438b2e 100644 --- a/docker/Dockerfile.package-cu116 +++ b/docker/Dockerfile.package-cu116 @@ -40,7 +40,6 @@ RUN conda create -n py310 python=3.10 -y COPY install/centos_install_python_package.sh /install/centos_install_python_package.sh RUN bash /install/centos_install_python_package.sh - COPY install/centos_install_auditwheel.sh /install/centos_install_auditwheel.sh RUN bash /install/centos_install_auditwheel.sh diff --git a/wheel/build_wheel_manylinux.sh b/wheel/build_wheel_manylinux.sh index b382b41..1c3b3d8 100755 --- a/wheel/build_wheel_manylinux.sh +++ b/wheel/build_wheel_manylinux.sh @@ -1,15 +1,26 @@ #!/usr/bin/env bash -set -ex +set -exo pipefail source /multibuild/manylinux_utils.sh TVM_PYTHON_DIR="/workspace/tvm/python" -PYTHON_VERSIONS_CPU=("3.7" "3.8" "3.9" "3.10") -PYTHON_VERSIONS_GPU=("3.7" "3.8" "3.9" "3.10") + +PYTHON_VERSIONS=$(conda env list | grep py | awk '{print $1}' | tr -d py3 | xargs -I {} printf "3.{}\n" | sort -V) +PYTHON_VERSIONS_CPU=$PYTHON_VERSIONS +PYTHON_VERSIONS_GPU=$PYTHON_VERSIONS CUDA_OPTIONS=("none" "10.2" "11.0" "11.1" "11.3" "11.6") CUDA="none" + +function activate() { + eval "$(conda shell.bash hook)" + PYTHON_VERSION=$1 + ENV_NAME=$(echo "$PYTHON_VERSION" | tr -d 3. | xargs -I {} printf "py3{}\n") + conda activate "$ENV_NAME" +} + + function usage() { echo "Usage: $0 [--cuda CUDA]" echo @@ -73,18 +84,6 @@ while [[ $# -gt 0 ]]; do esac done -for python_version in ${PYTHON_VERSIONS[*]} -do - echo "> Looking for Python ${python_version}." - cpython_dir="$(cpython_path ${python_version} ${UNICODE_WIDTH} 2> /dev/null)" - if [ -d "${cpython_dir}" ]; then - echo "Python ${python_version} found"; - else - echo "Python ${python_version} not found"; - exit 1 - fi -done - if ! in_array "${CUDA}" "${CUDA_OPTIONS[*]}" ; then echo "Invalid CUDA option: ${CUDA}" @@ -130,7 +129,7 @@ fi mkdir -p build cd build cmake .. -make -j$(nproc) +make -j"$(nproc)" UNICODE_WIDTH=32 # Dummy value, irrelevant for Python 3 @@ -138,6 +137,7 @@ UNICODE_WIDTH=32 # Dummy value, irrelevant for Python 3 # so check the existing python versions before generating packages for python_version in ${PYTHON_VERSIONS[*]} do + activate "$python_version" echo "> Looking for Python ${python_version}." cpython_dir="$(cpython_path ${python_version} ${UNICODE_WIDTH} 2> /dev/null)" if [ -d "${cpython_dir}" ]; then