#include <lbann/macros/common_cereal_registration.hpp>
#include <cereal/types/polymorphic.hpp>
#include "lbann/macros/instantiate_device.hpp"

Include dependency graph for register_operator_with_cereal.hpp:

Macros
#define	LBANN_COMMA ,

#define	LBANN_REGISTER_OPERATOR_WITH_CEREAL_BASE(NAME, TYPE, DEVICE)

#define	LBANN_REGISTER_OPERATOR_WITH_CEREAL(NAME, TYPE, DEVICE) LBANN_REGISTER_OPERATOR_WITH_CEREAL_BASE(NAME, TYPE, DEVICE)

#define	PROTO_DEVICE(T, D) LBANN_REGISTER_OPERATOR_WITH_CEREAL(LBANN_OPERATOR_NAME, T, D)

Functions
	LBANN_REGISTER_DYNAMIC_INIT (LBANN_OPERATOR_NAME)

Detailed Description

Define LBANN_OPERATOR_NAME to be the full operator class name before including this file. Don't include this file inside the lbann namespace.

Definition in file register_operator_with_cereal.hpp.

Macro Definition Documentation

◆ LBANN_COMMA

#define LBANN_COMMA ,

Definition at line 44 of file register_operator_with_cereal.hpp.

◆ LBANN_REGISTER_OPERATOR_WITH_CEREAL

#define LBANN_REGISTER_OPERATOR_WITH_CEREAL	(	NAME,
		TYPE,
		DEVICE
	)	LBANN_REGISTER_OPERATOR_WITH_CEREAL_BASE(NAME, TYPE, DEVICE)

Definition at line 51 of file register_operator_with_cereal.hpp.

◆ LBANN_REGISTER_OPERATOR_WITH_CEREAL_BASE

#define LBANN_REGISTER_OPERATOR_WITH_CEREAL_BASE	(	NAME,
		TYPE,
		DEVICE
	)

Value:

LBANN_ADD_ALL_SERIALIZE_ETI(::lbann::NAME<TYPE LBANN_COMMA DEVICE>);         \
  CEREAL_REGISTER_TYPE_WITH_NAME(::lbann::NAME<TYPE LBANN_COMMA DEVICE>,       \
                                 #NAME "(" #TYPE "," #DEVICE ")")

Definition at line 46 of file register_operator_with_cereal.hpp.

◆ PROTO_DEVICE

#define PROTO_DEVICE	(	T,
		D
	)	LBANN_REGISTER_OPERATOR_WITH_CEREAL(LBANN_OPERATOR_NAME, T, D)

Definition at line 54 of file register_operator_with_cereal.hpp.

Function Documentation

◆ LBANN_REGISTER_DYNAMIC_INIT()

LBANN_REGISTER_DYNAMIC_INIT ( LBANN_OPERATOR_NAME )

Macros

Functions