Zoltan2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
Public Member Functions | Protected Member Functions | Protected Attributes | List of all members
Zoltan2::TpetraRowMatrixAdapter< User, UserCoord > Class Template Reference

Provides access for Zoltan2 to Tpetra::RowMatrix data. More...

#include <Zoltan2_TpetraRowMatrixAdapter.hpp>

Inheritance diagram for Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >:
Inheritance graph
[legend]
Collaboration diagram for Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >:
Collaboration graph
[legend]

Public Member Functions

 TpetraRowMatrixAdapter (const RCP< const User > &inmatrix, int nWeightsPerRow=0)
 Constructor. More...
 
void setWeights (const scalar_t *weightVal, int stride, int idx=0)
 Specify a weight for each entity of the primaryEntityType. More...
 
void setWeightsDevice (typename Base::ConstWeightsDeviceView1D val, int idx)
 Provide a device view of weights for the primary entity type. More...
 
void setWeightsHost (typename Base::ConstWeightsHostView1D val, int idx)
 Provide a host view of weights for the primary entity type. More...
 
void setRowWeights (const scalar_t *weightVal, int stride, int idx=0)
 Specify a weight for each row. More...
 
void setRowWeightsDevice (typename Base::ConstWeightsDeviceView1D val, int idx)
 Provide a device view to row weights. More...
 
void setRowWeightsHost (typename Base::ConstWeightsHostView1D val, int idx)
 Provide a host view to row weights. More...
 
void setWeightIsDegree (int idx)
 Specify an index for which the weight should be the degree of the entity. More...
 
void setRowWeightIsNumberOfNonZeros (int idx)
 Specify an index for which the row weight should be the global number of nonzeros in the row. More...
 
size_t getLocalNumRows () const
 Returns the number of rows on this process. More...
 
size_t getLocalNumColumns () const
 Returns the number of columns on this process. More...
 
size_t getLocalNumEntries () const
 Returns the number of nonzeros on this process. More...
 
bool CRSViewAvailable () const
 Indicates whether the MatrixAdapter implements a view of the matrix in compressed sparse row (CRS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
void getRowIDsView (const gno_t *&rowIds) const override
 
void getRowIDsHostView (typename Base::ConstIdsHostView &rowIds) const override
 
void getRowIDsDeviceView (typename Base::ConstIdsDeviceView &rowIds) const override
 
void getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds) const
 
void getCRSHostView (typename Base::ConstOffsetsHostView &offsets, typename Base::ConstIdsHostView &colIds) const override
 
void getCRSDeviceView (typename Base::ConstOffsetsDeviceView &offsets, typename Base::ConstIdsDeviceView &colIds) const override
 
void getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds, ArrayRCP< const scalar_t > &values) const
 
void getCRSHostView (typename Base::ConstOffsetsHostView &offsets, typename Base::ConstIdsHostView &colIds, typename Base::ConstScalarsHostView &values) const override
 
void getCRSDeviceView (typename Base::ConstOffsetsDeviceView &offsets, typename Base::ConstIdsDeviceView &colIds, typename Base::ConstScalarsDeviceView &values) const override
 
int getNumWeightsPerRow () const
 Returns the number of weights per row (0 or greater). Row weights may be used when partitioning matrix rows. More...
 
void getRowWeightsView (const scalar_t *&weights, int &stride, int idx=0) const
 Provide a pointer to the row weights, if any. More...
 
void getRowWeightsDeviceView (typename Base::WeightsDeviceView1D &weights, int idx=0) const
 
void getRowWeightsDeviceView (typename Base::WeightsDeviceView &weights) const override
 
void getRowWeightsHostView (typename Base::WeightsHostView1D &weights, int idx=0) const
 
void getRowWeightsHostView (typename Base::WeightsHostView &weights) const override
 
bool useNumNonzerosAsRowWeight (int idx) const
 Indicate whether row weight with index idx should be the global number of nonzeros in the row. More...
 
template<typename Adapter >
void applyPartitioningSolution (const User &in, User *&out, const PartitioningSolution< Adapter > &solution) const
 
template<typename Adapter >
void applyPartitioningSolution (const User &in, RCP< User > &out, const PartitioningSolution< Adapter > &solution) const
 
- Public Member Functions inherited from Zoltan2::MatrixAdapter< User, UserCoord >
enum BaseAdapterType adapterType () const override
 Returns the type of adapter. More...
 
 MatrixAdapter ()
 
virtual void getRowIDsView (const gno_t *&rowIds) const
 Sets pointer to this process' rows' global IDs. More...
 
virtual void getRowIDsHostView (typename BaseAdapter< User >::ConstIdsHostView &rowIds) const
 
virtual void getRowIDsDeviceView (typename BaseAdapter< User >::ConstIdsDeviceView &rowIds) const
 
virtual void getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds) const
 Sets pointers to this process' matrix entries using compressed sparse row (CRS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
virtual void getCRSHostView (typename BaseAdapter< User >::ConstOffsetsHostView &offsets, typename BaseAdapter< User >::ConstIdsHostView &colIds) const
 
virtual void getCRSDeviceView (typename BaseAdapter< User >::ConstOffsetsDeviceView &offsets, typename BaseAdapter< User >::ConstIdsDeviceView &colIds) const
 
virtual void getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds, ArrayRCP< const scalar_t > &values) const
 Sets pointers to this process' matrix entries and their values using compressed sparse row (CRS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
virtual void getCRSHostView (typename BaseAdapter< User >::ConstOffsetsHostView &offsets, typename BaseAdapter< User >::ConstIdsHostView &colIds, typename BaseAdapter< User >::ConstScalarsHostView &values) const
 
virtual void getCRSDeviceView (typename BaseAdapter< User >::ConstOffsetsDeviceView &offsets, typename BaseAdapter< User >::ConstIdsDeviceView &colIds, typename BaseAdapter< User >::ConstScalarsDeviceView &values) const
 
virtual void getRowWeightsHostView (typename BaseAdapter< User >::WeightsHostView1D &weights, int=0) const
 
virtual void getRowWeightsHostView (typename BaseAdapter< User >::WeightsHostView &weights) const
 
virtual void getRowWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView1D &weights, int=0) const
 
virtual void getRowWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView &weights) const
 
virtual bool CCSViewAvailable () const
 Indicates whether the MatrixAdapter implements a view of the matrix in compressed sparse column (CCS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
virtual void getColumnIDsView (const gno_t *&colIds) const
 Sets pointer to this process' columns' global IDs. More...
 
virtual void getColumnIDsHostView (typename BaseAdapter< User >::ConstIdsHostView &colIds) const
 
virtual void getColumnIDsDeviceView (typename BaseAdapter< User >::ConstIdsDeviceView &colIds) const
 
virtual void getCCSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &rowIds) const
 Sets pointers to this process' matrix entries using compressed sparse column (CCS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
virtual void getCCSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &rowIds, ArrayRCP< const scalar_t > &values) const
 Sets pointers to this process' matrix entries and their values using compressed sparse column (CCS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required. More...
 
virtual int getNumWeightsPerColumn () const
 Returns the number of weights per column (0 or greater). Column weights may be used when partitioning matrix columns. More...
 
virtual void getColumnWeightsView (const scalar_t *&weights, int &stride, int idx=0) const
 Provide a pointer to the column weights, if any. More...
 
virtual void getColumnWeightsHostView (typename BaseAdapter< User >::WeightsHostView1D &weights, int=0) const
 
virtual void getColumnWeightsHostView (typename BaseAdapter< User >::WeightsHostView &weights) const
 
virtual void getColumnWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView1D &weights, int=0) const
 
virtual void getColumnWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView &weights) const
 
virtual bool useNumNonzerosAsColumnWeight (int idx) const
 Indicate whether column weight with index idx should be the global number of nonzeros in the column. More...
 
void setCoordinateInput (VectorAdapter< UserCoord > *coordData) override
 Allow user to provide additional data that contains coordinate info associated with the MatrixAdapter's primaryEntityType. Associated data must have the same parallel distribution and ordering of entries as the primaryEntityType. More...
 
bool coordinatesAvailable () const
 Indicate whether coordinate information has been set for this MatrixAdapter. More...
 
VectorAdapter< UserCoord > * getCoordinateInput () const override
 Obtain the coordinate data registered by the user. More...
 
