LAMA
lama::BlockDistribution Class Reference

Block distribution in contiguous chunks of same size for each partition. More...

#include <BlockDistribution.hpp>

Inheritance diagram for lama::BlockDistribution:

Public Member Functions

 BlockDistribution (const IndexType globalSize, const CommunicatorPtr communicator)
 Construct a block distribution for a number of elements on to the partitions of the passed communicator.
virtual ~BlockDistribution ()
virtual bool isLocal (const IndexType index) const
 Query if the given global index is local for the calling rank (e.g.
virtual PartitionId getOwner (const IndexType globalIndex) const
 Computes the owner of the passed globalIndex.
virtual IndexType getLocalSize () const
 This method returns the number of local elements on the calling processor.
virtual IndexType local2global (const IndexType localIndex) const
 Abstract method that translates a local index back to a global index.
virtual IndexType global2local (const IndexType globalIndex) const
 Abstract method that translates a global index into a local index.
virtual bool isEqual (const Distribution &other) const
virtual void writeAt (std::ostream &stream) const
 This method writes info about the distribution into an output stream.
virtual void computeOwners (const std::vector< IndexType > &requiredIndexes, std::vector< PartitionId > &owners) const
 Method to compute directly the owners of global indexes without any communication.
void printDistributionVector (std::string problem) const
 Master process prints out the distribution vector to file named "name.part".
const CommunicatorgetCommunicator () const
 Getter routine for the communicator of the distribution.
CommunicatorPtr getCommunicatorPtr () const
 Getter routine for the communicator as shared pointer.
PartitionId getNumPartitions () const
 Query for the number of partitions onto which the distribution is done.
bool isReplicated () const
 Query whether the distribution is a replication.
IndexType getGlobalSize () const
 Getter for the global number of elements that are distributed.
bool operator== (const Distribution &other) const
 Check for equality of two distributions.
bool operator!= (const Distribution &other) const
 Check for inequality.
template<typename T1 , typename T2 >
void replicate (T1 *allValues, const T2 *localValues) const
template<typename T >
void replicate (T *allValues, const IndexType *allOffsets, const T *localValues) const
template<typename T1 , typename T2 >
void replicateN (T1 *allValues, const T2 *localValues, const IndexType n) const

Static Public Member Functions

static void getRange (IndexType &lb, IndexType &ub, const IndexType n, const PartitionId rank, const PartitionId size)
 Static method that allows to compute the range of a block distribution for arbitrary rank and communicators.
static DistributionPtr create (const IndexType globalSize, const CommunicatorPtr communicator)
 Static method to construct a new block distribution.

Protected Member Functions

 LAMA_LOG_DECL_STATIC_LOGGER (logger)

Protected Attributes

IndexType mGlobalSize
CommunicatorPtr mCommunicator

Private Member Functions

 BlockDistribution ()

Private Attributes

IndexType mBlockSize
 block size of each partition
IndexType lb
IndexType ub
 local range of full size in global values

Detailed Description

Block distribution in contiguous chunks of same size for each partition.

BlockDistribution is noncopyable as Distribution is noncopyable


Constructor & Destructor Documentation

lama::BlockDistribution::BlockDistribution ( const IndexType  globalSize,
const CommunicatorPtr  communicator 
)

Construct a block distribution for a number of elements on to the partitions of the passed communicator.

Parameters:
[in]globalSizenumber of elements to distribute
[in]communicatorused for the partitions onto which elements are distributed.

References lama::Distribution::getGlobalSize(), getRange(), lb, mBlockSize, lama::Distribution::mCommunicator, and ub.


Member Function Documentation

void lama::BlockDistribution::computeOwners ( const std::vector< IndexType > &  requiredIndexes,
std::vector< PartitionId > &  owners 
) const [virtual]

Method to compute directly the owners of global indexes without any communication.

Reimplemented from lama::Distribution.

References getOwner().

DistributionPtr lama::BlockDistribution::create ( const IndexType  globalSize,
const CommunicatorPtr  communicator 
) [static]

Static method to construct a new block distribution.

References BlockDistribution().

Getter routine for the communicator as shared pointer.

References lama::Distribution::mCommunicator.

Referenced by lama::DenseMatrix< T >::invert().

This method returns the number of local elements on the calling processor.

Implements lama::Distribution.

References lb, and ub.

Referenced by printDistributionVector().

PartitionId lama::BlockDistribution::getOwner ( const IndexType  globalIndex) const [virtual]

Computes the owner of the passed globalIndex.

Parameters:
[in]globalIndexthe global index to compute the owner for.
Returns:
the owner of the passed globalIndex.

References mBlockSize.

Referenced by computeOwners().

void lama::BlockDistribution::getRange ( IndexType lb,
IndexType ub,
const IndexType  n,
const PartitionId  rank,
const PartitionId  size 
) [static]

Static method that allows to compute the range of a block distribution for arbitrary rank and communicators.

References LAMA_ASSERT_DEBUG, and lama::min().

Referenced by BlockDistribution(), and lama::MatrixCreator< T >::buildPoisson().

IndexType lama::BlockDistribution::global2local ( const IndexType  globalIndex) const [virtual]

Abstract method that translates a global index into a local index.

Parameters:
[in]globalIndexwith 0 <= globalIndex < getGlobalSize
Returns:
localIndex with 0 <= localIndex < getLocalSize() if local, nIndex otherwise

This method must be implemented by all base classes. It should throw an exception if the argument is not in the valid range.

Implements lama::Distribution.

References lb, lama::nIndex, and ub.

bool lama::BlockDistribution::isLocal ( const IndexType  index) const [virtual]

Query if the given global index is local for the calling rank (e.g.

process for an MPI Communicator

Parameters:
[in]indexthe global index to query for locality to the calling rank
Returns:
if the passed global index is local to the calling rank

Implements lama::Distribution.

References lb, and ub.

Reimplemented from lama::Distribution.

IndexType lama::BlockDistribution::local2global ( const IndexType  localIndex) const [virtual]

Abstract method that translates a local index back to a global index.

Parameters:
[in]localIndexis the local index, 0 <= localIndex < getLocalSize()
Returns:
globalIndex with 0 <= globalIndex < getGlobalSize

This method must be implemented by all base classes. It should throw an exception if the argument is not in the valid range.

Implements lama::Distribution.

References lb.

bool lama::Distribution::operator!= ( const Distribution other) const [inherited]

Check for inequality.

Even if two distributions are not equal it might be the case that the mapping of elements to partititons is the same.

bool lama::Distribution::operator== ( const Distribution other) const [inherited]

Check for equality of two distributions.

As verification of same distribution can be rather expensive, the operator might return false.

Attention: The operator must be conservative and only return true if the distributions are really equal.

References lama::Distribution::getGlobalSize(), lama::Distribution::isEqual(), and lama::Distribution::isReplicated().

void lama::BlockDistribution::printDistributionVector ( std::string  name) const [virtual]

Master process prints out the distribution vector to file named "name.part".

Every row contains a single number: the index of the process, where the row is local.

Implements lama::Distribution.

References getLocalSize(), and lama::Distribution::mCommunicator.

void lama::BlockDistribution::writeAt ( std::ostream &  stream) const [virtual]

This method writes info about the distribution into an output stream.

The method should be overwritten by base classes to give more specific information about the object.

Reimplemented from lama::Distribution.

References mBlockSize, and lama::Distribution::mGlobalSize.


Field Documentation

block size of each partition

Referenced by BlockDistribution(), getOwner(), and writeAt().

local range of full size in global values

Referenced by BlockDistribution(), getLocalSize(), global2local(), and isLocal().


The documentation for this class was generated from the following files: