Explicitly swap the pointers on container move.

rapidsai · Nov 1, 2024 · e0068cd · e0068cd
1 parent 281a6f9
commit e0068cd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/cpp/include/raft/core/pinned_container_policy.hpp b/cpp/include/raft/core/pinned_container_policy.hpp
@@ -62,8 +62,8 @@ struct pinned_container {
     if (data_ != nullptr) { RAFT_CUDA_TRY_NO_THROW(cudaFreeHost(data_)); }
   }
 
-  pinned_container(pinned_container&&)            = default;
-  pinned_container& operator=(pinned_container&&) = default;
+  pinned_container(pinned_container&& other) { std::swap(this->data_, other.data_); }
+  pinned_container& operator=(pinned_container&& other) { std::swap(this->data_, other.data_); }
   pinned_container(pinned_container const&) = delete;  // Copying disallowed: one array one owner
   pinned_container& operator=(pinned_container const&) = delete;