enum MatrixEntityType getPrimaryEntityType () const
 Returns the entity to be partitioned, ordered, colored, etc. Valid values are MATRIX_ROW, MATRIX_COLUMN, MATRIX_NONZERO. More...
 
void setPrimaryEntityType (std::string typestr)
 Sets the primary entity type. Called by algorithm based on parameter value in parameter list from application. Also sets to adjacencyEntityType to something reasonable: opposite of primaryEntityType. More...
 
size_t getLocalNumIDs () const override
 Returns the number of objects on this process. More...
 
void getIDsView (const gno_t *&Ids) const override
 Provide a pointer to this process' identifiers. More...
 
void getIDsHostView (typename BaseAdapter< User >::ConstIdsHostView &ids) const override
 
void getIDsDeviceView (typename BaseAdapter< User >::ConstIdsDeviceView &ids) const override
 
int getNumWeightsPerID () const override
 Returns the number of weights per object. Number of weights per object should be zero or greater. If zero, then it is assumed that all objects are equally weighted. Default is zero weights per ID. More...
 
void getWeightsView (const scalar_t *&wgt, int &stride, int idx=0) const override
 Provide pointer to a weight array with stride. More...
 
void getWeightsHostView (typename BaseAdapter< User >::WeightsHostView &hostWgts) const override
 
void getWeightsHostView (typename BaseAdapter< User >::WeightsHostView1D &hostWgts, int idx=0) const override
 
void getWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView &deviceWgts) const override
 
void getWeightsDeviceView (typename BaseAdapter< User >::WeightsDeviceView1D &deviceWgts, int idx=0) const override
 
bool useDegreeAsWeight (int idx) const
 
- Public Member Functions inherited from Zoltan2::BaseAdapter< User >
virtual void getIDsKokkosView (ConstIdsDeviceView &ids) const
 Provide a Kokkos view to this process' identifiers. More...
 
virtual void getIDsHostView (ConstIdsHostView &hostIds) const
 Provide a Kokkos view (Host side) to this process' identifiers. More...
 
virtual void getIDsDeviceView (ConstIdsDeviceView &deviceIds) const
 Provide a Kokkos view (Device side) to this process' identifiers. More...
 
virtual void getWeightsKokkosView (Kokkos::View< scalar_t **, device_t > &wgt) const
 Provide kokkos view of weights. More...
 
virtual void getWeightsHostView (WeightsHostView1D &hostWgts, int idx=0) const
 Provide a Kokkos view (Host side) of the weights. More...
 
virtual void getWeightsHostView (WeightsHostView &hostWgts) const
 Provide a Kokkos view (Host side) of the weights. More...
 
virtual void getWeightsDeviceView (WeightsDeviceView1D &deviceWgts, int idx=0) const
 Provide a Kokkos view (Device side) of the weights. More...
 
virtual void getWeightsDeviceView (WeightsDeviceView &deviceWgts) const
 Provide a Kokkos view (Device side) of the weights. More...
 
virtual void getPartsView (const part_t *&inputPart) const
 Provide pointer to an array containing the input part assignment for each ID. The input part information may be used for re-partitioning to reduce data movement, or for mapping parts to processes. Adapters may return NULL for this pointer (the default behavior); if NULL is returned, algorithms will assume the rank. More...
 
virtual void getPartsHostView (Kokkos::View< part_t *, host_t > &inputPart) const
 
virtual void getPartsDeviceView (Kokkos::View< part_t *, device_t > &inputPart) const
 
template<typename Adapter >
void applyPartitioningSolution (const User &in, User *&out, const PartitioningSolution< Adapter > &solution) const
 Apply a PartitioningSolution to an input. More...
 
- Public Member Functions inherited from Zoltan2::BaseAdapterRoot
virtual ~BaseAdapterRoot ()=default
 

Protected Member Functions

 TpetraRowMatrixAdapter (int nWeightsPerRow, const RCP< const User > &inmatrix)
 
virtual RCP< User > doMigration (const User &from, size_t numLocalRows, const gno_t *myNewRows) const
 
- Protected Member Functions inherited from Zoltan2::BaseAdapter< User >
void generateWeightFileOnly (const char *fileprefix, const Teuchos::Comm< int > &comm) const
 

