Xpetra_CrsGraph.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//             Xpetra: A linear algebra interface package
//
// Copyright 2012 NTESS and the Xpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef XPETRA_CRSGRAPH_HPP
#define XPETRA_CRSGRAPH_HPP
 
#include <Teuchos_ParameterList.hpp>
 
#include <Teuchos_Describable.hpp>
#include <Tpetra_KokkosCompat_DefaultNode.hpp>
#include "Xpetra_ConfigDefs.hpp"
#include "Xpetra_DistObject.hpp"
#include "Xpetra_Exceptions.hpp"
 
#include "Xpetra_Map.hpp"
 
#ifdef HAVE_XPETRA_TPETRA
#include <KokkosSparse_StaticCrsGraph.hpp>
#endif
 
namespace Xpetra {
 
using Teuchos::ParameterList;
 
struct RowInfo {
  size_t localRow;
  size_t allocSize;
  size_t numEntries;
  size_t offset1D;
};
 
enum ELocalGlobal {
  LocalIndices,
  GlobalIndices
};
 
template <class LocalOrdinal,
          class GlobalOrdinal,
          class Node = Tpetra::KokkosClassic::DefaultNode::DefaultNodeType>
class CrsGraph
  : /*public RowGraph<>,*/ public DistObject<GlobalOrdinal, LocalOrdinal, GlobalOrdinal, Node> {
 public:
  typedef LocalOrdinal local_ordinal_type;
  typedef GlobalOrdinal global_ordinal_type;
  typedef Node node_type;
 
  //@
 
  virtual ~CrsGraph() {}
 
 
 
 
  virtual void insertGlobalIndices(GlobalOrdinal globalRow, const ArrayView<const GlobalOrdinal> &indices) = 0;
 
  virtual void insertLocalIndices(const LocalOrdinal localRow, const ArrayView<const LocalOrdinal> &indices) = 0;
 
  virtual void removeLocalIndices(LocalOrdinal localRow) = 0;
 
  virtual void allocateAllIndices(size_t numNonZeros, ArrayRCP<size_t> &rowptr, ArrayRCP<LocalOrdinal> &colind) = 0;
 
  virtual void setAllIndices(const ArrayRCP<size_t> &rowptr, const ArrayRCP<LocalOrdinal> &colind) = 0;
 
  virtual void getAllIndices(ArrayRCP<const size_t> &rowptr, ArrayRCP<const LocalOrdinal> &colind) const = 0;
 
 
 
 
  virtual void fillComplete(const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > &domainMap, const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > &rangeMap, const RCP<ParameterList> &params = null) = 0;
 
  virtual void fillComplete(const RCP<ParameterList> &params = null) = 0;
 
  virtual void
  expertStaticFillComplete(const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > &domainMap,
                           const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > &rangeMap,
                           const RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> > &importer = null,
                           const RCP<const Export<LocalOrdinal, GlobalOrdinal, Node> > &exporter = null,
                           const RCP<Teuchos::ParameterList> &params                             = null) = 0;
 
 
 
 
  virtual RCP<const Comm<int> > getComm() const = 0;
 
  virtual RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getRowMap() const = 0;
 
  virtual RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getColMap() const = 0;
 
  virtual RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getDomainMap() const = 0;
 
  virtual RCP<const Map<LocalOrdinal, GlobalOrdinal, Node> > getRangeMap() const = 0;
 
  virtual RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> > getImporter() const = 0;
 
  virtual RCP<const Export<LocalOrdinal, GlobalOrdinal, Node> > getExporter() const = 0;
 
  virtual global_size_t getGlobalNumRows() const = 0;
 
  virtual global_size_t getGlobalNumCols() const = 0;
 
  virtual size_t getLocalNumRows() const = 0;
 
  virtual size_t getLocalNumCols() const = 0;
 
  virtual GlobalOrdinal getIndexBase() const = 0;
 
  virtual global_size_t getGlobalNumEntries() const = 0;
 
  virtual size_t getLocalNumEntries() const = 0;
 
  virtual size_t getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const = 0;
 
  virtual size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const = 0;
 
  virtual size_t getNumAllocatedEntriesInGlobalRow(GlobalOrdinal globalRow) const = 0;
 
  virtual size_t getNumAllocatedEntriesInLocalRow(LocalOrdinal localRow) const = 0;
 
  virtual size_t getGlobalMaxNumRowEntries() const = 0;
 
  virtual size_t getLocalMaxNumRowEntries() const = 0;
 
  virtual bool hasColMap() const = 0;
 
  virtual bool isLocallyIndexed() const = 0;
 
  virtual bool isGloballyIndexed() const = 0;
 
  virtual bool isFillComplete() const = 0;
 
  virtual bool isStorageOptimized() const = 0;
 
  virtual void getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView<const GlobalOrdinal> &Indices) const = 0;
 
  virtual void getLocalRowView(LocalOrdinal LocalRow, ArrayView<const LocalOrdinal> &indices) const = 0;
 
  virtual void computeGlobalConstants() = 0;
 
 
 
#ifdef HAVE_XPETRA_TPETRA
  typedef typename node_type::execution_space execution_space;
  typedef typename node_type::device_type device_type;
  typedef KokkosSparse::StaticCrsGraph<LocalOrdinal, Kokkos::LayoutLeft, device_type, void, size_t> local_graph_type;
 
#if KOKKOS_VERSION >= 40799
  virtual typename local_graph_type::host_mirror_type getLocalGraphHost() const = 0;
#else
  virtual typename local_graph_type::HostMirror getLocalGraphHost() const = 0;
#endif
  virtual local_graph_type getLocalGraphDevice() const = 0;
 
  virtual void getLocalDiagOffsets(const Kokkos::View<size_t *, device_type, Kokkos::MemoryUnmanaged> &offsets) const = 0;
 
#else
#ifdef __GNUC__
#warning "Xpetra Kokkos interface for CrsMatrix is enabled (HAVE_XPETRA_KOKKOS_REFACTOR) but Tpetra is disabled. The Kokkos interface needs Tpetra to be enabled, too."
#endif
#endif
 
 
 
 
  virtual std::string description() const = 0;
 
  virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel = Teuchos::Describable::verbLevel_default) const = 0;
 
 
};  // CrsGraph class
 
}  // namespace Xpetra
 
#define XPETRA_CRSGRAPH_SHORT
#endif  // XPETRA_CRSGRAPH_HPP