device_memory_resource.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
 
// Local include(s).
#include "vecmem/memory/memory_resource.hpp"
#include "vecmem/vecmem_hip_export.hpp"
 
namespace vecmem::hip {
 
class device_memory_resource final : public memory_resource {
 
public:
    static constexpr int INVALID_DEVICE = -1;
 
    VECMEM_HIP_EXPORT
    device_memory_resource(int device = INVALID_DEVICE);
    VECMEM_HIP_EXPORT
    ~device_memory_resource();
 
private:
 
    VECMEM_HIP_EXPORT
    virtual void* do_allocate(std::size_t nbytes,
                              std::size_t alignment) override final;
 
    VECMEM_HIP_EXPORT
    virtual void do_deallocate(void* ptr, std::size_t nbytes,
                               std::size_t alignment) override final;
 
    VECMEM_HIP_EXPORT
    virtual bool do_is_equal(
        const memory_resource& other) const noexcept override final;
 
 
    const int m_device;
 
};  // class device_memory_resource
 
}  // namespace vecmem::hip