async_copy.hpp

/*
 * VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2021-2026 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// VecMem include(s).
#include "vecmem/utils/copy.hpp"
#include "vecmem/utils/cuda/stream_wrapper.hpp"
#include "vecmem/vecmem_cuda_export.hpp"
 
// System include(s).
#include <memory>
 
namespace vecmem::cuda {
 
class async_copy : public vecmem::copy {
 
public:
    VECMEM_CUDA_EXPORT
    explicit async_copy(const stream_wrapper& stream);
    VECMEM_CUDA_EXPORT
    async_copy(const stream_wrapper& stream, unsigned int event_flags);
    VECMEM_CUDA_EXPORT
    async_copy(async_copy&&) noexcept;
    VECMEM_CUDA_EXPORT
    ~async_copy() noexcept override;
 
    VECMEM_CUDA_EXPORT
    async_copy& operator=(async_copy&&) noexcept;
 
private:
    VECMEM_CUDA_EXPORT
    void do_copy(std::size_t size, const void* from, void* to,
                 type::copy_type cptype) const final;
    VECMEM_CUDA_EXPORT
    void do_memset(std::size_t size, void* ptr, int value) const final;
    VECMEM_CUDA_EXPORT
    event_type create_event() const final;
 
    struct impl;
    std::unique_ptr<impl> m_impl;
 
};  // class async_copy
 
}  // namespace vecmem::cuda