From 2feb7d85c72bd0e1c5134024c861073d08e4bb0d Mon Sep 17 00:00:00 2001 From: Robert Maynard Date: Thu, 31 Oct 2024 12:50:27 -0400 Subject: [PATCH] cuml SINGLEGPU now tells cuvs to not build with nccl/mg support (#6127) 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: https://github.com/rapidsai/cuml/pull/6127 --- cpp/cmake/thirdparty/get_cuvs.cmake | 7 +++++++ 1 file changed, 7 insertions(+) diff --git a/cpp/cmake/thirdparty/get_cuvs.cmake b/cpp/cmake/thirdparty/get_cuvs.cmake index 1d319206ba..a46879c3e7 100644 --- a/cpp/cmake/thirdparty/get_cuvs.cmake +++ b/cpp/cmake/thirdparty/get_cuvs.cmake @@ -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 @@ -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)