Skip to content

Commit

Permalink
cuml SINGLEGPU now tells cuvs to not build with nccl/mg support (#6127)
Browse files Browse the repository at this point in the history
Corrects compilation failures when `SINGLEGPU` is enabled on machines without NCCL.

Authors:
  - Robert Maynard (https://github.com/robertmaynard)

Approvers:
  - Dante Gama Dessavre (https://github.com/dantegd)

URL: #6127
  • Loading branch information
robertmaynard authored Oct 31, 2024
1 parent a5e35c5 commit 2feb7d8
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions cpp/cmake/thirdparty/get_cuvs.cmake
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,11 @@ function(find_and_configure_cuvs)
set(CUVS_LIB cuvs::cuvs PARENT_SCOPE)
endif()

set(CUVS_BUILD_MG_ALGOS ON)
if(SINGLEGPU)
set(CUVS_BUILD_MG_ALGOS OFF)
endif()

rapids_cpm_find(cuvs ${PKG_VERSION}
GLOBAL_TARGETS cuvs::cuvs
BUILD_EXPORT_SET cuml-exports
Expand All @@ -50,6 +55,8 @@ function(find_and_configure_cuvs)
OPTIONS
"BUILD_TESTS OFF"
"BUILD_BENCH OFF"
"BUILD_MG_ALGOS ${CUVS_BUILD_MG_ALGOS}"

)

if(cuvs_ADDED)
Expand Down

0 comments on commit 2feb7d8

Please sign in to comment.