gpu/sync_info_helpers.hpp File Reference

#include <El.hpp>

Include dependency graph for gpu/sync_info_helpers.hpp:

Go to the source code of this file.

Namespaces
	lbann
	lbann::gpu

Functions
SyncInfo extractors.
template<typename TensorDataType >
El::SyncInfo< El::Device::GPU >	lbann::gpu::get_sync_info (El::AbstractMatrix< TensorDataType > const &m)
	Get a SyncInfo from an AbstractMatrix. More...
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. More...
template<typename TensorDataType >
El::SyncInfo< El::Device::GPU >	lbann::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 >	lbann::gpu::get_sync_info (El::DistMatrix< TensorDataType, RowDist, ColDist, El::ELEMENT, El::Device::GPU > const &m) noexcept
	Get a SyncInfo from a DistMatrix. More...