LCAO_deepks_io.cpp File Reference

#include "source_io/module_parameter/parameter.h"
#include "LCAO_deepks_io.h"
#include "source_base/tool_quit.h"
#include "npy.hpp"
#include <mpi.h>

Include dependency graph for LCAO_deepks_io.cpp:

Functions
template void	LCAO_deepks_io::print_dm< std::complex< double > > (const int nks, const int nlocal, const int nrow, const std::vector< std::vector< std::complex< double > > > &dm)
template void	LCAO_deepks_io::save_npy_h< std::complex< double > > (const std::vector< ModuleBase::ComplexMatrix > &hamilt, const std::string &h_file, const int nlocal, const int nks, const int rank, const double unit_scale)
template void	LCAO_deepks_io::save_tensor2npy< std::complex< double > > (const std::string &file_name, const torch::Tensor &tensor, const int rank)

Function Documentation

LCAO_deepks_io::print_dm< std::complex< double > >()

template void LCAO_deepks_io::print_dm< std::complex< double > >	(	const int	nks,
		const int	nlocal,
		const int	nrow,
		const std::vector< std::vector< std::complex< double > > > &	dm
	)

LCAO_deepks_io::save_npy_h< std::complex< double > >()

template void LCAO_deepks_io::save_npy_h< std::complex< double > >	(	const std::vector< ModuleBase::ComplexMatrix > &	hamilt,
		const std::string &	h_file,
		const int	nlocal,
		const int	nks,
		const int	rank,
		const double	unit_scale
	)

LCAO_deepks_io::save_tensor2npy< std::complex< double > >()

template void LCAO_deepks_io::save_tensor2npy< std::complex< double > >	(	const std::string &	file_name,
		const torch::Tensor &	tensor,
		const int	rank
	)