54 #ifndef AMESOS2_KOKKOSCRSMATRIX_MATRIXADAPTER_DECL_HPP
55 #define AMESOS2_KOKKOSCRSMATRIX_MATRIXADAPTER_DECL_HPP
57 #include "Amesos2_config.h"
59 #include "Amesos2_MatrixAdapter_decl.hpp"
60 #include "KokkosSparse_CrsMatrix.hpp"
73 template <
typename Scalar,
74 typename LocalOrdinal,
75 typename ExecutionSpace>
76 class ConcreteMatrixAdapter<KokkosSparse::CrsMatrix<Scalar, LocalOrdinal, ExecutionSpace>> :
77 public MatrixAdapter<KokkosSparse::CrsMatrix<Scalar, LocalOrdinal, ExecutionSpace>>
80 typedef KokkosSparse::CrsMatrix<Scalar, LocalOrdinal, ExecutionSpace> matrix_t;
83 typedef typename MatrixTraits<matrix_t>::scalar_t scalar_t;
84 typedef typename MatrixTraits<matrix_t>::local_ordinal_t local_ordinal_t;
85 typedef typename MatrixTraits<matrix_t>::global_ordinal_t global_ordinal_t;
86 typedef typename MatrixTraits<matrix_t>::node_t node_t;
87 typedef typename MatrixTraits<matrix_t>::global_size_t global_size_t;
90 typedef ConcreteMatrixAdapter<matrix_t> type;
91 typedef typename MatrixTraits<matrix_t>::major_access major_access;
93 ConcreteMatrixAdapter(Teuchos::RCP<matrix_t> m);
95 Teuchos::RCP<const MatrixAdapter<matrix_t> > get_impl(
const Teuchos::Ptr<
const Tpetra::Map<local_ordinal_t,global_ordinal_t,node_t> > map,
EDistribution distribution = ROOTED)
const;
97 const Teuchos::RCP<const Teuchos::Comm<int> > getComm_impl()
const;
98 global_size_t getGlobalNumRows_impl()
const;
99 global_size_t getGlobalNumCols_impl()
const;
100 global_size_t getGlobalNNZ_impl()
const;
103 void getSparseRowPtr_kokkos_view(KV & view)
const {
104 deep_copy_or_assign_view(view, this->mat_->graph.row_map);
108 void getSparseColInd_kokkos_view(KV & view)
const {
109 deep_copy_or_assign_view(view, this->mat_->graph.entries);
113 void getSparseValues_kokkos_view(KV & view)
const {
114 deep_copy_or_assign_view(view, this->mat_->values);
117 size_t getGlobalRowNNZ_impl(global_ordinal_t row)
const;
118 size_t getLocalRowNNZ_impl(local_ordinal_t row)
const;
119 size_t getGlobalColNNZ_impl(global_ordinal_t col)
const;
120 size_t getLocalColNNZ_impl(local_ordinal_t col)
const;
122 global_size_t getRowIndexBase()
const;
123 global_size_t getColumnIndexBase()
const;
125 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
130 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
133 getRowMap_impl()
const;
135 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
138 getColMap_impl()
const;
141 void getGlobalRowCopy_impl(global_ordinal_t row,
142 const Teuchos::ArrayView<global_ordinal_t>& indices,
143 const Teuchos::ArrayView<scalar_t>& vals,
146 void getGlobalColCopy_impl(global_ordinal_t col,
147 const Teuchos::ArrayView<global_ordinal_t>& indices,
148 const Teuchos::ArrayView<scalar_t>& vals,
152 template <
typename KV_GO,
typename KV_S>
153 void getGlobalRowCopy_kokkos_view_impl(global_ordinal_t row,
162 #endif // AMESOS2_KOKKOSCRSMATRIX_MATRIXADAPTER_DECL_HPP
Copy or assign views based on memory spaces.
A Matrix adapter interface for Amesos2.
Definition: Amesos2_MatrixAdapter_decl.hpp:76
Indicates that the concrete class can use the generic getC{c|r}s methods implemented in MatrixAdapter...
Definition: Amesos2_TypeDecl.hpp:91
EDistribution
Definition: Amesos2_TypeDecl.hpp:123