diff --git a/.github/workflows/build.yaml b/.github/workflows/build.yaml index 31a1bd35f..0189aebb2 100644 --- a/.github/workflows/build.yaml +++ b/.github/workflows/build.yaml @@ -74,7 +74,7 @@ jobs: package-name: rmm package-dir: python skbuild-configure-options: "-DRMM_BUILD_WHEELS=ON" - uses-versioneer: false + uses-setup-env-vars: false wheel-publish: needs: wheel-build secrets: inherit diff --git a/.github/workflows/pr.yaml b/.github/workflows/pr.yaml index 858c02691..240344bc1 100644 --- a/.github/workflows/pr.yaml +++ b/.github/workflows/pr.yaml @@ -68,7 +68,7 @@ jobs: package-dir: python package-name: rmm skbuild-configure-options: "-DRMM_BUILD_WHEELS=ON" - uses-versioneer: false + uses-setup-env-vars: false wheel-tests: needs: wheel-build secrets: inherit diff --git a/ci/release/apply_wheel_modifications.sh b/ci/release/apply_wheel_modifications.sh index aad6ae84f..c981cd162 100755 --- a/ci/release/apply_wheel_modifications.sh +++ b/ci/release/apply_wheel_modifications.sh @@ -4,6 +4,9 @@ # Usage: bash apply_wheel_modifications.sh VERSION=${1} +CUDA_SUFFIX=${2} sed -i "s/__version__ = .*/__version__ = \"${VERSION}\"/g" python/rmm/__init__.py sed -i "s/version=.*,/version=\"${VERSION}\",/g" python/setup.py + +sed -i "s/name=\"rmm\",/name=\"rmm${CUDA_SUFFIX}\",/g" python/setup.py diff --git a/python/setup.py b/python/setup.py index a60f87e75..221908184 100644 --- a/python/setup.py +++ b/python/setup.py @@ -1,12 +1,10 @@ # Copyright (c) 2019-2022, NVIDIA CORPORATION. -import os - from setuptools import find_packages from skbuild import setup setup( - name="rmm" + os.getenv("RAPIDS_PY_WHEEL_CUDA_SUFFIX", default=""), + name="rmm", version="23.04.00", description="rmm - RAPIDS Memory Manager", url="https://github.com/rapidsai/rmm",