vecmem 1.21.0
Loading...
Searching...
No Matches
async_copy.hpp
1/*
2 * VecMem project, part of the ACTS project (R&D line)
3 *
4 * (c) 2021-2025 CERN for the benefit of the ACTS project
5 *
6 * Mozilla Public License Version 2.0
7 */
8#pragma once
9
10// VecMem include(s).
11#include "vecmem/utils/copy.hpp"
12#include "vecmem/utils/cuda/stream_wrapper.hpp"
13#include "vecmem/vecmem_cuda_export.hpp"
14
15namespace vecmem::cuda {
16
27class async_copy : public vecmem::copy {
28
29public:
32 explicit async_copy(const stream_wrapper& stream);
36
40 void do_copy(std::size_t size, const void* from, void* to,
41 type::copy_type cptype) const final;
44 void do_memset(std::size_t size, void* ptr, int value) const final;
48
50 stream_wrapper m_stream;
51
52}; // class async_copy
53
54} // namespace vecmem::cuda
An allocator class that wraps a memory resource.
Definition allocator.hpp:37
Class implementing (synchronous) host <-> device memory copies.
Definition copy.hpp:42
std::unique_ptr< abstract_event > event_type
Event type used by the copy class.
Definition copy.hpp:70
data::vector_buffer< std::remove_cv_t< TYPE > > to(const data::vector_view< TYPE > &data, memory_resource &resource, type::copy_type cptype=type::unknown) const
Copy a 1-dimensional vector to the specified memory resource.
Definition copy.ipp:66
Specialisation of vecmem::copy for CUDA.
Definition async_copy.hpp:27
VECMEM_CUDA_EXPORT event_type create_event() const final
Create an event for synchronization.
Definition async_copy.cpp:157
VECMEM_CUDA_EXPORT void do_memset(std::size_t size, void *ptr, int value) const final
Fill a memory area using CUDA asynchronously.
Definition async_copy.cpp:136
VECMEM_CUDA_EXPORT void do_copy(std::size_t size, const void *from, void *to, type::copy_type cptype) const final
Perform an asynchronous memory copy using CUDA.
Definition async_copy.cpp:109
VECMEM_CUDA_EXPORT ~async_copy() noexcept override
Destructor.
Wrapper class for cudaStream_t.
Definition stream_wrapper.hpp:24
Namespace holding types that work on/with CUDA.
Definition cuda_device_atomic_ref.hpp:19
Wrapper struct around the copy_type enumeration.
Definition copy.hpp:51