From 5c3390771c45acd9094073a00610feb46dd3e3f7 Mon Sep 17 00:00:00 2001
From: Vyas Ramasubramani <vyasr@nvidia.com>
Date: Wed, 27 Nov 2024 23:17:35 +0000
Subject: [PATCH] Also add target to Python packages

---
 python/cuvs/cuvs/common/CMakeLists.txt    | 4 ++++
 python/cuvs/cuvs/distance/CMakeLists.txt  | 4 ++++
 python/cuvs/cuvs/neighbors/CMakeLists.txt | 4 ++++
 3 files changed, 12 insertions(+)

diff --git a/python/cuvs/cuvs/common/CMakeLists.txt b/python/cuvs/cuvs/common/CMakeLists.txt
index 202919e01..f007bc922 100644
--- a/python/cuvs/cuvs/common/CMakeLists.txt
+++ b/python/cuvs/cuvs/common/CMakeLists.txt
@@ -22,3 +22,7 @@ rapids_cython_create_modules(
   SOURCE_FILES "${cython_sources}"
   LINKED_LIBRARIES "${linked_libraries}" ASSOCIATED_TARGETS cuvs MODULE_PREFIX common_
 )
+
+foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
+  target_link_library(${tgt} PRIVATE rmm::rmm_logger_impl)
+endforeach()
diff --git a/python/cuvs/cuvs/distance/CMakeLists.txt b/python/cuvs/cuvs/distance/CMakeLists.txt
index 363778a9c..71cad5601 100644
--- a/python/cuvs/cuvs/distance/CMakeLists.txt
+++ b/python/cuvs/cuvs/distance/CMakeLists.txt
@@ -22,3 +22,7 @@ rapids_cython_create_modules(
   SOURCE_FILES "${cython_sources}"
   LINKED_LIBRARIES "${linked_libraries}" ASSOCIATED_TARGETS cuvs MODULE_PREFIX distance_
 )
+
+foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
+  target_link_library(${tgt} PRIVATE rmm::rmm_logger_impl)
+endforeach()
diff --git a/python/cuvs/cuvs/neighbors/CMakeLists.txt b/python/cuvs/cuvs/neighbors/CMakeLists.txt
index f68bbea53..fe2965070 100644
--- a/python/cuvs/cuvs/neighbors/CMakeLists.txt
+++ b/python/cuvs/cuvs/neighbors/CMakeLists.txt
@@ -29,3 +29,7 @@ rapids_cython_create_modules(
   SOURCE_FILES "${cython_sources}"
   LINKED_LIBRARIES "${linked_libraries}" ASSOCIATED_TARGETS cuvs MODULE_PREFIX neighbors_refine_
 )
+
+foreach(tgt IN LISTS RAPIDS_CYTHON_CREATED_TARGETS)
+  target_link_library(${tgt} PRIVATE rmm::rmm_logger_impl)
+endforeach()