async_copy.hpp

/*
 * VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2021-2023 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"
 
namespace vecmem::cuda {
 
class async_copy : public vecmem::copy {
 
public:
    VECMEM_CUDA_EXPORT
    async_copy(const stream_wrapper& stream);
    VECMEM_CUDA_EXPORT
    ~async_copy();
 
protected:
    VECMEM_CUDA_EXPORT
    virtual void do_copy(std::size_t size, const void* from, void* to,
                         type::copy_type cptype) const override final;
    VECMEM_CUDA_EXPORT
    virtual void do_memset(std::size_t size, void* ptr,
                           int value) const override final;
    VECMEM_CUDA_EXPORT
    virtual event_type create_event() const override final;
 
private:
    stream_wrapper m_stream;
 
};  // class async_copy
 
}  // namespace vecmem::cuda