#include <PKmanagers.h>

Inheritance diagram for psi::pk::PKMgrYoshimine:

Public Member Functions
	PKMgrYoshimine (std::shared_ptr< PSIO > psio, std::shared_ptr< BasisSet > primary, size_t memory, Options &options)
	Constructor. More...

	~PKMgrYoshimine () override
	Destructor. More...

void	prestripe_files () override
	Pre-striping the IWL pre-sorted bucket files. More...

void	prestripe_files_wK () override
	Pre-striping for IWL wK pre-sorted files. More...

void	form_PK () override
	Gather all steps to form PK. More...

void	form_PK_wK () override
	Steps to form the supermatrix for wK. More...

void	allocate_buffers () override
	Allocating the buffers for each thread. More...

void	allocate_buffers_wK () override
	Allocating buffers for wK integrals. More...

void	compute_integrals (bool wK=false) override
	Computing the integrals. More...

void	compute_integrals_wK () override
	computing wK integrals More...

void	write () override
	Writing of the last partially filled buffers. More...

void	write_wK () override
	Writing of the last partially filled buffers for wK. More...

void	sort_ints (bool wK=false)
	Reading and sorting integrals to generate PK file. More...

void	sort_ints_wK ()
	Reading and sorting wK integrals for PK file. More...

void	close_iwl_buckets ()
	Close the IWL bucket files. More...

void	close_iwl_buckets_wK ()
	Close the IWL bucket file for wK. More...

void	generate_J_PK (double *twoel_ints, size_t max_size)
	Generate the J PK supermatrix from IWL integrals. More...

void	generate_K_PK (double *twoel_ints, size_t max_size)
	Generate the K PK supermatrix from IWL integrals. More...

void	generate_wK_PK (double *twoel_ints, size_t max_size)
	Generate the wK PK supermatrix from IWL integrals. More...

Public Member Functions inherited from psi::pk::PKMgrDisk
	PKMgrDisk (std::shared_ptr< PSIO > psio, std::shared_ptr< BasisSet > primary, size_t memory, Options &options)
	Constructor for PKMgrDisk. More...

	~PKMgrDisk () override
	Destructor for PKMgrDisk, does nothing. More...

void	set_writing (bool tmp)
	Setter/Getter functions. More...

bool	writing () const

int	pk_file () const

std::vector< size_t > &	batch_ind_min ()

std::vector< size_t > &	batch_ind_max ()

std::vector< size_t > &	batch_pq_min ()

std::vector< size_t > &	batch_pq_max ()

std::vector< int > &	batch_for_pq ()

std::shared_ptr< AIOHandler >	AIO () const

std::shared_ptr< PSIO >	psio () const

void	finalize_PK () override
	Finalize the PK file formation. More...

void	initialize () override
	Initialize sequence for Disk algorithms. More...

void	initialize_wK () override

void	prepare_JK (std::vector< SharedMatrix > D, std::vector< SharedMatrix > Cl, std::vector< SharedMatrix > Cr) override
	Prepare the JK formation for disk algorithms. More...

void	batch_sizing ()
	Determining the batch sizes. More...

void	print_batches () override
	Printing out the batches. More...

void	open_PK_file ()
	Opening the PK file. More...

virtual void	close_PK_file (bool keep)
	Closing the files. More...

void	form_J (std::vector< SharedMatrix > J, std::string exch="", std::vector< SharedMatrix > K=std::vector< SharedMatrix >()) override
	Form J from PK supermatrix, shared_ptr() initialized to null. More...

void	finalize_JK () override
	Finalize JK matrix formation. More...

Public Member Functions inherited from psi::pk::PKManager
	PKManager (std::shared_ptr< BasisSet > primary, size_t memory, Options &options)
	Base constructor. More...

virtual	~PKManager ()
	Base destructor, does nothing. More...

bool	do_wk () const
	Accessor functions for simple data. More...

double	omega () const

double	cutoff () const

int	nthreads () const

int	nbf () const

std::shared_ptr< ERISieve >	sieve () const

size_t	pk_pairs () const

size_t	pk_size () const

size_t	ntasks () const

size_t	memory () const

SharedPKWrkr	buffer (int i) const

double *	D_glob_vecs (int i) const

double *	JK_glob_vecs (int i) const

std::shared_ptr< BasisSet >	primary () const

bool	is_sym (int i) const

bool	all_sym () const

SharedMatrix	original_D (int N) const

SharedPKWrkr	get_buffer ()
	Accessor that returns buffer corresponding to current thread. More...

void	set_ntasks (size_t tmp)

virtual void	print_batches_wK ()
	Some printing for wK integrals;. More...

void	integrals_buffering (const double *buffer, size_t P, size_t Q, size_t R, size_t S)
	Store the computed integrals in the appropriate buffers. More...

void	integrals_buffering_wK (const double *buffer, size_t P, size_t Q, size_t R, size_t S)
	Store the computed wK integrals in the appropriate buffers. More...

void	form_D_vec (std::vector< SharedMatrix > D, std::vector< SharedMatrix > Cl, std::vector< SharedMatrix > Cr)

void	make_J_vec (std::vector< SharedMatrix > J)
	Preparing triangular vector for J/K. More...

void	get_results (std::vector< SharedMatrix > J, std::string exch)
	Extracting results from vectors to matrix. More...

void	form_K (std::vector< SharedMatrix > K)
	Forming K. More...

