meta_op.h

Go to the documentation of this file.

#ifndef MODULE_HAMILT_META_H
#define MODULE_HAMILT_META_H
 
#include "source_base/module_device/types.h"
#include "source_psi/psi.h"
#include <complex>
 
namespace hamilt {
    template <typename FPTYPE, typename Device>
    struct meta_pw_op {
        void operator() (
                const Device* dev,
                const int& ik,
                const int& pol,
                const int& npw,
                const int& npwx,
                const FPTYPE& tpiba,
                const FPTYPE* gcar,
                const FPTYPE* kvec_c,
                const std::complex<FPTYPE>* in,
                std::complex<FPTYPE>* out,
                const bool add = false);
    };
 
#if __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM
    // Partially specialize functor for base_device::GpuDevice.
    template <typename FPTYPE>
    struct meta_pw_op<FPTYPE, base_device::DEVICE_GPU>
    {
        void operator()(const base_device::DEVICE_GPU* dev,
                        const int& ik,
                        const int& pol,
                        const int& npw,
                        const int& npwx,
                        const FPTYPE& tpiba,
                        const FPTYPE* gcar,
                        const FPTYPE* kvec_c,
                        const std::complex<FPTYPE>* in,
                        std::complex<FPTYPE>* out,
                        const bool add = false);
    };
#endif // __CUDA || __UT_USE_CUDA || __ROCM || __UT_USE_ROCM
} // namespace hamilt
#endif //MODULE_HAMILT_META_H