overset_types.H

Go to the documentation of this file.

#ifndef OVERSET_TYPES_H
#define OVERSET_TYPES_H
 
#include "AMReX_Gpu.H"
 
namespace kynema_sgf::tioga_iface {

template <typename T>
struct AmrDualArray
{
    using ArrayType = amrex::Gpu::DeviceVector<T>;
    using HostArrayType = amrex::Vector<T>;

    ArrayType d_view;

    HostArrayType h_view;
 
    AmrDualArray() : d_view(), h_view() {}

    explicit AmrDualArray(const size_t len) : d_view(len), h_view(len) {}

    void resize(const size_t len)
    {
        d_view.resize(len);
        h_view.resize(len);
    }

    [[nodiscard]] size_t size() const { return d_view.size(); }

    void copy_to_device()
    {
        amrex::Gpu::copy(
            amrex::Gpu::hostToDevice, h_view.begin(), h_view.end(),
            d_view.begin());
    }

    void copy_to_host()
    {
        amrex::Gpu::copy(
            amrex::Gpu::deviceToHost, d_view.begin(), d_view.end(),
            h_view.begin());
    }
};
 
} // namespace kynema_sgf::tioga_iface
 
#endif /* OVERSET_TYPES_H */