QedWrapperCommons.H

Go to the documentation of this file.

/* Copyright 2019 Luca Fedeli
 *
 * This file is part of WarpX.
 *
 * License: BSD-3-Clause-LBNL
 */
#ifndef WARPX_amrex_qed_wrapper_commons_h_
#define WARPX_amrex_qed_wrapper_commons_h_
/*
 * This header contains some common #define directives
 * used by the QED library wrappers and related components.
 * It also provides a vector type suitable to be used
 * as the building block  of the picsar QED lookup tables
 * when managed memory is not used.
 */

#include <AMReX_AmrCore.H>
#include <AMReX_Extension.H>
#include <AMReX_GpuDevice.H>
#include <AMReX_GpuQualifiers.H>
#include <AMReX_Vector.H>

#include <vector>


#define PXRMP_WITH_GPU
#define PXRMP_GPU_QUALIFIER AMREX_GPU_HOST_DEVICE
//_________________________


#define PXRMP_RESTRICT AMREX_RESTRICT
#define PXRMP_FORCE_INLINE AMREX_FORCE_INLINE
//_________________________


#ifdef AMREX_USE_GPU

template <typename Real>
class PicsarQedVector : public std::vector<Real>
{
    using HV = std::vector<Real>;
    using AGDV = amrex::Gpu::DeviceVector<Real>;

    public:

    template<typename... Args>
    PicsarQedVector(Args&&... args) : HV(std::forward<Args>(args)...)
    {}

    void pxr_sync()
    {
        m_device_data.resize(this->size());
        amrex::Gpu::copyAsync(amrex::Gpu::hostToDevice,
            this->begin(), this->end(), m_device_data.begin());
        amrex::Gpu::Device::streamSynchronize();
    }

    const Real* data() const
    {
        return m_device_data.data();
    }

    private:
    AGDV m_device_data;
};
#else

template <typename Real>
using PicsarQedVector = std::vector<Real>;

#endif

#endif //WARPX_amrex_qed_wrapper_commons_h_