lbann::persist Class Reference

#include <persist.hpp>

Public Member Functions
	persist ()
	~persist ()
template<class Archive >
void	serialize (Archive &ar)
callback_type	get_cb_type () const
void	set_cb_type (callback_type type)
void	open_checkpoint_dir (const std::string &dir, bool create_dir)
void	open_checkpoint (const std::string &dir, bool create_dir)
void	close_checkpoint ()
void	open_restart (const std::string &dir)
void	close_restart ()
void	set_restart_dir (const std::string &dir)
uint64_t	get_bytes () const
void	reset_bytes ()
template<typename TensorDataType >
bool	write_rank_distmat (persist_type type, const char *name, const El::AbstractDistMatrix< TensorDataType > &M)
template<typename TensorDataType >
bool	read_rank_distmat (persist_type type, const char *name, El::AbstractDistMatrix< TensorDataType > &M)
template<typename TensorDataType >
bool	write_distmat (persist_type type, const char *name, El::AbstractDistMatrix< TensorDataType > *M)
template<typename TensorDataType >
bool	read_distmat (persist_type type, const char *name, El::AbstractDistMatrix< TensorDataType > *M)
const std::string &	get_checkpoint_dir () const
std::string	get_filename (persist_type type) const

Public Attributes
std::string	m_checkpoint_dir

Private Attributes
std::map< persist_type, uint64_t >	m_bytes
std::map< persist_type, std::string >	m_filenames
callback_type	ckpt_type

Detailed Description

Definition at line 72 of file persist.hpp.

Constructor & Destructor Documentation

persist()

lbann::persist::persist

(

)

~persist()

lbann::persist::~persist ( )

inline

Definition at line 84 of file persist.hpp.

Here is the call graph for this function:

Member Function Documentation

close_checkpoint()

void lbann::persist::close_checkpoint

(

)

close_restart()

void lbann::persist::close_restart

(

)

get_bytes()

uint64_t lbann::persist::get_bytes ( ) const

inline

Definition at line 102 of file persist.hpp.

get_cb_type()

callback_type lbann::persist::get_cb_type ( ) const

inline

Definition at line 90 of file persist.hpp.

get_checkpoint_dir()

const std::string& lbann::persist::get_checkpoint_dir ( ) const

inline

Definition at line 136 of file persist.hpp.

Here is the call graph for this function:

Here is the caller graph for this function:

get_filename()

std::string lbann::persist::get_filename

(

persist_type

type

)

const

Here is the caller graph for this function:

open_checkpoint()

void lbann::persist::open_checkpoint	(	const std::string &	dir,
		bool	create_dir
	)

open_checkpoint_dir()

void lbann::persist::open_checkpoint_dir	(	const std::string &	dir,
		bool	create_dir
	)

open_restart()

void lbann::persist::open_restart

(

const std::string &

dir

)

read_distmat()

template<typename TensorDataType >

bool lbann::persist::read_distmat	(	persist_type	type,
		const char *	name,
		El::AbstractDistMatrix< TensorDataType > *	M
	)

read_rank_distmat()

template<typename TensorDataType >

bool lbann::persist::read_rank_distmat	(	persist_type	type,
		const char *	name,
		El::AbstractDistMatrix< TensorDataType > &	M
	)

reset_bytes()

void lbann::persist::reset_bytes ( )

inline

Definition at line 111 of file persist.hpp.

serialize()

template<class Archive >

void lbann::persist::serialize

(

Archive &

ar

)

Archive for checkpoint and restart

Definition at line 91 of file persist_impl.hpp.

set_cb_type()

void lbann::persist::set_cb_type ( callback_type  type )

inline

Definition at line 92 of file persist.hpp.

Here is the call graph for this function:

set_restart_dir()

void lbann::persist::set_restart_dir ( const std::string &  dir )

inline

Definition at line 100 of file persist.hpp.

write_distmat()

template<typename TensorDataType >

bool lbann::persist::write_distmat	(	persist_type	type,
		const char *	name,
		El::AbstractDistMatrix< TensorDataType > *	M
	)

write_rank_distmat()

template<typename TensorDataType >

bool lbann::persist::write_rank_distmat	(	persist_type	type,
		const char *	name,
		const El::AbstractDistMatrix< TensorDataType > &	M
	)

Member Data Documentation

ckpt_type

callback_type lbann::persist::ckpt_type

private

Definition at line 77 of file persist.hpp.

m_bytes

std::map<persist_type, uint64_t> lbann::persist::m_bytes

private

Definition at line 75 of file persist.hpp.

m_checkpoint_dir

std::string lbann::persist::m_checkpoint_dir

Definition at line 80 of file persist.hpp.

m_filenames

std::map<persist_type, std::string> lbann::persist::m_filenames

private

Definition at line 76 of file persist.hpp.

---

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

• persist.hpp
• persist_impl.hpp