forked from rapidsai/cugraph
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Initial version of build.sh for cugraph, plus updates to correspondin…
…g scripts and docs.
- Loading branch information
Showing
6 changed files
with
175 additions
and
70 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,118 @@ | ||
| #!/bin/bash | ||
|
|
||
| # Copyright (c) 2019, NVIDIA CORPORATION. | ||
|
|
||
| # cugraph build script | ||
|
|
||
| # This script is used to build the component(s) in this repo from | ||
| # source, and can be called with various options to customize the | ||
| # build as needed (see the help output for details) | ||
|
|
||
| # Abort script on first error | ||
| set -e | ||
|
|
||
| NUMARGS=$# | ||
| ARGS=$* | ||
|
|
||
| # NOTE: ensure all dir changes are relative to the location of this | ||
| # script, and that this script resides in the repo dir! | ||
| REPODIR=$(cd $(dirname $0); pwd) | ||
|
|
||
| VALIDARGS="clean libcugraph cugraph -v -g -n -h --help" | ||
| HELP="$0 [<target> ...] [<flag> ...] | ||
| where <target> is: | ||
| clean - remove all existing build artifacts and configuration (start over) | ||
| libcugraph - build the cugraph C++ code only | ||
| cugraph - build the cugraph Python package | ||
| and <flag> is: | ||
| -v - verbose build mode | ||
| -g - build for debug | ||
| -n - no install step | ||
| -h - print this text | ||
| default action (no args) is to build and install 'libcugraph' then 'cugraph' targets | ||
| " | ||
| LIBCUGRAPH_BUILD_DIR=${REPODIR}/cpp/build | ||
| CUGRAPH_BUILD_DIR=${REPODIR}/python/build | ||
| BUILD_DIRS="${LIBCUGRAPH_BUILD_DIR} ${CUGRAPH_BUILD_DIR}" | ||
|
|
||
| # Set defaults for vars modified by flags to this script | ||
| VERBOSE="" | ||
| BUILD_TYPE=Release | ||
| INSTALL_TARGET=install | ||
|
|
||
| # Set defaults for vars that may not have been defined externally | ||
| # FIXME: if INSTALL_PREFIX is not set, check PREFIX, then check | ||
| # CONDA_PREFIX, but there is no fallback from there! | ||
| INSTALL_PREFIX=${INSTALL_PREFIX:=${PREFIX:=${CONDA_PREFIX}}} | ||
| PARALLEL_LEVEL=${PARALLEL_LEVEL:=""} | ||
| BUILD_ABI=${BUILD_ABI:=ON} | ||
|
|
||
| function hasArg { | ||
| (( ${NUMARGS} != 0 )) && (echo " ${ARGS} " | grep -q " $1 ") | ||
| } | ||
|
|
||
| if hasArg -h || hasArg --help; then | ||
| echo "${HELP}" | ||
| exit 0 | ||
| fi | ||
|
|
||
| # Check for valid usage | ||
| if (( ${NUMARGS} != 0 )); then | ||
| for a in ${ARGS}; do | ||
| if ! (echo " ${VALIDARGS} " | grep -q " ${a} "); then | ||
| echo "Invalid option: ${a}" | ||
| exit 1 | ||
| fi | ||
| done | ||
| fi | ||
|
|
||
| # Process flags | ||
| if hasArg -v; then | ||
| VERBOSE=1 | ||
| fi | ||
| if hasArg -g; then | ||
| BUILD_TYPE=Debug | ||
| fi | ||
| if hasArg -n; then | ||
| INSTALL_TARGET="" | ||
| fi | ||
|
|
||
| # If clean given, run it prior to any other steps | ||
| if hasArg clean; then | ||
| # If the dirs to clean are mounted dirs in a container, the | ||
| # contents should be removed but the mounted dirs will remain. | ||
| # The find removes all contents but leaves the dirs, the rmdir | ||
| # attempts to remove the dirs but can fail safely. | ||
| for bd in ${BUILD_DIRS}; do | ||
| if [ -d ${bd} ]; then | ||
| find ${bd} -mindepth 1 -delete | ||
| rmdir ${bd} || true | ||
| fi | ||
| done | ||
| fi | ||
|
|
||
| ################################################################################ | ||
| # Configure, build, and install libcugraph | ||
| if (( ${NUMARGS} == 0 )) || hasArg libcugraph; then | ||
|
|
||
| mkdir -p ${LIBCUGRAPH_BUILD_DIR} | ||
| cd ${LIBCUGRAPH_BUILD_DIR} | ||
| cmake -DCMAKE_INSTALL_PREFIX=${INSTALL_PREFIX} \ | ||
| -DCMAKE_CXX11_ABI=${BUILD_ABI} \ | ||
| -DNVG_PLUGIN=True \ | ||
| -DCMAKE_BUILD_TYPE=${BUILD_TYPE} .. | ||
| make -j${PARALLEL_LEVEL} VERBOSE=${VERBOSE} ${INSTALL_TARGET} | ||
| fi | ||
|
|
||
| # Build and install the cugraph Python package | ||
| if (( ${NUMARGS} == 0 )) || hasArg cugraph; then | ||
|
|
||
| cd ${REPODIR}/python | ||
| if [[ ${INSTALL_TARGET} != "" ]]; then | ||
| python setup.py build_ext --inplace | ||
| python setup.py install --single-version-externally-managed --record=record.txt | ||
| else | ||
| python setup.py build_ext --inplace --library-dir=${LIBCUGRAPH_BUILD_DIR} | ||
| fi | ||
| fi |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,4 +1,4 @@ | ||
| #!/usr/bin/env bash | ||
|
|
||
| cd python | ||
| python setup.py install | ||
| # This assumes the script is executed from the root of the repo directory | ||
| ./build.sh cugraph |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,16 +1,12 @@ | ||
| #!/usr/bin/env bash | ||
| CMAKE_COMMON_VARIABLES=" -DCMAKE_INSTALL_PREFIX=$PREFIX -DCMAKE_BUILD_TYPE=Release -DCMAKE_CXX11_ABI=$BUILD_ABI -DNVG_PLUGIN=True" | ||
|
|
||
| # This assumes the script is executed from the root of the repo directory | ||
|
|
||
| # show environment | ||
| printenv | ||
| # Cleanup local git | ||
| if [ -d .git ]; then | ||
| git clean -xdf | ||
| fi | ||
| # Use CMake-based build procedure | ||
| mkdir -p cpp/build | ||
| cd cpp/build | ||
| # configure | ||
| cmake $CMAKE_COMMON_VARIABLES .. | ||
| # build | ||
| make -j${PARALLEL_LEVEL} VERBOSE=1 install | ||
|
|
||
| ./build.sh libcugraph -v |