lbann::gpu Namespace Reference

Functions
SyncInfo extractors.
template<typename TensorDataType >
El::SyncInfo< El::Device::GPU >	get_sync_info (El::AbstractMatrix< TensorDataType > const &m)
	Get a SyncInfo from an AbstractMatrix. More...
template<typename TensorDataType >
El::SyncInfo< El::Device::GPU >	get_sync_info (El::Matrix< TensorDataType, El::Device::GPU > const &m) noexcept
	Get a SyncInfo from an Matrix. More...
template<typename TensorDataType >
El::SyncInfo< El::Device::GPU >	get_sync_info (El::AbstractDistMatrix< TensorDataType > const &m)
	Get a SyncInfo from an AbstractDistMatrix. More...
template<typename TensorDataType , El::Dist RowDist, El::Dist ColDist>
El::SyncInfo< El::Device::GPU >	get_sync_info (El::DistMatrix< TensorDataType, RowDist, ColDist, El::ELEMENT, El::Device::GPU > const &m) noexcept
	Get a SyncInfo from a DistMatrix. More...

Function Documentation

get_sync_info() [1/4]

template<typename TensorDataType >

El::SyncInfo<El::Device::GPU> lbann::gpu::get_sync_info

(

El::AbstractMatrix< TensorDataType > const &

m

)

Get a SyncInfo from an AbstractMatrix.

Exceptions

std::bad_cast

If m is not a GPU matrix.

Definition at line 43 of file gpu/sync_info_helpers.hpp.

Here is the caller graph for this function:

get_sync_info() [2/4]

template<typename TensorDataType >

El::SyncInfo<El::Device::GPU> lbann::gpu::get_sync_info ( El::Matrix< TensorDataType, El::Device::GPU > const &  m )

noexcept

Get a SyncInfo from an Matrix.

Definition at line 52 of file gpu/sync_info_helpers.hpp.

get_sync_info() [3/4]

template<typename TensorDataType >

El::SyncInfo<El::Device::GPU> lbann::gpu::get_sync_info

(

El::AbstractDistMatrix< TensorDataType > const &

m

)

Get a SyncInfo from an AbstractDistMatrix.

Exceptions

std::bad_cast

If m is not a GPU matrix.

Definition at line 63 of file gpu/sync_info_helpers.hpp.

get_sync_info() [4/4]

template<typename TensorDataType , El::Dist RowDist, El::Dist ColDist>

El::SyncInfo<El::Device::GPU> lbann::gpu::get_sync_info ( El::DistMatrix< TensorDataType, RowDist, ColDist, El::ELEMENT, El::Device::GPU > const &  m )

noexcept

Get a SyncInfo from a DistMatrix.

This saves a dynamic_cast over the AbstractDistMatrix version.

Definition at line 76 of file gpu/sync_info_helpers.hpp.