From 1f291a9ae0c2e6496e99ee8ba08ff32bb2931248 Mon Sep 17 00:00:00 2001 From: Charles Coulombe Date: Fri, 25 Nov 2022 14:55:14 -0500 Subject: [PATCH 1/2] Added cuda-python config --- config/cuda_python.sh | 2 ++ 1 file changed, 2 insertions(+) create mode 100644 config/cuda_python.sh diff --git a/config/cuda_python.sh b/config/cuda_python.sh new file mode 100644 index 0000000..2779a0f --- /dev/null +++ b/config/cuda_python.sh @@ -0,0 +1,2 @@ +MODULE_BUILD_DEPS='cuda/11.7' +PACKAGE_DOWNLOAD_ARGUMENT="https://github.com/NVIDIA/cuda-python/archive/refs/tags/v${VERSION:?version required}.tar.gz" From 46eaf5a3f785e51b74e6fecd8e89b11558b538e7 Mon Sep 17 00:00:00 2001 From: Charles Coulombe Date: Fri, 25 Nov 2022 16:45:29 -0500 Subject: [PATCH 2/2] Added RMM config --- config/RMM.sh | 14 ++++++++++++++ 1 file changed, 14 insertions(+) create mode 100644 config/RMM.sh diff --git a/config/RMM.sh b/config/RMM.sh new file mode 100644 index 0000000..fcadf33 --- /dev/null +++ b/config/RMM.sh @@ -0,0 +1,14 @@ +# The root of repo does not contains a setup nor pyproject. Use git instead. +PACKAGE_DOWNLOAD_ARGUMENT="https://github.com/rapidsai/rmm.git" +PACKAGE_DOWNLOAD_NAME="$PACKAGE-$VERSION.tar.gz" +PACKAGE_DOWNLOAD_METHOD="Git" +PACKAGE_DOWNLOAD_CMD="git clone --recursive $PACKAGE_DOWNLOAD_ARGUMENT --branch v${VERSION:?version required} $PACKAGE_FOLDER_NAME" +POST_DOWNLOAD_COMMANDS="tar -zcf ${PACKAGE}-${VERSION}.tar.gz $PACKAGE_FOLDER_NAME" + +PYTHON_DEPS="scikit-build>=0.13.1 cuda-python>=11.7.1,<12.0 cython ninja" +MODULE_BUILD_DEPS="cmake cuda/11.7 spdlog" +PRE_BUILD_COMMANDS=" + export RAPIDS_PY_WHEEL_VERSIONEER_OVERRIDE=$VERSION; + cd python; + sed -i -e 's/version=\".*\"/version=\"$VERSION\"/' setup.py; +"