diago_cg_float_test.cpp File Reference

#include "gtest/gtest.h"
#include "source_io/module_parameter/parameter.h"
#include "source_base/inverse_matrix.h"
#include "source_base/module_external/lapack_connector.h"
#include "source_pw/module_pwdft/structure_factor.h"
#include "source_psi/psi.h"
#include "source_hamilt/hamilt.h"
#include "source_pw/module_pwdft/hamilt_pw.h"
#include "../diago_cg.h"
#include "../diago_iter_assist.h"
#include "diago_mock.h"
#include "mpi.h"
#include "source_basis/module_pw/test/test_tool.h"
#include <complex>
#include <random>
#include <ATen/core/tensor_map.h>

Include dependency graph for diago_cg_float_test.cpp:

Classes
class	DiagoCGPrepare
class	DiagoCGFloatTest

Macros
#define	private   public

Functions
void	lapackEigen (int &npw, std::vector< std::complex< float > > &hm, float *e, bool outtime=false)
	TEST_P (DiagoCGFloatTest, RandomHamilt)
	INSTANTIATE_TEST_SUITE_P (VerifyCG, DiagoCGFloatTest, ::testing::Values(DiagoCGPrepare(10, 200, 0, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 200, 6, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 400, 8, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 600, 8, true, 1e-6, 300, 1e-0)))
	TEST (DiagoCGFloatTest, Hamilt)
	TEST (DiagoCGFloatTest, readH)
int	main (int argc, char **argv)

Macro Definition Documentation

private

#define private   public

Function Documentation

INSTANTIATE_TEST_SUITE_P()

INSTANTIATE_TEST_SUITE_P	(	VerifyCG	,
		DiagoCGFloatTest	,
		::testing::Values(DiagoCGPrepare(10, 200, 0, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 200, 6, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 400, 8, true, 1e-6, 300, 1e-0), DiagoCGPrepare(10, 600, 8, true, 1e-6, 300, 1e-0))
	)

lapackEigen()

void lapackEigen	(	int &	npw,
		std::vector< std::complex< float > > &	hm,
		float *	e,
		bool	outtime = false
	)

Class Diago_CG is an approach for eigenvalue problems This unittest test the function Diago_CG::diag() for FPTYPE=float, Device=cpu with different examples.

• the Hermite matrices (npw=500,1000) produced using random numbers and with sparsity of 0%, 60%, 80%
• the Hamiltonian matrix read from "data-H", produced by using out_hs in INPUT of a LCAO calculation
• a 2x2 Hermite matrix for learning and checking

Note: The test is passed when the eignvalues are closed to these calculated by LAPACK. It is used together with a header file diago_mock.h. The default Hermite matrix generated here is real symmetric, one can add an imaginary part by changing two commented out lines in diago_mock.h.

Here is the call graph for this function:

main()

int main	(	int	argc,
		char **	argv
	)

Here is the call graph for this function:

TEST() [1/2]

TEST	(	DiagoCGFloatTest	,
		Hamilt
	)

Here is the call graph for this function:

TEST() [2/2]

TEST	(	DiagoCGFloatTest	,
		readH
	)

Here is the call graph for this function:

TEST_P()

TEST_P	(	DiagoCGFloatTest	,
		RandomHamilt
	)

Here is the call graph for this function: