iterative-solver 0.0
molpro::linalg::array::DistrArraySpan Class Reference

#include <DistrArraySpan.h>

Inheritance diagram for molpro::linalg::array::DistrArraySpan:
Collaboration diagram for molpro::linalg::array::DistrArraySpan:

Detailed Description

Classes

struct  LocalBufferSpan
 

Public Member Functions

 DistrArraySpan ()=delete
 
 DistrArraySpan (size_t dimension, Span< value_type > buffer, MPI_Comm commun=comm_global())
 
 DistrArraySpan (std::unique_ptr< Distribution > distribution, Span< value_type > buffer, MPI_Comm commun=comm_global())
 
 DistrArraySpan (const DistrArraySpan &source)
 
 DistrArraySpan (const DistrArray &source)
 
 DistrArraySpan (DistrArraySpan &&source) noexcept
 
DistrArraySpanoperator= (const DistrArraySpan &source)
 
DistrArraySpanoperator= (DistrArraySpan &&source) noexcept
 
void allocate_buffer (Span< value_type > buffer)
 
const Distributiondistribution () const override
 Access distribution of the array among processes. More...
 
std::unique_ptr< LocalBufferlocal_buffer () override
 Access the buffer local to this process. More...
 
std::unique_ptr< const LocalBufferlocal_buffer () const override
 
value_type at (index_type ind) const override
 
void set (index_type ind, value_type val) override
 Set one element to a scalar. Global operation. More...
 
void get (index_type lo, index_type hi, value_type *buf) const override
 Gets buffer[lo:hi) from global array (hi is past-the-end). Blocking. More...
 
std::vector< value_typeget (index_type lo, index_type hi) const override
 
void put (index_type lo, index_type hi, const value_type *data) override
 array[lo:hi) = data[:] (hi is past-the-end). Blocking More...
 
void acc (index_type lo, index_type hi, const value_type *data) override
 array[lo:hi) += scaling_constant * data[:] (hi is past-the-end). Blocking More...
 
std::vector< value_typegather (const std::vector< index_type > &indices) const override
 gets elements with discontinuous indices from array. Blocking More...
 
void scatter (const std::vector< index_type > &indices, const std::vector< value_type > &data) override
 array[indices[i]] = data[i] Puts vals of elements with discontinuous indices of array. Blocking. More...
 
void scatter_acc (std::vector< index_type > &indices, const std::vector< value_type > &data) override
 array[indices[i]] += vals[i] Accumulates vals of elements into discontinuous indices of array. Atomic, blocking, with on-sided communication More...
 
std::vector< value_typevec () const override
 Copies the whole buffer into a vector. Blocking. More...
 
- Public Member Functions inherited from molpro::linalg::array::DistrArray
virtual ~DistrArray ()=default
 
MPI_Comm communicator () const
 return a copy of the communicator More...
 
virtual void sync () const
 Synchronizes all process in this group and ensures any outstanding operations on the array have completed. More...
 
size_t size () const
 total number of elements, same as overall dimension of array More...
 
bool compatible (const DistrArray &other) const
 Checks that arrays are of the same dimensionality. More...
 
virtual void zero ()
 Set all local elements to zero. More...
 
virtual void error (const std::string &message) const
 stops application with an error More...
 
value_type operator[] (size_t index)
 
virtual void fill (value_type val)
 
virtual void copy (const DistrArray &y)
 
virtual void copy_patch (const DistrArray &y, index_type start, index_type end)
 Copies elements in a patch of y. If both arrays are empty than does nothing. If only one is empty, throws an error. More...
 
virtual void axpy (value_type a, const DistrArray &y)
 this[:] += a * y[:]. Throws an error if any array is empty. Add a multiple of another array to this one. Blocking, collective. More...
 
virtual void axpy (value_type a, const SparseArray &y)
 
virtual void scal (value_type a)
 Scale by a constant. Local. More...
 
virtual void add (const DistrArray &y)
 Add another array to this. Local. Throws error if any array is empty. More...
 
virtual void add (value_type a)
 Add a constant. Local. More...
 
virtual void sub (const DistrArray &y)
 Subtract another array from this. Local. Throws error if any array is empty. More...
 
