From b8273bbe0e9218e72eab4bc0b2dc833f1fb98b5a Mon Sep 17 00:00:00 2001
From: Bradley Dice <bdice@bradleydice.com>
Date: Fri, 28 Aug 2026 12:40:22 -0500
Subject: [PATCH] Adapt pooled stream use to cuda::stream_ref

---
 cpp/src/neighbors/detail/knn_brute_force.cuh      | 2 +-
 cpp/src/stats/detail/batched/silhouette_score.cuh | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/cpp/src/neighbors/detail/knn_brute_force.cuh b/cpp/src/neighbors/detail/knn_brute_force.cuh
index 76ed71fe57..584a88ef54 100644
--- a/cpp/src/neighbors/detail/knn_brute_force.cuh
+++ b/cpp/src/neighbors/detail/knn_brute_force.cuh
@@ -465,7 +465,7 @@ void brute_force_knn_impl(
                  k,
                  rowMajorIndex,
                  rowMajorQuery,
-                 stream,
+                 stream.get(),
                  metric,
                  input_norms ? (*input_norms)[i] : nullptr,
                  search_norms);
diff --git a/cpp/src/stats/detail/batched/silhouette_score.cuh b/cpp/src/stats/detail/batched/silhouette_score.cuh
index f41b843d5c..568773a3f6 100644
--- a/cpp/src/stats/detail/batched/silhouette_score.cuh
+++ b/cpp/src/stats/detail/batched/silhouette_score.cuh
@@ -241,7 +241,7 @@ value_t silhouette_score(
                           distances.data(),
                           n_left_rows,
                           n_right_rows,
-                          chunk_stream);
+                          chunk_stream.get());
     }
   }