Skip to content

Commit 46eaf5a

Browse files
committed
Added RMM config
1 parent 1f291a9 commit 46eaf5a

1 file changed

Lines changed: 14 additions & 0 deletions

File tree

config/RMM.sh

Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
# The root of repo does not contains a setup nor pyproject. Use git instead.
2+
PACKAGE_DOWNLOAD_ARGUMENT="https://github.com/rapidsai/rmm.git"
3+
PACKAGE_DOWNLOAD_NAME="$PACKAGE-$VERSION.tar.gz"
4+
PACKAGE_DOWNLOAD_METHOD="Git"
5+
PACKAGE_DOWNLOAD_CMD="git clone --recursive $PACKAGE_DOWNLOAD_ARGUMENT --branch v${VERSION:?version required} $PACKAGE_FOLDER_NAME"
6+
POST_DOWNLOAD_COMMANDS="tar -zcf ${PACKAGE}-${VERSION}.tar.gz $PACKAGE_FOLDER_NAME"
7+
8+
PYTHON_DEPS="scikit-build>=0.13.1 cuda-python>=11.7.1,<12.0 cython ninja"
9+
MODULE_BUILD_DEPS="cmake cuda/11.7 spdlog"
10+
PRE_BUILD_COMMANDS="
11+
export RAPIDS_PY_WHEEL_VERSIONEER_OVERRIDE=$VERSION;
12+
cd python;
13+
sed -i -e 's/version=\".*\"/version=\"$VERSION\"/' setup.py;
14+
"

0 commit comments

Comments
 (0)