Zoltan2
 All Classes Namespaces Files Functions Variables Typedefs Enumerations Enumerator Macros Pages
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::AdapterWithCoords< User >
using scalar_t = typename BaseAdapter< User >::scalar_t
 
using device_t = typename BaseAdapter< User >::node_t::device_type
 
using host_t = typename Kokkos::HostSpace::memory_space
 
using CoordsDeviceView = Kokkos::View< scalar_t **, Kokkos::LayoutLeft, device_t >
 
using CoordsHostView = typename CoordsDeviceView::HostMirror
 
- 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
 

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. More...
 
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. More...
 
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. More...
 
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 >
enum BaseAdapterType adapterType () const override
 Returns the type of adapter. More...
 
virtual void getEntriesView (const scalar_t *&elements, int &stride, int idx=0) const
 Provide a pointer to the elements of the specified vector. More...
 
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. More...
 
virtual void getEntriesHostView (typename AdapterWithCoords< User >::CoordsHostView &elements) const
 Provide a Kokkos view (Host side) to the elements of the specified vector. More...
 
virtual void getEntriesDeviceView (typename AdapterWithCoords< User >::CoordsDeviceView &elements) const
 Provide a Kokkos view (Device side) to the elements of the specified vector. More...
 
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. More...
 
int getDimension () const
 
void getCoordinatesView (const scalar_t *&elements, int &stride, int idx=0) const override
 
void getCoordinatesKokkosView (typename AdapterWithCoords< User >::CoordsDeviceView &elements) const override
 
void getCoordinatesHostView (typename AdapterWithCoords< User >::CoordsHostView &elements) const override
 
void getCoordinatesDeviceView (typename AdapterWithCoords< User >::CoordsDeviceView &elements) const override
 
- Public Member Functions inherited from Zoltan2::AdapterWithCoords< User >
virtual void getCoordinatesKokkosView (CoordsDeviceView &elements) const =0
 
virtual void getCoordinatesHostView (CoordsHostView &) const
 
virtual void getCoordinatesDeviceView (CoordsDeviceView &elements) const
 
- Public Member Functions inherited from Zoltan2::BaseAdapter< User >
virtual void getIDsView (const gno_t *&ids) const
 Provide a pointer to this process' identifiers. More...
 
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 getWeightsView (const scalar_t *&wgt, int &stride, int idx=0) const
 Provide pointer to a weight array with stride. 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
 

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

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

Definition at line 154 of file mj_backwardcompat.cpp.

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

Definition at line 155 of file mj_backwardcompat.cpp.

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

Definition at line 156 of file mj_backwardcompat.cpp.

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

Definition at line 157 of file mj_backwardcompat.cpp.

Constructor & Destructor Documentation

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

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.

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

Definition at line 220 of file mj_backwardcompat.cpp.

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.

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.

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.

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.

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.

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.

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:
Generated by   doxygen 1.8.5