Fixes #687: Unified interface of regular and async variants of `memor…

…y::copy_single<T>()`
eyalroz · Oct 20, 2024 · 5c7b607 · 5c7b607
1 parent d601d29
commit 5c7b607
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/cuda/api/memory.hpp b/src/cuda/api/memory.hpp
@@ -1094,9 +1094,9 @@ void copy(const array_t<T, NumDimensions>&  destination, const T* source, stream
  * @param stream_handle A stream on which to enqueue the copy operation
  */
 template <typename T>
-void copy_single(T& destination, const T& source, stream::handle_t stream_handle)
+void copy_single(T* destination, const T* source, stream::handle_t stream_handle)
 {
-	copy(&destination, &source, sizeof(T), stream_handle);
+	copy(destination, source, sizeof(T), stream_handle);
 }
 
 } // namespace detail_
@@ -1364,7 +1364,7 @@ inline void copy(T(&destination)[N], const_region_t source, const stream_t& stre
  * @param stream The CUDA command queue on which this copying will be enqueued
  */
 template <typename T>
-void copy_single(T& destination, const T& source, const stream_t& stream);
+void copy_single(T* destination, const T* source, const stream_t& stream);
 
 } // namespace async
 

diff --git a/src/cuda/api/multi_wrapper_impls/memory.hpp b/src/cuda/api/multi_wrapper_impls/memory.hpp
@@ -81,9 +81,9 @@ inline void copy(span<T> destination, const array_t<T, NumDimensions>& source, c
 }
 
 template <typename T>
-inline void copy_single(T& destination, const T& source, const stream_t& stream)
+inline void copy_single(T* destination, const T* source, const stream_t& stream)
 {
-	detail_::copy_single(&destination, &source, sizeof(T), stream.handle());
+	detail_::copy_single(destination, source, sizeof(T), stream.handle());
 }
 
 } // namespace async