ekinetic_pw.h

Go to the documentation of this file.

#ifndef EKINETICPW_H
#define EKINETICPW_H
 
#include "operator_pw.h"
#include "source_pw/module_pwdft/kernels/ekinetic_op.h"
 
#include <source_base/macros.h>
 
namespace hamilt {
 
// Not needed anymore
#ifndef __EKINETICTEMPLATE
#define __EKINETICTEMPLATE
 
template<class T> class Ekinetic : public T {};
// template<typename R, typename Device = base_device::DEVICE_CPU>
// class Ekinetic : public OperatorPW<T, Device> {};
 
#endif
 
// template<typename R, typename Device = base_device::DEVICE_CPU>
// class Ekinetic : public OperatorPW<T, Device>
template<typename T, typename Device>
class Ekinetic<OperatorPW<T, Device>> : public OperatorPW<T, Device>
{
  private: 
    using Real = typename GetTypeReal<T>::type;
  public:
    Ekinetic(
        Real tpiba2_in, 
        const Real* gk2_in,
        const int gk2_row, 
        const int gk2_col);
 
    template<typename T_in, typename Device_in = Device>
    explicit Ekinetic(const Ekinetic<OperatorPW<T_in, Device_in>>* ekinetic);
 
    virtual ~Ekinetic();
 
    virtual void act(const int nbands,
        const int nbasis,
        const int npol,
        const T* tmpsi_in,
        T* tmhpsi,
        const int ngk_ik = 0,
        const bool is_first_node = false)const override;
 
    // denghuilu added for copy construct at 20221105
    int get_gk2_row() const {return this->gk2_row;}
    int get_gk2_col() const {return this->gk2_col;}
    Real get_tpiba2() const {return this->tpiba2;}
    const Real* get_gk2() const {return this->gk2;}
    Device* get_ctx() const {return this->ctx;}
 
  private:
 
    Real tpiba2 = 0.0;
    const Real* gk2 = nullptr;
    int gk2_row = 0;
    int gk2_col = 0;
 
    Device* ctx = {};
    base_device::DEVICE_CPU* cpu_ctx = {};
    base_device::AbacusDevice_t device = {};
 
    using ekinetic_op = ekinetic_pw_op<Real, Device>;
    using resmem_var_op = base_device::memory::resize_memory_op<Real, Device>;
    using delmem_var_op = base_device::memory::delete_memory_op<Real, Device>;
    using syncmem_var_h2d_op = base_device::memory::synchronize_memory_op<Real, Device, base_device::DEVICE_CPU>;
};
 
} // namespace hamilt
 
#endif