upsi.cpp File Reference

#include "upsi.h"
#include "source_base/module_container/ATen/kernels/blas.h"
#include "source_base/module_external/scalapack_connector.h"
#include <complex>
#include <iostream>

Include dependency graph for upsi.cpp:

Namespaces
namespace	module_rt

Functions
void	module_rt::upsi (const Parallel_Orbitals *pv, const int nband, const int nlocal, const std::complex< double > *U_operator, const std::complex< double > *psi_k_laststep, std::complex< double > *psi_k, std::ofstream &ofs_running, const int print_matrix)
	apply U_operator to the wave function of the previous step for new wave function
void	module_rt::upsi_tensor (const Parallel_Orbitals *pv, const int nband, const int nlocal, const ct::Tensor &U_operator, const ct::Tensor &psi_k_laststep, ct::Tensor &psi_k, std::ofstream &ofs_running, const int print_matrix)
template<typename Device >
void	module_rt::upsi_tensor_lapack (const Parallel_Orbitals *pv, const int nband, const int nlocal, const ct::Tensor &U_operator, const ct::Tensor &psi_k_laststep, ct::Tensor &psi_k, std::ofstream &ofs_running, const int print_matrix)
template void	module_rt::upsi_tensor_lapack< base_device::DEVICE_CPU > (const Parallel_Orbitals *pv, const int nband, const int nlocal, const ct::Tensor &U_operator, const ct::Tensor &psi_k_laststep, ct::Tensor &psi_k, std::ofstream &ofs_running, const int print_matrix)