ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
|
#include <bpcg_kernel_op.h>
Public Member Functions | |
void | operator() (T *grad_out, T *hgrad_out, T *psi_out, T *hpsi_out, const int &n_basis, const int &n_basis_max, const int &n_band) |
dot_real_op computes the dot product of the given complex arrays(treated as float arrays). And there's may have MPI communications while enabling planewave parallization strategy. | |
void hsolver::line_minimize_with_block_op< T, Device >::operator() | ( | T * | grad_out, |
T * | hgrad_out, | ||
T * | psi_out, | ||
T * | hpsi_out, | ||
const int & | n_basis, | ||
const int & | n_basis_max, | ||
const int & | n_band | ||
) |
dot_real_op computes the dot product of the given complex arrays(treated as float arrays). And there's may have MPI communications while enabling planewave parallization strategy.
Input Parameters
dev | : the type of computing device |
A | : input array arr |
dim | : size of A |
lda | : leading dimention of A |
batch | : batch size, the size of the result array res |