Remove function

rapidsai · Jul 8, 2021 · ac79888 · ac79888
1 parent e571293
commit ac79888
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/cpp/include/raft/handle.hpp b/cpp/include/raft/handle.hpp
@@ -140,14 +140,10 @@ class handle_t {
     return cusparse_handle_;
   }
 
-  thrust_exec_policy_t get_thrust_policy() const {
+  rmm::exec_policy get_thrust_policy() const {
     std::lock_guard<std::mutex> _(mutex_);
     if (!thrust_policy_initialized_) {
-      if (!thrust_policy_) {
-        thrust_policy_ =
-          (thrust_exec_policy_t*)malloc(sizeof(thrust_exec_policy_t));
-      }
-      *thrust_policy_ = rmm::exec_policy(this->get_stream());
+      thrust_policy_ = new rmm::exec_policy(get_stream());
       thrust_policy_initialized_ = true;
     }
     return *thrust_policy_;
@@ -240,7 +236,7 @@ class handle_t {
   mutable bool cusolver_sp_initialized_{false};
   mutable cusparseHandle_t cusparse_handle_;
   mutable bool cusparse_initialized_{false};
-  mutable thrust_exec_policy_t* thrust_policy_{nullptr};
+  mutable rmm::exec_policy* thrust_policy_{nullptr};
   mutable bool thrust_policy_initialized_{false};
   cudaStream_t user_stream_{nullptr};
   cudaEvent_t event_;