Protected Attributes

RCP< const User > matrix_
 
ArrayRCP< offset_toffset_
 
ArrayRCP< gno_tcolumnIds_
 
ArrayRCP< scalar_tvalues_
 
Base::ConstOffsetsHostView offsHost_
 
Base::ConstIdsHostView colIdsHost_
 
Base::ScalarsHostView valuesHost_
 
Base::ConstOffsetsDeviceView offsDevice_
 
Base::ConstIdsDeviceView colIdsDevice_
 
Base::ScalarsDeviceView valuesDevice_
 
int nWeightsPerRow_
 
ArrayRCP< StridedData< lno_t,
scalar_t > > 
rowWeights_
 
Base::WeightsDeviceView rowWeightsDevice_
 
Kokkos::View< bool *, host_tnumNzWeight_
 
bool mayHaveDiagonalEntries
 

Additional Inherited Members

- Public Types inherited from Zoltan2::BaseAdapter< User >
using lno_t = typename InputTraits< User >::lno_t
 
using gno_t = typename InputTraits< User >::gno_t
 
using scalar_t = typename InputTraits< User >::scalar_t
 
using node_t = typename InputTraits< User >::node_t
 
using part_t = typename InputTraits< User >::part_t
 
using offset_t = typename InputTraits< User >::offset_t
 
using host_t = typename Kokkos::HostSpace::memory_space
 
using device_t = typename node_t::device_type
 
using ConstIdsDeviceView = Kokkos::View< const gno_t *, device_t >
 
using ConstIdsHostView = typename ConstIdsDeviceView::HostMirror
 
using IdsDeviceView = Kokkos::View< gno_t *, device_t >
 
using IdsHostView = typename IdsDeviceView::HostMirror
 
using ConstOffsetsDeviceView = Kokkos::View< const offset_t *, device_t >
 
using ConstOffsetsHostView = typename ConstOffsetsDeviceView::HostMirror
 
using OffsetsDeviceView = Kokkos::View< offset_t *, device_t >
 
using OffsetsHostView = typename OffsetsDeviceView::HostMirror
 
using ConstScalarsDeviceView = Kokkos::View< const scalar_t *, device_t >
 
using ConstScalarsHostView = typename ConstScalarsDeviceView::HostMirror
 
using ScalarsDeviceView = Kokkos::View< scalar_t *, device_t >
 
using ScalarsHostView = typename ScalarsDeviceView::HostMirror
 
using ConstWeightsDeviceView1D = Kokkos::View< const scalar_t *, device_t >
 
using ConstWeightsHostView1D = typename ConstWeightsDeviceView1D::HostMirror
 
using WeightsDeviceView1D = Kokkos::View< scalar_t *, device_t >
 
using WeightsHostView1D = typename WeightsDeviceView1D::HostMirror
 
using ConstWeightsDeviceView = Kokkos::View< const scalar_t **, device_t >
 
using ConstWeightsHostView = typename ConstWeightsDeviceView::HostMirror
 
using WeightsDeviceView = Kokkos::View< scalar_t **, device_t >
 
using WeightsHostView = typename WeightsDeviceView::HostMirror
 

Detailed Description

template<typename User, typename UserCoord = User>
class Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >

Provides access for Zoltan2 to Tpetra::RowMatrix data.

The scalar_t type, representing user data such as matrix values, is used by Zoltan2 for weights, coordinates, part sizes and quality metrics. Some User types (like Tpetra::RowMatrix) have an inherent scalar type, and some (like Tpetra::RowGraph) do not. For such objects, the scalar type is set by Zoltan2 to float. If you wish to change it to double, set the second template parameter to double.

Definition at line 78 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

Constructor & Destructor Documentation

template<typename User , typename UserCoord >
Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::TpetraRowMatrixAdapter ( const RCP< const User > &  inmatrix,
int  nWeightsPerRow = 0 
)

Constructor.

Parameters
inmatrixThe user's Tpetra RowMatrix object
nWeightsPerRowIf row weights will be provided in setRowWeights(), then set nWeightsPerRow to the number of weights per row.

