From 270111ce11bbbd44523aaf075454c0b0c3f1451d Mon Sep 17 00:00:00 2001
From: abc99lr <abc99lr@gmail.com>
Date: Thu, 9 May 2024 17:34:26 +0000
Subject: [PATCH] Use nosync policy by default.

---
 cpp/include/raft/core/device_resources.hpp           | 2 +-
 cpp/include/raft/core/resource/thrust_policy.hpp     | 8 ++++----
 cpp/include/raft/spectral/detail/matrix_wrappers.hpp | 2 +-
 3 files changed, 6 insertions(+), 6 deletions(-)

diff --git a/cpp/include/raft/core/device_resources.hpp b/cpp/include/raft/core/device_resources.hpp
index 496c65d91f..856ecc96d7 100644
--- a/cpp/include/raft/core/device_resources.hpp
+++ b/cpp/include/raft/core/device_resources.hpp
@@ -121,7 +121,7 @@ class device_resources : public resources {
 
   cusparseHandle_t get_cusparse_handle() const { return resource::get_cusparse_handle(*this); }
 
-  rmm::exec_policy& get_thrust_policy() const { return resource::get_thrust_policy(*this); }
+  rmm::exec_policy_nosync& get_thrust_policy() const { return resource::get_thrust_policy(*this); }
 
   /**
    * @brief synchronize a stream on the current container
diff --git a/cpp/include/raft/core/resource/thrust_policy.hpp b/cpp/include/raft/core/resource/thrust_policy.hpp
index f81898be8a..c728f0a00e 100644
--- a/cpp/include/raft/core/resource/thrust_policy.hpp
+++ b/cpp/include/raft/core/resource/thrust_policy.hpp
@@ -24,7 +24,7 @@ namespace raft::resource {
 class thrust_policy_resource : public resource {
  public:
   thrust_policy_resource(rmm::cuda_stream_view stream_view)
-    : thrust_policy_(std::make_unique<rmm::exec_policy>(stream_view))
+    : thrust_policy_(std::make_unique<rmm::exec_policy_nosync>(stream_view))
   {
   }
   void* get_resource() override { return thrust_policy_.get(); }
@@ -32,7 +32,7 @@ class thrust_policy_resource : public resource {
   ~thrust_policy_resource() override {}
 
  private:
-  std::unique_ptr<rmm::exec_policy> thrust_policy_;
+  std::unique_ptr<rmm::exec_policy_nosync> thrust_policy_;
 };
 
 /**
@@ -60,13 +60,13 @@ class thrust_policy_resource_factory : public resource_factory {
  * @param res raft res object for managing resources
  * @return thrust execution policy
  */
-inline rmm::exec_policy& get_thrust_policy(resources const& res)
+inline rmm::exec_policy_nosync& get_thrust_policy(resources const& res)
 {
   if (!res.has_resource_factory(resource_type::THRUST_POLICY)) {
     rmm::cuda_stream_view stream = get_cuda_stream(res);
     res.add_resource_factory(std::make_shared<thrust_policy_resource_factory>(stream));
   }
-  return *res.get_resource<rmm::exec_policy>(resource_type::THRUST_POLICY);
+  return *res.get_resource<rmm::exec_policy_nosync>(resource_type::THRUST_POLICY);
 };
 
 /**
diff --git a/cpp/include/raft/spectral/detail/matrix_wrappers.hpp b/cpp/include/raft/spectral/detail/matrix_wrappers.hpp
index 30dd6e5e69..1fe078bd32 100644
--- a/cpp/include/raft/spectral/detail/matrix_wrappers.hpp
+++ b/cpp/include/raft/spectral/detail/matrix_wrappers.hpp
@@ -129,7 +129,7 @@ class vector_t {
  private:
   using thrust_exec_policy_t =
     thrust::detail::execute_with_allocator<rmm::mr::thrust_allocator<char>,
-                                           thrust::cuda_cub::execute_on_stream_base>;
+                                           thrust::cuda_cub::execute_on_stream_nosync_base>;
   rmm::device_uvector<value_type> buffer_;
   const thrust_exec_policy_t thrust_policy;
 };