diff --git a/build.sh b/build.sh index e342dec3ba..2b88403606 100755 --- a/build.sh +++ b/build.sh @@ -20,7 +20,7 @@ ARGS=$* REPODIR=$(cd "$(dirname "$0")"; pwd) VALIDTARGETS="clean libcuml cuml cpp-mgtests prims bench prims-bench cppdocs pydocs" -VALIDFLAGS="-v -g -n --allgpuarch --singlegpu --nolibcumltest --nvtx --show_depr_warn --codecov --ccache --configure-only --build-metrics --incl-cache-stats -h --help " +VALIDFLAGS="-v -g -n --allgpuarch --singlegpu --nolibcumltest --nvtx --show_depr_warn --codecov --ccache --configure-only --static-treelite --build-metrics --incl-cache-stats -h --help " VALIDARGS="${VALIDTARGETS} ${VALIDFLAGS}" HELP="$0 [ ...] [ ...] where is: @@ -154,6 +154,7 @@ LONG_ARGUMENT_LIST=( "ccache" "nolibcumltest" "configure-only" + "static-treelite" "build-metrics" "incl-cache-stats" ) diff --git a/cpp/cmake/thirdparty/get_nvforest.cmake b/cpp/cmake/thirdparty/get_nvforest.cmake index 38065ef3b7..dee2d52765 100644 --- a/cpp/cmake/thirdparty/get_nvforest.cmake +++ b/cpp/cmake/thirdparty/get_nvforest.cmake @@ -28,6 +28,7 @@ function(find_and_configure_nvforest) EXCLUDE_FROM_ALL ${PKG_EXCLUDE_FROM_ALL} OPTIONS "BUILD_NVFOREST_TESTS OFF" + "NVFOREST_USE_TREELITE_STATIC ${CUML_USE_TREELITE_STATIC}" ) if(nvforest_ADDED)