gatherchi_op< FPTYPE, Device > Struct Template Reference

gather chi from all processors More...

#include <sto_tool.h>

Collaboration diagram for gatherchi_op< FPTYPE, Device >:

Public Member Functions
psi::Psi< std::complex< FPTYPE >, Device > *	operator() (psi::Psi< std::complex< FPTYPE >, Device > &chi, psi::Psi< std::complex< FPTYPE >, Device > &chi_all, const int &npwx, int *nrecv_sto, int *displs_sto, const int perbands_sto)

Detailed Description

template<typename FPTYPE, typename Device>
struct gatherchi_op< FPTYPE, Device >

gather chi from all processors

Parameters

chi	stochasitc wave function of this processor
chi_all	gathered stochastic wave function
npwx	maximum number of plane waves on all processors
nrecv_sto	number of stochastic orbitals on each processor
displs_sto	displacement of stochastic orbitals on each processor
perbands_sto	number of stochastic bands of this processor

Returns

psi::Psi<std::complex<float>> pointer to gathered stochastic wave function

Member Function Documentation

operator()()

template<typename FPTYPE , typename Device >

psi::Psi< std::complex< FPTYPE >, Device > * gatherchi_op< FPTYPE, Device >::operator()	(	psi::Psi< std::complex< FPTYPE >, Device > &	chi,
		psi::Psi< std::complex< FPTYPE >, Device > &	chi_all,
		const int &	npwx,
		int *	nrecv_sto,
		int *	displs_sto,
		const int	perbands_sto
	)

Here is the call graph for this function:

---

The documentation for this struct was generated from the following files:

• /home/runner/work/abacus-develop/abacus-develop/source/source_pw/module_stodft/sto_tool.h
• /home/runner/work/abacus-develop/abacus-develop/source/source_pw/module_stodft/sto_tool.cpp