virtual void sub (value_type a)
 Subtract a constant. Local. More...
 
virtual void recip ()
 Take element-wise reciprocal of this. Local. No checks are made for zero values. More...
 
virtual void times (const DistrArray &y)
 this[i] *= y[i]. Throws error if any array is empty. More...
 
virtual void times (const DistrArray &y, const DistrArray &z)
 this[i] = y[i]*z[i]. Throws error if any array is empty. More...
 
virtual value_type dot (const DistrArray &y) const
 Scalar product of two arrays. Collective. Throws error if any array is empty. Both arrays should be part of the same processor group (same communicator). The result is broadcast to each process. More...
 
virtual value_type dot (const SparseArray &y) const
 
void divide (const DistrArray &y, const DistrArray &z, value_type shift=0, bool append=false, bool negative=false)
 this[i] = y[i]/(z[i]+shift). Collective. Throws error if any array is empty. More...
 
std::list< std::pair< index_type, value_type > > min_n (int n) const
 returns n smallest elements in array x Collective operation, must be called by all processes in the group. More...
 
std::list< std::pair< index_type, value_type > > max_n (int n) const
 returns n largest elements in array x Collective operation, must be called by all processes in the group. More...
 
std::list< std::pair< index_type, value_type > > min_abs_n (int n) const
 returns n elements that are largest by absolute value in array x Collective operation, must be called by all processes in the group. More...
 
std::list< std::pair< index_type, value_type > > max_abs_n (int n) const
 returns n elements that are largest by absolute value in array x Collective operation, must be called by all processes in the group. More...
 
std::vector< index_typemin_loc_n (int n) const
 find the index of n smallest components in array x Collective operation, must be called by all processes in the group. More...
 
std::map< size_t, value_typeselect_max_dot (size_t n, const DistrArray &y) const
 
std::map< size_t, value_typeselect_max_dot (size_t n, const SparseArray &y) const
 
std::map< size_t, value_typeselect (size_t n, bool max=false, bool ignore_sign=false) const
 

Protected Attributes

std::unique_ptr< Distributionm_distribution
 describes distribution of array among processes More...
 
bool m_allocated = false
 whether the window has been created More...
 
Span< value_typem_span
 Span over provided buffer. More...
 
- Protected Attributes inherited from molpro::linalg::array::DistrArray
index_type m_dimension = 0
 number of elements in the array More...
 
MPI_Comm m_communicator
 

Friends

void swap (DistrArraySpan &a1, DistrArraySpan &a2) noexcept
 

Additional Inherited Members

- Public Types inherited from molpro::linalg::array::DistrArray
using distributed_array = void
 a compile time tag that this is a distributed array More...
 
using value_type = double
 
using index_type = size_t
 
using SparseArray = std::map< index_type, double >
 
using Distribution = util::Distribution< index_type >
 
- Protected Member Functions inherited from molpro::linalg::array::DistrArray
 DistrArray (size_t dimension, MPI_Comm commun)
 Initializes array without allocating any memory. More...
 
 DistrArray ()=default
 
virtual void _divide (const DistrArray &y, const DistrArray &z, value_type shift, bool append, bool negative)
 

Constructor & Destructor Documentation

