LBANN  0.103.0
LivermoreBigArtificialNeuralNetworkToolkit
serialize_matrices.hpp File Reference
#include "cereal_utils.hpp"
#include "rooted_archive_adaptor.hpp"
#include <lbann/utils/exception.hpp>
#include <El.hpp>
#include <stdexcept>
Include dependency graph for serialize_matrices.hpp:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes

struct  lbann::utils::grid_manager
 RAII grid management. More...
 
struct  cereal::LoadAndConstruct<::El::DistMatrix< DataT, CDist, RDist, Wrap, D > >
 Construct DistMatrix object from Cereal archives. More...
 

Namespaces

 El
 
 El::details
 
 lbann
 
 lbann::utils
 
 cereal
 

Macros

#define LBANN_UTILS_SERIALIZATION_SERIALIZE_MATRICES_HPP_
 

Functions

template<typename ArchiveT , typename T >
void El::save (ArchiveT &ar, ::El::AbstractMatrix< T > const &mat)
 Save a matrix to a text-based archive. More...
 
template<typename ArchiveT , typename T , ::El::Device D, lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::save (ArchiveT &ar, ::El::Matrix< T, D > const &mat)
 Save a matrix to a binary archive. More...
 
template<typename ArchiveT , typename T , lbann::utils::WhenNotTextArchive< ArchiveT > = 1>
void El::details::do_save (ArchiveT &ar, ::El::Matrix< T, ::El::Device::CPU > const &mat)
 Save a CPU matrix to a non-text-based archive. More...
 
template<typename ArchiveT , typename T , ::El::Device D>
void El::save (lbann::RootedOutputArchiveAdaptor< ArchiveT > &ar, ::El::Matrix< T, D > const &mat)
 
template<typename ArchiveT , typename T , lbann::utils::WhenNotTextArchive< ArchiveT > = 1>
void El::load (ArchiveT &archive, ::El::AbstractMatrix< T > &mat)
 
template<typename ArchiveT , typename T , ::El::Device D, lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::load (ArchiveT &archive, ::El::Matrix< T, D > &mat)
 "Load" a CPU Matrix from a text-based archive. More...
 
template<typename ArchiveT , typename T , lbann::utils::WhenNotTextArchive< ArchiveT > = 1>
void El::load (ArchiveT &archive, ::El::Matrix< T, ::El::Device::CPU > &mat)
 Load a CPU Matrix from a non-text archive. More...
 
template<typename ArchiveT , typename T , ::El::Device D>
void El::load (lbann::RootedInputArchiveAdaptor< ArchiveT > &ar, ::El::Matrix< T, D > &mat)
 
template<typename ArchiveT , typename T , lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::save (ArchiveT &ar, ::El::AbstractDistMatrix< T > const &mat)
 Save a distributed matrix to a text-based archive. More...
 
template<typename ArchiveT , typename T , lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::load (ArchiveT &ar, ::El::AbstractDistMatrix< T > &mat)
 Load a DistMatrix from a text-based archive. More...
 
template<typename ArchiveT , typename T , lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::save (lbann::RootedOutputArchiveAdaptor< ArchiveT > &ar, ::El::AbstractDistMatrix< T > const &mat)
 
template<typename ArchiveT , typename T , lbann::utils::WhenTextArchive< ArchiveT > = 1>
void El::load (lbann::RootedInputArchiveAdaptor< ArchiveT > &ar, ::El::AbstractDistMatrix< T > &mat)
 
template<typename ArchiveT , typename T , lbann::utils::WhenNotTextArchive< ArchiveT > = 1>
void El::save (lbann::RootedOutputArchiveAdaptor< ArchiveT > &ar, ::El::DistMatrix< T, ::El::CIRC, ::El::CIRC > const &mat)
 
template<typename ArchiveT , typename T , lbann::utils::WhenNotTextArchive< ArchiveT > = 1>
void El::load (lbann::RootedInputArchiveAdaptor< ArchiveT > &ar, ::El::DistMatrix< T, ::El::CIRC, ::El::CIRC > &mat)
 
Grid const & lbann::utils::get_current_grid () noexcept
 Get the current grid being used for deserialization. More...
 
lbann_comm & lbann::utils::get_current_comm () noexcept
 

Macro Definition Documentation

◆ LBANN_UTILS_SERIALIZATION_SERIALIZE_MATRICES_HPP_

#define LBANN_UTILS_SERIALIZATION_SERIALIZE_MATRICES_HPP_

Definition at line 28 of file serialize_matrices.hpp.