Definition at line 313 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::TpetraRowMatrixAdapter ( int  nWeightsPerRow,
const RCP< const User > &  inmatrix 
)
inlineprotected

Definition at line 279 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

Member Function Documentation

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeights ( const scalar_t weightVal,
int  stride,
int  idx = 0 
)

Specify a weight for each entity of the primaryEntityType.

Parameters
weightValA pointer to the weights for this index. A stride to be used in reading the values. The index idx weight for entity should be found at weightVal[k*stride].
idxA value between zero and one less that the nWeightsPerRow argument to the constructor.

The order of weights should correspond to the order of the primary entity type; see, e.g., setRowWeights below.

Definition at line 373 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeightsDevice ( typename Base::ConstWeightsDeviceView1D  val,
int  idx 
)

Provide a device view of weights for the primary entity type.

Parameters
valA view to the weights for index idx.
idxA number from 0 to one less than weight idx specified in the constructor.

The order of the weights should match the order that entities appear in the input data structure.

Definition at line 389 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeightsHost ( typename Base::ConstWeightsHostView1D  val,
int  idx 
)

Provide a host view of weights for the primary entity type.

Parameters
valA view to the weights for index idx.
idxA number from 0 to one less than weight idx specified in the constructor.

The order of the weights should match the order that entities appear in the input data structure.

Definition at line 405 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setRowWeights ( const scalar_t weightVal,
int  stride,
int  idx = 0 
)

Specify a weight for each row.

Parameters
weightValA pointer to the weights for this index.
strideA stride to be used in reading the values. The index idx weight for row should be found at weightVal[k*stride].
idxA value between zero and one less that the nWeightsPerRow argument to the constructor.

The order of weights should correspond to the order of rows returned by

TheMatrix->getRowMap()->getLocalElementList();

Definition at line 421 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setRowWeightsDevice ( typename Base::ConstWeightsDeviceView1D  val,
int  idx 
)

Provide a device view to row weights.

Parameters
valA pointer to the weights for index idx.
idxA number from 0 to one less than number of row weights specified in the constructor.

The order of the row weights should match the order that rows appear in the input data structure.

TheMatrix->getRowMap()->getLocalElementList()

Definition at line 434 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setRowWeightsHost ( typename Base::ConstWeightsHostView1D  val,
int  idx 
)

Provide a host view to row weights.

Parameters
valA pointer to the weights for index idx.
idxA number from 0 to one less than number of row weights specified in the constructor.

The order of the row weights should match the order that rows appear in the input data structure.

TheMatrix->getRowMap()->getLocalElementList()

Definition at line 450 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeightIsDegree ( int  idx)

Specify an index for which the weight should be the degree of the entity.

Parameters
idxZoltan2 will use the entity's degree as the entity weight for index idx.

Definition at line 463 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setRowWeightIsNumberOfNonZeros ( int  idx)

Specify an index for which the row weight should be the global number of nonzeros in the row.

Parameters
idxZoltan2 will use the global number of nonzeros in a row as the row weight for index idx.

Definition at line 478 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
size_t Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getLocalNumRows ( ) const
virtual

Returns the number of rows on this process.

Implements Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 492 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
size_t Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getLocalNumColumns ( ) const
virtual

Returns the number of columns on this process.

Implements Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 498 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
size_t Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getLocalNumEntries ( ) const
virtual

Returns the number of nonzeros on this process.

Implements Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 504 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
bool Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::CRSViewAvailable ( ) const
virtual

Indicates whether the MatrixAdapter implements a view of the matrix in compressed sparse row (CRS) format. All matrix adapters must implement either getCRSView or getCCSView, but implementation of both is not required.

Reimplemented from Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 510 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowIDsView ( const gno_t *&  rowIds) const
override

Definition at line 514 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowIDsHostView ( typename Base::ConstIdsHostView &  rowIds) const
override

Definition at line 521 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowIDsDeviceView ( typename Base::ConstIdsDeviceView &  rowIds) const
override

Definition at line 533 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSView ( ArrayRCP< const offset_t > &  offsets,
ArrayRCP< const gno_t > &  colIds 
) const

Definition at line 546 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSHostView ( typename Base::ConstOffsetsHostView &  offsets,
typename Base::ConstIdsHostView &  colIds 
) const
override

