ABACUS develop
Atomic-orbital Based Ab-initio Computation at UStc
Loading...
Searching...
No Matches
Namespaces | Typedefs | Functions
lr_util_print.h File Reference
#include <ATen/tensor.h>
#include "source_psi/psi.h"
Include dependency graph for lr_util_print.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Namespaces

namespace  LR_Util
 

Typedefs

using LR_Util::TC = std::array< int, 3 >
 
using LR_Util::TAC = std::pair< int, TC >
 
template<typename T >
using LR_Util::TLRIX = std::map< int, std::map< TAC, std::vector< T > > >
 

Functions

template<typename T >
constexpr T LR_Util::filter (const T &v, const double &threshold=1e-10)
 
template<typename T >
int LR_Util::read_value (std::ifstream &ifs, T *ptr, const int &size)
 
template<typename T , typename... Args>
int LR_Util::read_value (std::ifstream &ifs, T *ptr, const int &size, Args &&... args)
 
template<typename T , typename... Args>
int LR_Util::read_value (const std::string &file, T *ptr, const int &size, Args &&... args)
 
template<typename T >
int LR_Util::write_value (std::ofstream &ofs, const T *ptr, const int &size)
 
template<typename T , typename... Args>
int LR_Util::write_value (std::ofstream &ofs, const T *ptr, const int &size, Args &&... args)
 
template<typename T , typename... Args>
int LR_Util::write_value (const std::string &file, const int &prec, const T *ptr, const int &size, Args &&... args)
 
template<typename T >
int LR_Util::print_value (const T *ptr, const int &size)
 
template<typename T , typename... Args>
int LR_Util::print_value (const T *ptr, const int &size, Args &&... args)
 
template<typename T >
void LR_Util::print_psi_bandfirst (const psi::Psi< T > &psi, const std::string &label, const int &ib, const double &threshold=1e-10)
 
template<typename T >
void LR_Util::write_psi_bandfirst (const psi::Psi< T > &psi, const std::string &filename, const int &rank, const double &threshold=1e-10, const int &precision=8)
 
template<typename T >
void LR_Util::write_psi_bandfirst (const T *psi, const int &nband, const int &nk, const int &nbasis, const std::string &filename, const int &rank, const double &threshold=1e-10, const int &precision=8)
 
template<typename T >
psi::Psi< TLR_Util::read_psi_bandfirst (const std::string &filename, const int &rank)
 
template<typename T >
void LR_Util::print_psi_kfirst (const psi::Psi< T > &psi, const std::string &label, const double &threshold=1e-10)
 
template<typename T >
void LR_Util::print_tensor (const container::Tensor &t, const std::string &label, const Parallel_2D *pmat, const double &threshold=1e-10)
 
template<typename T >
void LR_Util::print_grid_nonzero (T *rho, const int &nrxx, const int &nnz, const std::string &label, const double &threshold=1e-5)
 
template<typename T >
void LR_Util::print_CsX (const TLRIX< T > &CsX, const int nvirt, const std::string &label, const double &threshold=1e-10)