Skip to content

Commit

Permalink
go back to installing everything
Browse files Browse the repository at this point in the history
  • Loading branch information
jameslamb committed Oct 14, 2024
1 parent 28e34eb commit 8251aaa
Showing 1 changed file with 15 additions and 9 deletions.
24 changes: 15 additions & 9 deletions ci/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -33,31 +33,36 @@ rapids-generate-version > ./VERSION

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

# Using env variable PIP_CONSTRAINT is necessary to ensure the constraints
# are used when installing build dependencies.
export PIP_CONSTRAINT="/tmp/constraints.txt"
touch "${PIP_CONSTRAINT}"
touch /tmp/constraints-build.txt

if [[ "${package_name}" == "cuspatial" ]]; then
# Downloads libcuspatial wheel from this current build,
# then ensures 'cuspatial' wheel builds always use the 'libcuspatial' just built in the same CI run.
RAPIDS_PY_WHEEL_NAME="libcuspatial_${RAPIDS_PY_CUDA_SUFFIX}" rapids-download-wheels-from-s3 cpp /tmp/libcuspatial_dist
echo "libcuspatial-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo /tmp/libcuspatial_dist/libcuspatial_*.whl)" >> "${PIP_CONSTRAINT}"
echo "libcuspatial-${RAPIDS_PY_CUDA_SUFFIX} @ file://$(echo /tmp/libcuspatial_dist/libcuspatial_*.whl)" > /tmp/constraints-build.txt
fi

rapids-logger "Generating build backend requirements"
rapids-logger "Generating build requirements"
declare -r matrix_selectors="cuda=${RAPIDS_CUDA_VERSION%.*};arch=$(arch);py=${RAPIDS_PY_VERSION};cuda_suffixed=true"

rapids-dependency-file-generator \
--output requirements \
--file-key "py_build_${package_name}" \
--matrix "${matrix_selectors}" \
| tee /tmp/requirements-build-backend.txt
| tee /tmp/requirements-build.txt

rapids-logger "Installing build backend requirements"
rapids-dependency-file-generator \
--output requirements \
--file-key "py_rapids_build_${package_name}" \
--matrix "${matrix_selectors}" \
| tee -a /tmp/requirements-build.txt

rapids-logger "Installing build requirements"
python -m pip install \
-v \
-r /tmp/requirements-build-backend.txt
--prefer-binary \
--constraint /tmp/constraints-build.txt \
-r /tmp/requirements-build.txt

cd "${package_dir}"

Expand All @@ -66,6 +71,7 @@ python -m pip wheel \
-w dist \
-v \
--no-build-isolation \
--no-deps \
--disable-pip-version-check \
.

Expand Down

0 comments on commit 8251aaa

Please sign in to comment.