Definition at line 554 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSDeviceView ( typename Base::ConstOffsetsDeviceView &  offsets,
typename Base::ConstIdsDeviceView &  colIds 
) const
override

Definition at line 568 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSView ( ArrayRCP< const offset_t > &  offsets,
ArrayRCP< const gno_t > &  colIds,
ArrayRCP< const scalar_t > &  values 
) const

Definition at line 577 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSHostView ( typename Base::ConstOffsetsHostView &  offsets,
typename Base::ConstIdsHostView &  colIds,
typename Base::ConstScalarsHostView &  values 
) const
override

Definition at line 587 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getCRSDeviceView ( typename Base::ConstOffsetsDeviceView &  offsets,
typename Base::ConstIdsDeviceView &  colIds,
typename Base::ConstScalarsDeviceView &  values 
) const
override

Definition at line 606 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
int Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getNumWeightsPerRow ( ) const
virtual

Returns the number of weights per row (0 or greater). Row weights may be used when partitioning matrix rows.

Reimplemented from Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 617 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowWeightsView ( const scalar_t *&  weights,
int &  stride,
int  idx = 0 
) const
virtual

Provide a pointer to the row weights, if any.

Parameters
weightsis the list of weights with a given index for the rows returned in getRowIDsView().
strideThe k'th weight is located at weights[stride*k]
idxranges from zero to one less than getNumWeightsPerRow().

Reimplemented from Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 621 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowWeightsDeviceView ( typename Base::WeightsDeviceView1D &  weights,
int  idx = 0 
) const

Definition at line 636 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowWeightsDeviceView ( typename Base::WeightsDeviceView &  weights) const
override

Definition at line 654 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowWeightsHostView ( typename Base::WeightsHostView1D &  weights,
int  idx = 0 
) const

Definition at line 662 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::getRowWeightsHostView ( typename Base::WeightsHostView &  weights) const
override

Definition at line 677 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
bool Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::useNumNonzerosAsRowWeight ( int  idx) const
virtual

Indicate whether row weight with index idx should be the global number of nonzeros in the row.

Reimplemented from Zoltan2::MatrixAdapter< User, UserCoord >.

Definition at line 686 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
template<typename Adapter >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::applyPartitioningSolution ( const User &  in,
User *&  out,
const PartitioningSolution< Adapter > &  solution 
) const

Definition at line 691 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
template<typename Adapter >
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::applyPartitioningSolution ( const User &  in,
RCP< User > &  out,
const PartitioningSolution< Adapter > &  solution 
) const

Definition at line 713 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord >
RCP< User > Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::doMigration ( const User &  from,
size_t  numLocalRows,
const gno_t myNewRows 
) const
protectedvirtual

Definition at line 732 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

Member Data Documentation

template<typename User , typename UserCoord = User>
RCP<const User> Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::matrix_
protected

Definition at line 283 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
ArrayRCP<offset_t> Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::offset_
protected

Definition at line 285 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
ArrayRCP<gno_t> Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::columnIds_
protected

Definition at line 286 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
ArrayRCP<scalar_t> Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::values_
protected

Definition at line 287 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ConstOffsetsHostView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::offsHost_
protected

Definition at line 289 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ConstIdsHostView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::colIdsHost_
protected

Definition at line 290 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ScalarsHostView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::valuesHost_
protected

Definition at line 291 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ConstOffsetsDeviceView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::offsDevice_
protected

Definition at line 293 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ConstIdsDeviceView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::colIdsDevice_
protected

Definition at line 294 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::ScalarsDeviceView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::valuesDevice_
protected

Definition at line 295 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
int Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::nWeightsPerRow_
protected

Definition at line 297 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
ArrayRCP<StridedData<lno_t, scalar_t> > Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::rowWeights_
protected

Definition at line 298 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Base::WeightsDeviceView Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::rowWeightsDevice_
protected

Definition at line 299 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
Kokkos::View<bool *, host_t> Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::numNzWeight_
protected

Definition at line 300 of file Zoltan2_TpetraRowMatrixAdapter.hpp.

template<typename User , typename UserCoord = User>
bool Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::mayHaveDiagonalEntries
protected

Definition at line 302 of file Zoltan2_TpetraRowMatrixAdapter.hpp.


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