utils/serialize_matrices/common.hpp File Reference

#include <cereal/types/polymorphic.hpp>
#include <lbann/utils/serialization/serialize_matrices.hpp>

Include dependency graph for utils/serialize_matrices/common.hpp:

Go to the source code of this file.

Macros
#define	LBANN_COMMA   ,
#define	REGISTER_DISTMATRIX(TYPE, CDIST, RDIST, DEVICE)

Macro Definition Documentation

LBANN_COMMA

#define LBANN_COMMA   ,

Definition at line 31 of file utils/serialize_matrices/common.hpp.

REGISTER_DISTMATRIX

#define REGISTER_DISTMATRIX	(		TYPE,
			CDIST,
			RDIST,
			DEVICE
	)

Value:

CEREAL_REGISTER_TYPE_WITH_NAME(                                              \
    El::DistMatrix<TYPE LBANN_COMMA El::CDIST LBANN_COMMA El::RDIST            \
                     LBANN_COMMA El::ELEMENT LBANN_COMMA El::Device::DEVICE>,  \
    "DistMatrix(" #TYPE "," #CDIST "," #RDIST "," #DEVICE ")")                 \
  CEREAL_REGISTER_POLYMORPHIC_RELATION(                                        \
    El::AbstractDistMatrix<TYPE>,                                              \
    El::DistMatrix<TYPE LBANN_COMMA El::CDIST LBANN_COMMA El::RDIST            \
                     LBANN_COMMA El::ELEMENT LBANN_COMMA El::Device::DEVICE>)

Definition at line 34 of file utils/serialize_matrices/common.hpp.