◆ DistrArraySpan() [1/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( )
delete

◆ DistrArraySpan() [2/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( size_t  dimension,
Span< value_type buffer,
MPI_Comm  commun = comm_global() 
)

◆ DistrArraySpan() [3/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( std::unique_ptr< Distribution distribution,
Span< value_type buffer,
MPI_Comm  commun = comm_global() 
)

◆ DistrArraySpan() [4/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( const DistrArraySpan source)

◆ DistrArraySpan() [5/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( const DistrArray source)
explicit

◆ DistrArraySpan() [6/6]

molpro::linalg::array::DistrArraySpan::DistrArraySpan ( DistrArraySpan &&  source)
noexcept

Member Function Documentation

◆ acc()

void molpro::linalg::array::DistrArraySpan::acc ( index_type  lo,
index_type  hi,
const value_type data 
)
overridevirtual

array[lo:hi) += scaling_constant * data[:] (hi is past-the-end). Blocking

Implements molpro::linalg::array::DistrArray.

◆ allocate_buffer()

void molpro::linalg::array::DistrArraySpan::allocate_buffer ( Span< value_type buffer)

◆ at()

DistrArray::value_type molpro::linalg::array::DistrArraySpan::at ( index_type  ind) const
overridevirtual

get element at the offset. Blocking.

Implements molpro::linalg::array::DistrArray.

◆ distribution()

const DistrArray::Distribution & molpro::linalg::array::DistrArraySpan::distribution ( ) const
overridevirtual

Access distribution of the array among processes.

Implements molpro::linalg::array::DistrArray.

◆ gather()

std::vector< DistrArraySpan::value_type > molpro::linalg::array::DistrArraySpan::gather ( const std::vector< index_type > &  indices) const
overridevirtual

gets elements with discontinuous indices from array. Blocking

Returns
res[i] = array[indices[i]]

Implements molpro::linalg::array::DistrArray.

◆ get() [1/2]

std::vector< DistrArraySpan::value_type > molpro::linalg::array::DistrArraySpan::get ( DistrArray::index_type  lo,
DistrArray::index_type  hi 
) const
overridevirtual

◆ get() [2/2]

void molpro::linalg::array::DistrArraySpan::get ( index_type  lo,
index_type  hi,
value_type buf 
) const
overridevirtual

Gets buffer[lo:hi) from global array (hi is past-the-end). Blocking.

Implements molpro::linalg::array::DistrArray.

◆ local_buffer() [1/2]

std::unique_ptr< const DistrArray::LocalBuffer > molpro::linalg::array::DistrArraySpan::local_buffer ( ) const
overridevirtual

◆ local_buffer() [2/2]

std::unique_ptr< DistrArray::LocalBuffer > molpro::linalg::array::DistrArraySpan::local_buffer ( )
overridevirtual

Access the buffer local to this process.

Implements molpro::linalg::array::DistrArray.

◆ operator=() [1/2]

DistrArraySpan & molpro::linalg::array::DistrArraySpan::operator= ( const DistrArraySpan source)

◆ operator=() [2/2]

DistrArraySpan & molpro::linalg::array::DistrArraySpan::operator= ( DistrArraySpan &&  source)
noexcept

◆ put()

void molpro::linalg::array::DistrArraySpan::put ( index_type  lo,
index_type  hi,
const value_type data 
)
overridevirtual

array[lo:hi) = data[:] (hi is past-the-end). Blocking

Implements molpro::linalg::array::DistrArray.

◆ scatter()

void molpro::linalg::array::DistrArraySpan::scatter ( const std::vector< index_type > &  indices,
const std::vector< value_type > &  data 
)
overridevirtual

array[indices[i]] = data[i] Puts vals of elements with discontinuous indices of array. Blocking.

Implements molpro::linalg::array::DistrArray.

◆ scatter_acc()

void molpro::linalg::array::DistrArraySpan::scatter_acc ( std::vector< index_type > &  indices,
const std::vector< value_type > &  data 
)
overridevirtual

array[indices[i]] += vals[i] Accumulates vals of elements into discontinuous indices of array. Atomic, blocking, with on-sided communication

Implements molpro::linalg::array::DistrArray.

◆ set()

void molpro::linalg::array::DistrArraySpan::set ( index_type  ind,
value_type  val 
)
overridevirtual

Set one element to a scalar. Global operation.

Implements molpro::linalg::array::DistrArray.

◆ vec()

std::vector< DistrArraySpan::value_type > molpro::linalg::array::DistrArraySpan::vec ( ) const
overridevirtual

Copies the whole buffer into a vector. Blocking.

Note
This is only meant for debugging small arrays!

Implements molpro::linalg::array::DistrArray.

Friends And Related Function Documentation

◆ swap

void swap ( DistrArraySpan a1,
DistrArraySpan a2 
)
friend

Member Data Documentation

◆ m_allocated

bool molpro::linalg::array::DistrArraySpan::m_allocated = false
protected

whether the window has been created

◆ m_distribution

std::unique_ptr<Distribution> molpro::linalg::array::DistrArraySpan::m_distribution
protected

describes distribution of array among processes

◆ m_span

Span<value_type> molpro::linalg::array::DistrArraySpan::m_span
protected

Span over provided buffer.