data_reader_common_catch2.hpp File Reference

#include <sys/stat.h>
#include <sys/types.h>
#include <unistd.h>
#include "lbann/data_readers/data_reader.hpp"
#include <lbann/base.hpp>
#include "lbann/proto/lbann.pb.h"
#include <google/protobuf/text_format.h>

Include dependency graph for data_reader_common_catch2.hpp:

Go to the source code of this file.

Functions
std::string	create_test_directory (std::string base_name)
std::map< lbann::execution_mode, lbann::generic_data_reader * >	instantiate_data_readers (std::string prototext_in, lbann::lbann_comm &comm_in, lbann::generic_data_reader *&train_ptr, lbann::generic_data_reader *&validate_ptr, lbann::generic_data_reader *&test_ptr, lbann::generic_data_reader *&tournament_ptr)
void	write_file (std::string data, std::string dir, std::string fn)

Function Documentation

create_test_directory()

std::string create_test_directory

(

std::string

base_name

)

create a directory in /tmp; returns the pathname to the directory

instantiate_data_readers()

std::map<lbann::execution_mode, lbann::generic_data_reader*> instantiate_data_readers	(	std::string	prototext_in,
		lbann::lbann_comm &	comm_in,
		lbann::generic_data_reader *&	train_ptr,
		lbann::generic_data_reader *&	validate_ptr,
		lbann::generic_data_reader *&	test_ptr,
		lbann::generic_data_reader *&	tournament_ptr
	)

Instantiates one or more data readers from the input 'prototext' string. Users should ensure that the appropriate options (if any) are set prior to calling this function, i.e: lbann::options *opts = lbann::options::get(); opts->set_option("preload_data_store", true);

write_file()

void write_file	(	std::string	data,
		std::string	dir,
		std::string	fn
	)