Zoltan2
|
Provides access for Zoltan2 to Tpetra::RowMatrix data. More...
#include <Zoltan2_TpetraRowMatrixAdapter.hpp>
Public Member Functions | |
~TpetraRowMatrixAdapter () | |
Destructor. | |
TpetraRowMatrixAdapter (const RCP< const User > &inmatrix, int nWeightsPerRow=0) | |
Constructor. | |
void | setWeights (const scalar_t *weightVal, int stride, int idx=0) |
Specify a weight for each entity of the primaryEntityType. | |
void | setRowWeights (const scalar_t *weightVal, int stride, int idx=0) |
Specify a weight for each row. | |
void | setWeightIsDegree (int idx) |
Specify an index for which the weight should be the degree of the entity. | |
void | setRowWeightIsNumberOfNonZeros (int idx) |
Specify an index for which the row weight should be the global number of nonzeros in the row. | |
size_t | getLocalNumRows () const |
Returns the number of rows on this process. | |
size_t | getLocalNumColumns () const |
Returns the number of columns on this process. | |
size_t | getLocalNumEntries () const |
Returns the number of nonzeros on this process. | |
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. | |
void | getRowIDsView (const gno_t *&rowIds) const |
void | getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds) const |
void | getCRSView (ArrayRCP< const offset_t > &offsets, ArrayRCP< const gno_t > &colIds, ArrayRCP< const scalar_t > &values) const |
int | getNumWeightsPerRow () const |
Returns the number of weights per row (0 or greater). Row weights may be used when partitioning matrix rows. | |
void | getRowWeightsView (const scalar_t *&weights, int &stride, int idx=0) const |
Provide a pointer to the row weights, if any. | |
bool | useNumNonzerosAsRowWeight (int idx) const |
Indicate whether row weight with index idx should be the global number of nonzeros in the row. | |
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 |
![]() | |
enum BaseAdapterType | adapterType () const override |
Returns the type of adapter. | |
MatrixAdapter () | |
virtual | ~MatrixAdapter () |
Destructor. | |
virtual size_t | getLocalNumRows () const =0 |
Returns the number of rows on this process. | |
virtual size_t | getLocalNumColumns () const =0 |
Returns the number of columns on this process. | |
virtual size_t | getLocalNumEntries () const =0 |
Returns the number of nonzeros on this process. | |
virtual 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. | |
virtual void | getRowIDsView (const gno_t *&rowIds) const |
Sets pointer to this process' rows' global IDs. | |
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. | |
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. | |
virtual int | getNumWeightsPerRow () const |
Returns the number of weights per row (0 or greater). Row weights may be used when partitioning matrix rows. | |
virtual void | getRowWeightsView (const scalar_t *&weights, int &stride, int idx=0) const |
Provide a pointer to the row weights, if any. | |
virtual bool | useNumNonzerosAsRowWeight (int idx) const |
Indicate whether row weight with index idx should be the global number of nonzeros in the row. | |
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. | |
virtual void | getColumnIDsView (const gno_t *&colIds) const |
Sets pointer to this process' columns' global IDs. | |
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. | |
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. | |
virtual int | getNumWeightsPerColumn () const |
Returns the number of weights per column (0 or greater). Column weights may be used when partitioning matrix columns. | |
virtual void | getColumnWeightsView (const scalar_t *&weights, int &stride, int idx=0) const |
Provide a pointer to the column weights, if any. | |
virtual bool | useNumNonzerosAsColumnWeight (int idx) const |
Indicate whether column weight with index idx should be the global number of nonzeros in the column. | |
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. | |
bool | coordinatesAvailable () const |
Indicate whether coordinate information has been set for this MatrixAdapter. | |
VectorAdapter< UserCoord > * | getCoordinateInput () const override |
Obtain the coordinate data registered by the user. | |
enum MatrixEntityType | getPrimaryEntityType () const |
Returns the entity to be partitioned, ordered, colored, etc. Valid values are MATRIX_ROW, MATRIX_COLUMN, MATRIX_NONZERO. | |
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. | |
size_t | getLocalNumIDs () const override |
Returns the number of objects on this process. | |
void | getIDsView (const gno_t *&Ids) const override |
Provide a pointer to this process' identifiers. | |
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. | |
void | getWeightsView (const scalar_t *&wgt, int &stride, int idx=0) const override |
bool | useDegreeAsWeight (int idx) const |
virtual void | setCoordinateInput (VectorAdapter< UserCoord > *coordData)=0 |
virtual VectorAdapter< UserCoord > * | getCoordinateInput () const =0 |
![]() | |
virtual enum BaseAdapterType | adapterType () const =0 |
Returns the type of adapter. | |
virtual | ~BaseAdapter () |
Destructor. | |
virtual void | getIDsView (const gno_t *&ids) const |
Provide a pointer to this process' identifiers. | |
virtual void | getIDsKokkosView (Kokkos::View< const gno_t *, typename node_t::device_type > &ids) const |
Provide a Kokkos view to this process' identifiers. | |
void | getPartsView (const part_t *&inputPart) const |
Provide pointer to a weight array with stride. | |
template<typename Adapter > | |
void | applyPartitioningSolution (const User &in, User *&out, const PartitioningSolution< Adapter > &solution) const |
Apply a PartitioningSolution to an input. | |
![]() | |
virtual | ~BaseAdapterRoot () |
virtual size_t | getLocalNumIDs () const =0 |
Returns the number of objects on this process. | |
virtual int | getNumWeightsPerID () const |
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. | |
Additional Inherited Members | |
![]() | |
typedef InputTraits< User >::lno_t | lno_t |
typedef InputTraits< User >::gno_t | gno_t |
typedef InputTraits< User >::scalar_t | scalar_t |
typedef InputTraits< User >::node_t | node_t |
typedef InputTraits< User >::part_t | part_t |
typedef InputTraits< User >::offset_t | offset_t |
![]() | |
void | generateWeightFileOnly (const char *fileprefix, const Teuchos::Comm< int > &comm) const |
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 76 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inline |
Destructor.
Definition at line 92 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::TpetraRowMatrixAdapter | ( | const RCP< const User > & | inmatrix, |
int | nWeightsPerRow = 0 |
||
) |
Constructor.
inmatrix | The user's Tpetra RowMatrix object |
nWeightsPerRow | If row weights will be provided in setRowWeights(), the set nWeightsPerRow to the number of weights per row. |
Definition at line 239 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeights | ( | const scalar_t * | weightVal, |
int | stride, | ||
int | idx = 0 |
||
) |
Specify a weight for each entity of the primaryEntityType.
weightVal | A pointer to the weights for this index. \stride A stride to be used in reading the values. The index idx weight for entity \k should be found at weightVal[k*stride] . |
idx | A 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 285 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setRowWeights | ( | const scalar_t * | weightVal, |
int | stride, | ||
int | idx = 0 |
||
) |
Specify a weight for each row.
weightVal | A pointer to the weights for this index. \stride A stride to be used in reading the values. The index idx weight for row \k should be found at weightVal[k*stride] . |
idx | A 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
Definition at line 303 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::setWeightIsDegree | ( | int | idx | ) |
Specify an index for which the weight should be the degree of the entity.
idx | Zoltan2 will use the entity's degree as the entity weight for index idx . |
Definition at line 322 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
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.
idx | Zoltan2 will use the global number of nonzeros in a row as the row weight for index idx . |
Definition at line 339 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
Returns the number of rows on this process.
Implements Zoltan2::MatrixAdapter< User, UserCoord >.
Definition at line 151 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
Returns the number of columns on this process.
Implements Zoltan2::MatrixAdapter< User, UserCoord >.
Definition at line 155 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
Returns the number of nonzeros on this process.
Implements Zoltan2::MatrixAdapter< User, UserCoord >.
Definition at line 159 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
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 163 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inline |
Definition at line 165 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inline |
Definition at line 171 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inline |
Definition at line 177 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
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 187 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
Provide a pointer to the row weights, if any.
weights | is the list of weights with a given index for the rows returned in getRowIDsView(). |
stride | The k'th weight is located at weights[stride*k] |
idx | ranges from zero to one less than getNumWeightsPerRow(). |
Reimplemented from Zoltan2::MatrixAdapter< User, UserCoord >.
Definition at line 189 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
|
inlinevirtual |
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 205 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::applyPartitioningSolution | ( | const User & | in, |
User *& | out, | ||
const PartitioningSolution< Adapter > & | solution | ||
) | const |
Definition at line 357 of file Zoltan2_TpetraRowMatrixAdapter.hpp.
void Zoltan2::TpetraRowMatrixAdapter< User, UserCoord >::applyPartitioningSolution | ( | const User & | in, |
RCP< User > & | out, | ||
const PartitioningSolution< Adapter > & | solution | ||
) | const |
Definition at line 380 of file Zoltan2_TpetraRowMatrixAdapter.hpp.