hcontainer_funcs.h

Go to the documentation of this file.

#pragma once
 
#include "source_lcao/module_hcontainer/hcontainer.h"
 
namespace hamilt
{
template<typename TR>
void folding_HR(const hamilt::HContainer<TR>& hR,
                std::complex<double>* hk,
                const ModuleBase::Vector3<double>& kvec_d_in,
                const int hk_ld,
                const int hk_type);
 
void folding_HR(const hamilt::HContainer<double>& hR,
                double* hk,
                const ModuleBase::Vector3<double>& kvec_d_in,
                const int hk_ld,
                const int hk_type);
 
#ifdef __MPI
template<typename TR>
void transferSerial2Parallels(const hamilt::HContainer<TR>& hR_s,
                             hamilt::HContainer<TR>* hR_p,
                             const int serial_rank);
template<typename TR>
void transferParallels2Serial(const hamilt::HContainer<TR>& hR_p,
                             hamilt::HContainer<TR>* hR_s,
                             const int serial_rank);
 
template<typename TR>
void transferSerials2Parallels(const hamilt::HContainer<TR>& hR_s,
                             hamilt::HContainer<TR>* hR_p);
 
template<typename TR>
void transferParallels2Serials(const hamilt::HContainer<TR>& hR_p,
                             hamilt::HContainer<TR>* hR_s);
 
template<typename TR>
void gatherParallels(const hamilt::HContainer<TR>& hR_p,
                     hamilt::HContainer<TR>* hR_s,
                     const int serial_rank);
 
#endif // __MPI
 
} // namespace hamilt