virtual void	form_wK (std::vector< SharedMatrix > wK)
	Forming wK. More...

void	finalize_D ()
	Finalize and delete the density matrix vectors. More...

Private Attributes
int	iwl_file_J_
	Files of pre-sorted IWL integral buckets. More...

int	iwl_file_K_

int	iwl_file_wK_

size_t	ints_per_buf_
	Number of integrals per buckets. More...

size_t	iwl_int_size_
	Total size of one IWL buffer on disk in bytes. More...

Additional Inherited Members
Static Public Member Functions inherited from psi::pk::PKManager
static std::shared_ptr< PKManager >	build_PKManager (std::shared_ptr< PSIO > psio, std::shared_ptr< BasisSet > primary, size_t memory, Options &options, bool dowK, double omega_in=0)
	build_PKManager Static instance constructor, used to get a proper instance of PKManager through automatic selection and options provided More...

Protected Member Functions inherited from psi::pk::PKManager
void	fill_buffer (SharedPKWrkr tmp)
	Setter objects for internal data. More...

Detailed Description

Yoshimine sorting algorithm: the integrals are divided in N batches that can individually fit in memory. Then, all integrals are computed only once, and sorted in N little buffers, one for each batch. As soon as a buffer is full, it gets dumped onto one of N temporary files on disk in IWL format.

Once all integrals are computed and pre-sorted in the N IWL files, each file is read and its integrals sorted, then written onto the PK files.

This routine takes advantage of OMP parallelization, then each thread has N little buffers. All disk writing is handled by a dedicated thread generated by AIOHandler

Constructor & Destructor Documentation

psi::pk::PKMgrYoshimine::PKMgrYoshimine	(	std::shared_ptr< PSIO >	psio,
		std::shared_ptr< BasisSet >	primary,
		size_t	memory,
		Options &	options
	)

Constructor.

psi::pk::PKMgrYoshimine::~PKMgrYoshimine ( )

inlineoverride

Destructor.

Member Function Documentation

void psi::pk::PKMgrYoshimine::allocate_buffers ( )

overridevirtual

Allocating the buffers for each thread.

Implements psi::pk::PKManager.

void psi::pk::PKMgrYoshimine::allocate_buffers_wK ( )

overridevirtual

Allocating buffers for wK integrals.

Implements psi::pk::PKMgrDisk.

void psi::pk::PKMgrYoshimine::close_iwl_buckets ( )

Close the IWL bucket files.

void psi::pk::PKMgrYoshimine::close_iwl_buckets_wK ( )

Close the IWL bucket file for wK.

void psi::pk::PKMgrYoshimine::compute_integrals ( bool wK = false )

overridevirtual

Computing the integrals.

Reimplemented from psi::pk::PKManager.

void psi::pk::PKMgrYoshimine::compute_integrals_wK ( )

overridevirtual

computing wK integrals

Reimplemented from psi::pk::PKManager.

void psi::pk::PKMgrYoshimine::form_PK ( )

overridevirtual

Gather all steps to form PK.

Implements psi::pk::PKManager.

void psi::pk::PKMgrYoshimine::form_PK_wK ( )

overridevirtual

Steps to form the supermatrix for wK.

Implements psi::pk::PKManager.

void psi::pk::PKMgrYoshimine::generate_J_PK	(	double *	twoel_ints,
		size_t	max_size
	)

Generate the J PK supermatrix from IWL integrals.

void psi::pk::PKMgrYoshimine::generate_K_PK	(	double *	twoel_ints,
		size_t	max_size
	)

Generate the K PK supermatrix from IWL integrals.

void psi::pk::PKMgrYoshimine::generate_wK_PK	(	double *	twoel_ints,
		size_t	max_size
	)

Generate the wK PK supermatrix from IWL integrals.

void psi::pk::PKMgrYoshimine::prestripe_files ( )

overridevirtual

Pre-striping the IWL pre-sorted bucket files.

Implements psi::pk::PKMgrDisk.

void psi::pk::PKMgrYoshimine::prestripe_files_wK ( )

overridevirtual

Pre-striping for IWL wK pre-sorted files.

Implements psi::pk::PKMgrDisk.

void psi::pk::PKMgrYoshimine::sort_ints ( bool wK = false )

Reading and sorting integrals to generate PK file.

void psi::pk::PKMgrYoshimine::sort_ints_wK ( )

Reading and sorting wK integrals for PK file.

void psi::pk::PKMgrYoshimine::write ( )

overridevirtual

Writing of the last partially filled buffers.

Reimplemented from psi::pk::PKMgrDisk.

void psi::pk::PKMgrYoshimine::write_wK ( )

overridevirtual

Writing of the last partially filled buffers for wK.

Reimplemented from psi::pk::PKMgrDisk.

Member Data Documentation

size_t psi::pk::PKMgrYoshimine::ints_per_buf_

private

Number of integrals per buckets.

int psi::pk::PKMgrYoshimine::iwl_file_J_

private

Files of pre-sorted IWL integral buckets.

int psi::pk::PKMgrYoshimine::iwl_file_K_

private

int psi::pk::PKMgrYoshimine::iwl_file_wK_

private

size_t psi::pk::PKMgrYoshimine::iwl_int_size_

private

Total size of one IWL buffer on disk in bytes.

The documentation for this class was generated from the following files:

Public Member Functions

Private Attributes

Additional Inherited Members

Detailed Description

Constructor & Destructor Documentation

Member Function Documentation

Member Data Documentation