stream_wrapper.hpp

/* VecMem project, part of the ACTS project (R&D line)
 *
 * (c) 2021-2025 CERN for the benefit of the ACTS project
 *
 * Mozilla Public License Version 2.0
 */
#pragma once
 
// Local include(s).
#include "vecmem/vecmem_cuda_export.hpp"
 
// System include(s).
#include <memory>
#include <string>
 
namespace vecmem {
namespace cuda {
 
class stream_wrapper {
 
public:
    static constexpr int INVALID_DEVICE = -1;
 
    VECMEM_CUDA_EXPORT
    explicit stream_wrapper(int device = INVALID_DEVICE);
    VECMEM_CUDA_EXPORT
    stream_wrapper(void* stream);
 
    VECMEM_CUDA_EXPORT
    stream_wrapper(const stream_wrapper& parent);
    VECMEM_CUDA_EXPORT
    stream_wrapper(stream_wrapper&& parent) noexcept;
 
    VECMEM_CUDA_EXPORT
    ~stream_wrapper();
 
    VECMEM_CUDA_EXPORT
    stream_wrapper& operator=(const stream_wrapper& rhs);
    VECMEM_CUDA_EXPORT
    stream_wrapper& operator=(stream_wrapper&& rhs) noexcept;
 
    VECMEM_CUDA_EXPORT
    void* stream() const;
 
    VECMEM_CUDA_EXPORT
    void synchronize();
 
    VECMEM_CUDA_EXPORT
    std::string device_name() const;
 
private:
    struct impl;
    std::unique_ptr<impl> m_impl;
 
};  // class stream_wrapper
 
}  // namespace cuda
}  // namespace vecmem