Zoltan2
Loading...
Searching...
No Matches
Public Types | Public Member Functions | List of all members
KokkosVectorAdapter< User > Class Template Reference
Inheritance diagram for KokkosVectorAdapter< User >:
Inheritance graph
[legend]
Collaboration diagram for KokkosVectorAdapter< User >:
Collaboration graph
[legend]

Public Types

typedef Zoltan2::InputTraits< User >::gno_t gno_t
 
typedef Zoltan2::InputTraits< User >::scalar_t scalar_t
 
typedef Tpetra::Map ::node_type node_t
 
typedef node_t::device_type device_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

 KokkosVectorAdapter (const size_t nids_, const gno_t *gids_, const int dim_, const scalar_t *coords_, const scalar_t *weights_=NULL)
 
size_t getLocalNumIDs () const
 Returns the number of objects on this process.
 
void getIDsView (const gno_t *&ids) const override
 
virtual void getIDsKokkosView (Kokkos::View< const gno_t *, device_t > &ids) const
 
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.
 
void getWeightsView (const scalar_t *&wgt, int &stride, int idx=0) const override
 
virtual void getWeightsKokkosView (Kokkos::View< scalar_t **, device_t > &wgt) const
 
int getNumEntriesPerID () const
 Return the number of vectors.
 
void getEntriesView (const scalar_t *&elements, int &stride, int idx=0) const override
 
virtual void getEntriesKokkosView (Kokkos::View< scalar_t **, Kokkos::LayoutLeft, device_t > &coo) const
 
- Public Member Functions inherited from Zoltan2::VectorAdapter< User >
virtual ~VectorAdapter ()
 Destructor.
 
enum BaseAdapterType adapterType () const override
 Returns the type of adapter.
 
virtual void getEntriesView (const scalar_t *&elements, int &stride, int idx=0) const
 Provide a pointer to the elements of the specified vector.
 
virtual void getEntriesKokkosView (Kokkos::View< scalar_t **, Kokkos::LayoutLeft, typename node_t::device_type > &elements) const
 Provide a Kokkos view to the elements of the specified vector.
 
void generateFiles (const char *fileprefix, const Teuchos::Comm< int > &comm) const
 Write files that can be used as input to Zoltan or Zoltan2 driver Creates chaco-formatted input files for coordinates and weights that can be used as input for Zoltan or Zoltan2 drivers. This routine is SERIAL and can be quite slow. It is meant as a debugging tool only, to allow Zoltan developers to replicate performance that applications are seeing using the applicatios' input.
 
int getDimension () const
 
void getCoordinatesView (const scalar_t *&elements, int &stride, int idx=0) const override
 
void getCoordinatesKokkosView (Kokkos::View< scalar_t **, Kokkos::LayoutLeft, typename node_t::device_type > &elements) const override
 
- Public Member Functions inherited from Zoltan2::AdapterWithCoords< User >
virtual void getCoordinatesView (const typename BaseAdapter< User >::scalar_t *&coords, int &stride, int coordDim) const =0
 
virtual void getCoordinatesKokkosView (Kokkos::View< typename BaseAdapter< User >::scalar_t **, Kokkos::LayoutLeft, typename BaseAdapter< User >::node_t::device_type > &elements) const =0
 
- Public Member Functions inherited from Zoltan2::BaseAdapter< User >
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 ()
 

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 KokkosVectorAdapter< User >

Definition at line 151 of file mj_backwardcompat.cpp.

Member Typedef Documentation

◆ gno_t

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

Definition at line 154 of file mj_backwardcompat.cpp.

◆ scalar_t

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

Definition at line 155 of file mj_backwardcompat.cpp.

◆ node_t

template<typename User >
typedef Tpetra::Map ::node_type KokkosVectorAdapter< User >::node_t

Definition at line 156 of file mj_backwardcompat.cpp.

◆ device_t

template<typename User >
typedef node_t::device_type KokkosVectorAdapter< User >::device_t

Definition at line 157 of file mj_backwardcompat.cpp.

Constructor & Destructor Documentation

◆ KokkosVectorAdapter()

template<typename User >
KokkosVectorAdapter< User >::KokkosVectorAdapter ( const size_t nids_,
const gno_t * gids_,
const int dim_,
const scalar_t * coords_,
const scalar_t * weights_ = NULL )
inline

Definition at line 159 of file mj_backwardcompat.cpp.

Member Function Documentation

◆ getLocalNumIDs()

template<typename User >
size_t KokkosVectorAdapter< User >::getLocalNumIDs ( ) const
inlinevirtual

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 218 of file mj_backwardcompat.cpp.

◆ getIDsView()

template<typename User >
void KokkosVectorAdapter< User >::getIDsView ( const gno_t *& ids) const
inlineoverride

Definition at line 220 of file mj_backwardcompat.cpp.

◆ getIDsKokkosView()

template<typename User >
virtual void KokkosVectorAdapter< User >::getIDsKokkosView ( Kokkos::View< const gno_t *, device_t > & ids) const
inlinevirtual

Definition at line 225 of file mj_backwardcompat.cpp.

◆ getNumWeightsPerID()

template<typename User >
int KokkosVectorAdapter< User >::getNumWeightsPerID ( ) const
inlinevirtual

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 229 of file mj_backwardcompat.cpp.

◆ getWeightsView()

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

Definition at line 231 of file mj_backwardcompat.cpp.

◆ getWeightsKokkosView()

template<typename User >
virtual void KokkosVectorAdapter< User >::getWeightsKokkosView ( Kokkos::View< scalar_t **, device_t > & wgt) const
inlinevirtual

Definition at line 240 of file mj_backwardcompat.cpp.

◆ getNumEntriesPerID()

template<typename User >
int KokkosVectorAdapter< User >::getNumEntriesPerID ( ) const
inlinevirtual

Return the number of vectors.

Implements Zoltan2::VectorAdapter< User >.

Definition at line 244 of file mj_backwardcompat.cpp.

◆ getEntriesView()

template<typename User >
void KokkosVectorAdapter< User >::getEntriesView ( const scalar_t *& elements,
int & stride,
int idx = 0 ) const
inlineoverride

Definition at line 246 of file mj_backwardcompat.cpp.

◆ getEntriesKokkosView()

template<typename User >
virtual void KokkosVectorAdapter< User >::getEntriesKokkosView ( Kokkos::View< scalar_t **, Kokkos::LayoutLeft, device_t > & coo) const
inlinevirtual

Definition at line 252 of file mj_backwardcompat.cpp.


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