Added options to PLC and cugraph CMakeLists.txt for disabling cugraph…

…-ops, updated build.sh to pass option to disable cugraph-ops.
rapidsai · Nov 11, 2022 · 8e2f971 · 8e2f971
1 parent b412ea1
commit 8e2f971
Show file tree

Hide file tree

Showing 3 changed files with 28 additions and 4 deletions.
diff --git a/build.sh b/build.sh
@@ -250,8 +250,14 @@ if buildAll || hasArg pylibcugraph; then
     # setup.py references an env var CUGRAPH_BUILD_PATH to find the libcugraph
     # build. If not set by the user, set it to LIBCUGRAPH_BUILD_DIR
     CUGRAPH_BUILD_PATH=${CUGRAPH_BUILD_PATH:=${LIBCUGRAPH_BUILD_DIR}}
-    python setup.py build_ext --inplace -- -DFIND_CUGRAPH_CPP=ON \
-           -Dcugraph_ROOT=${LIBCUGRAPH_BUILD_DIR} -- -j${PARALLEL_LEVEL:-1}
+    python setup.py build_ext \
+	   --inplace \
+	   -- \
+	   -DFIND_CUGRAPH_CPP=ON \
+	   -DUSE_CUGRAPH_OPS=${BUILD_WITH_CUGRAPHOPS} \
+           -Dcugraph_ROOT=${LIBCUGRAPH_BUILD_DIR} \
+	   -- \
+	   -j${PARALLEL_LEVEL:-1}
     if [[ ${INSTALL_TARGET} != "" ]]; then
 	env CUGRAPH_BUILD_PATH=${CUGRAPH_BUILD_PATH} python setup.py ${PYTHON_INSTALL}
     fi
@@ -264,8 +270,14 @@ if buildAll || hasArg cugraph; then
     # setup.py references an env var CUGRAPH_BUILD_PATH to find the libcugraph
     # build. If not set by the user, set it to LIBCUGRAPH_BUILD_DIR
     CUGRAPH_BUILD_PATH=${CUGRAPH_BUILD_PATH:=${LIBCUGRAPH_BUILD_DIR}}
-    python setup.py build_ext --inplace -- -DFIND_CUGRAPH_CPP=ON \
-           -Dcugraph_ROOT=${LIBCUGRAPH_BUILD_DIR} -- -j${PARALLEL_LEVEL:-1}
+    python setup.py build_ext \
+	   --inplace \
+	   -- \
+	   -DFIND_CUGRAPH_CPP=ON \
+	   -DUSE_CUGRAPH_OPS=${BUILD_WITH_CUGRAPHOPS} \
+           -Dcugraph_ROOT=${LIBCUGRAPH_BUILD_DIR} \
+	   -- \
+	   -j${PARALLEL_LEVEL:-1}
     if [[ ${INSTALL_TARGET} != "" ]]; then
 	env CUGRAPH_BUILD_PATH=${CUGRAPH_BUILD_PATH} python setup.py ${PYTHON_INSTALL}
     fi

diff --git a/python/cugraph/CMakeLists.txt b/python/cugraph/CMakeLists.txt
@@ -33,6 +33,12 @@ project(
 option(FIND_CUGRAPH_CPP "Search for existing CUGRAPH C++ installations before defaulting to local files"
        OFF
 )
+option(USE_CUGRAPH_OPS "Enable all functions that call cugraph-ops" ON)
+
+if(NOT USE_CUGRAPH_OPS)
+    message(STATUS "Disabling libcugraph functions that reference cugraph-ops")
+    add_compile_definitions(NO_CUGRAPH_OPS)
+endif()
 
 # If the user requested it,  we attempt to find CUGRAPH.
 if(FIND_CUGRAPH_CPP)

diff --git a/python/pylibcugraph/CMakeLists.txt b/python/pylibcugraph/CMakeLists.txt
@@ -33,6 +33,12 @@ project(
 option(FIND_CUGRAPH_CPP "Search for existing CUGRAPH C++ installations before defaulting to local files"
        OFF
 )
+option(USE_CUGRAPH_OPS "Enable all functions that call cugraph-ops" ON)
+
+if(NOT USE_CUGRAPH_OPS)
+    message(STATUS "Disabling libcugraph functions that reference cugraph-ops")
+    add_compile_definitions(NO_CUGRAPH_OPS)
+endif()
 
 # If the user requested it we attempt to find CUGRAPH.