build_wheel.sh

#!/bin/bash
# Copyright (c) 2023, NVIDIA CORPORATION.

set -euo pipefail

package_name="cucim"
package_dir="python/cucim"
package_src_dir="${package_dir}/src/${package_name}"

CMAKE_BUILD_TYPE="release"

source rapids-configure-sccache
source rapids-date-string

rapids-generate-version > ./VERSION

RAPIDS_PY_CUDA_SUFFIX="$(rapids-wheel-ctk-name-gen ${RAPIDS_CUDA_VERSION})"

PACKAGE_CUDA_SUFFIX="-${RAPIDS_PY_CUDA_SUFFIX}"

# Install pip build dependencies (not yet using pyproject.toml)
rapids-dependency-file-generator \
  --file-key "py_build" \
  --output "requirements" \
  --matrix "cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION}" | tee build_requirements.txt
pip install -r build_requirements.txt

# First build the C++ lib using CMake via the run script
./run build_local all ${CMAKE_BUILD_TYPE}

cd "${package_dir}"

python -m pip wheel . -w dist -vvv --no-deps --disable-pip-version-check

mkdir -p final_dist
python -m auditwheel repair -w final_dist dist/*
ls -1 final_dist | grep -vqz 'none'

# Build the wheel for Windows if the current platform is x86_64 (#763)
if [ "$(arch)" = "x86_64" ]; then
  # Install the wheel package
  python -m pip install wheel

  # Create directories
  mkdir -p unpacked converted_dist
  # Unpack the wheel
  python -m wheel unpack final_dist/*_x86_64.whl -d unpacked

  # Delete unnecessary files/folder (.so files, .libs folder)
  find unpacked -name '*.so*' -exec rm {} +
  find unpacked -maxdepth 2 -type d -name '*.libs' -exec rm -r {} +

  # Pack the wheel again
  python -m wheel pack --dest-dir converted_dist unpacked/*

  # Change the platform tag to win_amd64 and store it in the final_dist folder
  python -m wheel tags --remove --python-tag=py3 --abi-tag=none --platform-tag=win_amd64 converted_dist/*_x86_64.whl
  mv converted_dist/*-win_amd64.whl final_dist
fi

RAPIDS_PY_WHEEL_NAME="${package_name}_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 final_dist