Skip to content

Commit

Permalink
account for upstream changes
Browse files Browse the repository at this point in the history
  • Loading branch information
divyegala committed Jul 19, 2023
1 parent 685834f commit af38cdf
Show file tree
Hide file tree
Showing 3 changed files with 6 additions and 13 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/build.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -68,7 +68,7 @@ jobs:
sha: ${{ inputs.sha }}
wheel-build-cuml:
secrets: inherit
uses: rapidsai/shared-action-workflows/.github/workflows/wheels-build.yml@local-script
uses: rapidsai/shared-action-workflows/.github/workflows/wheels-build.yaml@local-script
with:
build_type: ${{ inputs.build_type || 'branch' }}
branch: ${{ inputs.branch }}
Expand Down
2 changes: 1 addition & 1 deletion .github/workflows/pr.yaml
Original file line number Diff line number Diff line change
Expand Up @@ -95,7 +95,7 @@ jobs:
wheel-build-cuml:
needs: checks
secrets: inherit
uses: rapidsai/shared-action-workflows/.github/workflows/wheels-build.yml@local-script
uses: rapidsai/shared-action-workflows/.github/workflows/wheels-build.yaml@local-script
with:
build_type: pull-request
extra-repo: rapidsai/cumlprims_mg
Expand Down
15 changes: 4 additions & 11 deletions ci/build_wheel.sh
Original file line number Diff line number Diff line change
Expand Up @@ -3,19 +3,12 @@

set -euo pipefail

if [[ ! -d "/tmp/gha-tools" ]]; then
git clone https://github.com/divyegala/gha-tools.git -b wheel-local-runs /tmp/gha-tools
fi

if [[ -z "${RAPIDS_EPOCH_TIMESTAMP}" ]]; then
export RAPIDS_EPOCH_TIMESTAMP="$(date +%s)"
fi

source /tmp/gha-tools/tools/rapids-configure-sccache
source rapids-configure-sccache
source rapids-date-string

# Use gha-tools rapids-pip-wheel-version to generate wheel version then
# update the necessary files
version_override="$(rapids-pip-wheel-version ${RAPIDS_EPOCH_TIMESTAMP})"
version_override="$(rapids-pip-wheel-version ${RAPIDS_DATE_STRING})"

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

Expand All @@ -30,4 +23,4 @@ SKBUILD_CONFIGURE_OPTIONS="-DCUML_BUILD_WHEELS=ON -DDETECT_CONDA_ENV=OFF -DDISAB
mkdir -p final_dist
python -m auditwheel repair -w final_dist dist/*

RAPIDS_PY_WHEEL_NAME="cuml_${RAPIDS_PY_CUDA_SUFFIX}" /tmp/gha-tools/tools/rapids-upload-wheels-to-s3 final_dist
RAPIDS_PY_WHEEL_NAME="cuml_${RAPIDS_PY_CUDA_SUFFIX}" rapids-upload-wheels-to-s3 final_dist

0 comments on commit af38cdf

Please sign in to comment.