Fix raw CUDA stream boundaries in benchmarks and tests

Signed-off-by: Bradley Dice <bdice@bradleydice.com>

Author: bdice

cpp/bench/ann/src/common/util.hpp

@@ -171,7 +171,7 @@ inline auto get_stream_from_global_pool() -> cudaStream_t
       detail::global_stream_pool.emplace_back(rmm::cuda_stream::flags::non_blocking);
     }
   }
-  return detail::global_stream_pool[benchmark_thread_id].view();
+  return detail::global_stream_pool[benchmark_thread_id].value();
 #else
   return nullptr;
 #endif

cpp/bench/ann/src/cuvs/cuvs_brute_force_knn.cu

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION.
+ * SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
 #include <cuvs/distance/distance.hpp>
@@ -72,7 +72,7 @@ template <typename T, typename DistT = T>
 class BruteForceKNNBenchmark {
  public:
   BruteForceKNNBenchmark(const RandomKNNInputs& params, const std::string& type_str)
-    : stream_(raft::resource::get_cuda_stream(handle_)),
+    : stream_(raft::resource::get_cuda_stream(handle_).get()),
       params_(params),
       type_str_(type_str),
       database(params_.num_db_vecs * params_.dim, stream_),

cpp/bench/ann/src/cuvs/cuvs_cagra_hnswlib_wrapper.h

@@ -45,7 +45,7 @@ class cuvs_cagra_hnswlib : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   [[nodiscard]] auto uses_stream() const noexcept -> bool override

cpp/bench/ann/src/cuvs/cuvs_cagra_wrapper.h

@@ -205,7 +205,7 @@ class cuvs_cagra : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   [[nodiscard]] auto uses_stream() const noexcept -> bool override

cpp/bench/ann/src/cuvs/cuvs_ivf_flat_wrapper.h

@@ -57,7 +57,7 @@ class cuvs_ivf_flat : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   // to enable dataset access from GPU memory

cpp/bench/ann/src/cuvs/cuvs_ivf_pq_wrapper.h

@@ -70,7 +70,7 @@ class cuvs_ivf_pq : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   // to enable dataset access from GPU memory

cpp/bench/ann/src/cuvs/cuvs_ivf_rabitq_wrapper.h

@@ -55,7 +55,7 @@ class cuvs_ivf_rabitq : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   // to enable dataset access from GPU memory

cpp/bench/ann/src/cuvs/cuvs_ivf_sq_wrapper.h

@@ -52,7 +52,7 @@ class cuvs_ivf_sq : public algo<T>, public algo_gpu {
 
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
-    return handle_.get_sync_stream();
+    return handle_.get_sync_stream().get();
   }
 
   [[nodiscard]] auto get_preference() const -> algo_property override

cpp/bench/ann/src/cuvs/cuvs_mg_cagra_wrapper.h

@@ -56,7 +56,7 @@ class cuvs_mg_cagra : public algo<T>, public algo_gpu {
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
     auto stream = raft::resource::get_cuda_stream(clique_);
-    return stream;
+    return stream.get();
   }
 
   // to enable dataset access from GPU memory

cpp/bench/ann/src/cuvs/cuvs_mg_ivf_flat_wrapper.h

@@ -1,5 +1,5 @@
 /*
- * SPDX-FileCopyrightText: Copyright (c) 2024, NVIDIA CORPORATION.
+ * SPDX-FileCopyrightText: Copyright (c) 2024-2026, NVIDIA CORPORATION & AFFILIATES. All rights reserved.
  * SPDX-License-Identifier: Apache-2.0
  */
 
@@ -52,7 +52,7 @@ class cuvs_mg_ivf_flat : public algo<T>, public algo_gpu {
   [[nodiscard]] auto get_sync_stream() const noexcept -> cudaStream_t override
   {
     auto stream = raft::resource::get_cuda_stream(clique_);
-    return stream;
+    return stream.get();
   }
 
   [[nodiscard]] auto uses_stream() const noexcept -> bool override { return false; }