Skip to content

Commit

Permalink
Added RMM config
Browse files Browse the repository at this point in the history
  • Loading branch information
ccoulombe committed Nov 25, 2022
1 parent 1f291a9 commit 5544fb5
Showing 1 changed file with 13 additions and 0 deletions.
13 changes: 13 additions & 0 deletions config/RMM.sh
Original file line number Diff line number Diff line change
@@ -0,0 +1,13 @@
# 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;
"

0 comments on commit 5544fb5

Please sign in to comment.