Model_trainer

Executes a set of different contractions on different processor counts to train model parameters. More...

Collaboration diagram for Model_trainer:

Namespaces
	CTF_int

Macros
#define	TEST_SUITE

Functions
void	train_off_vec_mat (int64_t n, int64_t m, World &dw, bool sp_A, bool sp_B, bool sp_C)
void	train_ttm (int64_t sz, int64_t r, World &dw)
void	train_dns_vec_mat (int64_t n, int64_t m, World &dw)
void	train_sps_vec_mat (int64_t n, int64_t m, World &dw, bool sp_A, bool sp_B, bool sp_C)
void	train_ccsd (int64_t n, int64_t m, World &dw)
void	train_sparse_mp3 (int64_t n, int64_t m, World &dw)
void	train_world (double dtime, World &dw, double step_size)
void	frize (std::set< int > &ps, int p)
void	train_all (double time, bool write_coeff, bool dump_data, std::string coeff_file, std::string data_dir)
char *	getCmdOption (char **begin, char **end, const std::string &option)
int	main (int argc, char **argv)

Detailed Description

Executes a set of different contractions on different processor counts to train model parameters.

Macro Definition Documentation

#define TEST_SUITE

Definition at line 10 of file model_trainer.cxx.

Function Documentation

void frize	(	std::set< int > &	ps,
		int	p
	)

Definition at line 280 of file model_trainer.cxx.

char* getCmdOption	(	char **	begin,
		char **	end,
		const std::string &	option
	)

Definition at line 377 of file model_trainer.cxx.

Referenced by main().

int main	(	int	argc,
		char **	argv
	)

Definition at line 388 of file model_trainer.cxx.

References getCmdOption(), ctf.core::np(), ctf.core::rank(), ctf.core::string, and train_all().

void train_all	(	double	time,
		bool	write_coeff,
		bool	dump_data,
		std::string	coeff_file,
		std::string	data_dir
	)

Definition at line 289 of file model_trainer.cxx.

References CTF::World::comm, CTF_int::dump_all_models(), CTF::World::np, ctf.core::np(), ctf.core::rank(), train_world(), CTF_int::update_all_models(), ctf.core::w, and CTF_int::write_all_models().

Referenced by main().

void train_ccsd	(	int64_t	n,
		int64_t	m,
		World &	dw
	)

Definition at line 200 of file model_trainer.cxx.

References Amplitudes::abij, Amplitudes::ai, ccsd(), Integrals::fill_rand(), Amplitudes::fill_rand(), CTF::Tensor< dtype >::norm2(), CTF::Timer::start(), and CTF::Timer::stop().

Referenced by train_world().

void train_dns_vec_mat	(	int64_t	n,
		int64_t	m,
		World &	dw
	)

Definition at line 86 of file model_trainer.cxx.

References ctf.core::a, AS, ctf.core::b, f2(), CTF::Tensor< dtype >::fill_random(), CTF::World::rank, CTF::Timer::start(), CTF::Timer::stop(), and SY.

Referenced by train_world().

void train_off_vec_mat	(	int64_t	n,
		int64_t	m,
		World &	dw,
		bool	sp_A,
		bool	sp_B,
		bool	sp_C
	)

Definition at line 23 of file model_trainer_kernels.cxx.

References ctf.core::a, ctf.core::b, CTF::Tensor< dtype >::fill_random(), grp::op2_red(), grp::op2_t2(), CTF::World::rank, and CTF::Tensor< dtype >::sparsify().

Referenced by train_world().

void train_sparse_mp3	(	int64_t	n,
		int64_t	m,
		World &	dw
	)

Definition at line 216 of file model_trainer.cxx.

References sparse_mp3(), CTF::Timer::start(), and CTF::Timer::stop().

Referenced by train_world().

void train_sps_vec_mat	(	int64_t	n,
		int64_t	m,
		World &	dw,
		bool	sp_A,
		bool	sp_B,
		bool	sp_C
	)

Definition at line 142 of file model_trainer.cxx.

References ctf.core::a, ctf.core::b, f2(), CTF::Tensor< dtype >::fill_sp_random(), NS, CTF::World::rank, CTF::Timer::start(), and CTF::Timer::stop().

Referenced by train_world().

void train_ttm	(	int64_t	sz,
		int64_t	r,
		World &	dw
	)

Definition at line 22 of file model_trainer.cxx.

References CTF::Tensor< dtype >::fill_random(), CTF::Timer::start(), and CTF::Timer::stop().

Referenced by train_world().

void train_world	(	double	dtime,
		World &	dw,
		double	step_size
	)

Definition at line 230 of file model_trainer.cxx.

References CTF::World::comm, train_ccsd(), train_dns_vec_mat(), train_off_vec_mat(), train_sparse_mp3(), train_sps_vec_mat(), and train_ttm().

Referenced by train_all().