Zoltan2
Loading...
Searching...
No Matches
Public Types | Public Member Functions | List of all members
Zoltan2::BasicKokkosIdentifierAdapter< User > Class Template Reference

This class represents a collection of global Identifiers and their associated weights, if any. More...

#include <Zoltan2_BasicKokkosIdentifierAdapter.hpp>

Inheritance diagram for Zoltan2::BasicKokkosIdentifierAdapter< User >:
Inheritance graph
[legend]
Collaboration diagram for Zoltan2::BasicKokkosIdentifierAdapter< User >:
Collaboration graph
[legend]

Public Types

typedef InputTraits< User >::scalar_t scalar_t
 
typedef InputTraits< User >::lno_t lno_t
 
typedef InputTraits< User >::gno_t gno_t
 
typedef InputTraits< User >::part_t part_t
 
typedef InputTraits< User >::node_t node_t
 
typedef node_t::device_type device_t
 
typedef User user_t
 
- Public Types inherited from Zoltan2::BaseAdapter< User >
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
 

Public Member Functions

 BasicKokkosIdentifierAdapter (Kokkos::View< gno_t *, device_t > &ids, Kokkos::View< scalar_t **, device_t > &weights)
 Constructor.
 
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.
 
void getIDsKokkosView (Kokkos::View< const gno_t *, device_t > &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.
 
void getWeightsView (const scalar_t *&wgt, int &stride, int idx=0) const override
 
void getWeightsKokkosView (Kokkos::View< scalar_t **, device_t > &wgts) const override
 
- Public Member Functions inherited from Zoltan2::IdentifierAdapter< User >
virtual ~IdentifierAdapter ()
 Destructor.
 
enum BaseAdapterType adapterType () const
 Returns the type of adapter.
 
- Public Member Functions inherited from Zoltan2::BaseAdapter< User >
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.
 
- Public Member Functions inherited from Zoltan2::BaseAdapterRoot
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

- Protected Member Functions inherited from Zoltan2::BaseAdapter< User >
void generateWeightFileOnly (const char *fileprefix, const Teuchos::Comm< int > &comm) const
 

Detailed Description

template<typename User>
class Zoltan2::BasicKokkosIdentifierAdapter< User >

This class represents a collection of global Identifiers and their associated weights, if any.

The user supplies the identifiers and weights by way of pointers to arrays.

The template parameter (User) is a C++ class type which provides the actual data types with which the Zoltan2 library will be compiled, through a Traits mechanism. User may be the actual class used by application to represent coordinates, or it may be the empty helper class BasicUserTypes with which a Zoltan2 user can easily supply the data types for the library.

The scalar_t type, representing use data such as matrix values, is used by Zoltan2 for weights, coordinates, part sizes and quality metrics. Some User types (like Tpetra::CrsMatrix) have an inherent scalar type, and some (like Tpetra::CrsGraph) 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.

Examples
kokkosBlock.cpp.

Definition at line 84 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

Member Typedef Documentation

◆ scalar_t

template<typename User >
typedef InputTraits<User>::scalar_t Zoltan2::BasicKokkosIdentifierAdapter< User >::scalar_t

Definition at line 87 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ lno_t

template<typename User >
typedef InputTraits<User>::lno_t Zoltan2::BasicKokkosIdentifierAdapter< User >::lno_t

Definition at line 88 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ gno_t

template<typename User >
typedef InputTraits<User>::gno_t Zoltan2::BasicKokkosIdentifierAdapter< User >::gno_t

Definition at line 89 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ part_t

template<typename User >
typedef InputTraits<User>::part_t Zoltan2::BasicKokkosIdentifierAdapter< User >::part_t

Definition at line 90 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ node_t

template<typename User >
typedef InputTraits<User>::node_t Zoltan2::BasicKokkosIdentifierAdapter< User >::node_t

Definition at line 91 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ device_t

template<typename User >
typedef node_t::device_type Zoltan2::BasicKokkosIdentifierAdapter< User >::device_t

Definition at line 92 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ user_t

template<typename User >
typedef User Zoltan2::BasicKokkosIdentifierAdapter< User >::user_t

Definition at line 93 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

Constructor & Destructor Documentation

◆ BasicKokkosIdentifierAdapter()

template<typename User >
Zoltan2::BasicKokkosIdentifierAdapter< User >::BasicKokkosIdentifierAdapter ( Kokkos::View< gno_t *, device_t > &  ids,
Kokkos::View< scalar_t **, device_t > &  weights 
)

Constructor.

Parameters
idsshould point to a View of identifiers.
weightsa list of pointers to arrays of weights. The number of weights per identifier is assumed to be weights.extent(0).
weightStridesa list of strides for the weights. The weight for weight index n for ids[k] should be found at weights[n][weightStrides[n] * k]. If weightStrides.size() is zero, it is assumed all strides are one.

The values pointed to the arguments must remain valid for the lifetime of this Adapter.

Definition at line 156 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

Member Function Documentation

◆ getLocalNumIDs()

template<typename User >
size_t Zoltan2::BasicKokkosIdentifierAdapter< User >::getLocalNumIDs ( ) const
inlineoverridevirtual

Returns the number of objects on this process.

Objects may be coordinates, graph vertices, matrix rows, etc. They are the objects to be partitioned, ordered, or colored.

Implements Zoltan2::BaseAdapterRoot.

Definition at line 116 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ getIDsView()

template<typename User >
void Zoltan2::BasicKokkosIdentifierAdapter< User >::getIDsView ( const gno_t *&  ids) const
inlineoverridevirtual

Provide a pointer to this process' identifiers.

Parameters
idswill on return point to the list of the global Ids for this process.

Reimplemented from Zoltan2::BaseAdapter< User >.

Definition at line 120 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ getIDsKokkosView()

template<typename User >
void Zoltan2::BasicKokkosIdentifierAdapter< User >::getIDsKokkosView ( Kokkos::View< const gno_t *, device_t > &  ids) const
inlineoverride

Definition at line 125 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ getNumWeightsPerID()

template<typename User >
int Zoltan2::BasicKokkosIdentifierAdapter< User >::getNumWeightsPerID ( ) const
inlineoverridevirtual

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.

Reimplemented from Zoltan2::BaseAdapterRoot.

Definition at line 129 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ getWeightsView()

template<typename User >
void Zoltan2::BasicKokkosIdentifierAdapter< User >::getWeightsView ( const scalar_t *&  wgt,
int &  stride,
int  idx = 0 
) const
inlineoverride

Definition at line 133 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.

◆ getWeightsKokkosView()

template<typename User >
void Zoltan2::BasicKokkosIdentifierAdapter< User >::getWeightsKokkosView ( Kokkos::View< scalar_t **, device_t > &  wgts) const
inlineoverride

Definition at line 142 of file Zoltan2_BasicKokkosIdentifierAdapter.hpp.


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