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;
88 #ifdef TPETRA_ENABLE_DEPRECATED_CODE 89 typedef typename MatrixTraits<matrix_t>::sparse_ptr_type spmtx_ptr_t;
90 typedef typename MatrixTraits<matrix_t>::sparse_idx_type spmtx_idx_t;
91 typedef typename MatrixTraits<matrix_t>::sparse_values_type spmtx_vals_t;
95 typedef ConcreteMatrixAdapter<matrix_t> type;
96 typedef typename MatrixTraits<matrix_t>::major_access major_access;
98 ConcreteMatrixAdapter(Teuchos::RCP<matrix_t> m);
100 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;
102 const Teuchos::RCP<const Teuchos::Comm<int> > getComm_impl()
const;
103 global_size_t getGlobalNumRows_impl()
const;
104 global_size_t getGlobalNumCols_impl()
const;
105 global_size_t getGlobalNNZ_impl()
const;
106 #ifdef TPETRA_ENABLE_DEPRECATED_CODE 107 spmtx_ptr_t getSparseRowPtr()
const;
108 spmtx_idx_t getSparseColInd()
const;
109 spmtx_vals_t getSparseValues()
const;
113 void getSparseRowPtr_kokkos_view(KV & view)
const {
114 deep_copy_or_assign_view(view, this->mat_->graph.row_map);
118 void getSparseColInd_kokkos_view(KV & view)
const {
119 deep_copy_or_assign_view(view, this->mat_->graph.entries);
123 void getSparseValues_kokkos_view(KV & view)
const {
124 deep_copy_or_assign_view(view, this->mat_->values);
127 size_t getGlobalRowNNZ_impl(global_ordinal_t row)
const;
128 size_t getLocalRowNNZ_impl(local_ordinal_t row)
const;
129 size_t getGlobalColNNZ_impl(global_ordinal_t col)
const;
130 size_t getLocalColNNZ_impl(local_ordinal_t col)
const;
132 global_size_t getRowIndexBase()
const;
133 global_size_t getColumnIndexBase()
const;
135 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
140 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
143 getRowMap_impl()
const;
145 const Teuchos::RCP<
const Tpetra::Map<local_ordinal_t,
148 getColMap_impl()
const;
151 void getGlobalRowCopy_impl(global_ordinal_t row,
152 const Teuchos::ArrayView<global_ordinal_t>& indices,
153 const Teuchos::ArrayView<scalar_t>& vals,
156 void getGlobalColCopy_impl(global_ordinal_t col,
157 const Teuchos::ArrayView<global_ordinal_t>& indices,
158 const Teuchos::ArrayView<scalar_t>& vals,
162 template <
typename KV_GO,
typename KV_S>
163 void getGlobalRowCopy_kokkos_view_impl(global_ordinal_t row,
172 #endif // AMESOS2_KOKKOSCRSMATRIX_MATRIXADAPTER_DECL_HPP Copy or assign views based on memory spaces.
Definition: Amesos2_AbstractConcreteMatrixAdapter.hpp:48
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
Definition: Amesos2_TypeDecl.hpp:127
EDistribution
Definition: Amesos2_TypeDecl.hpp:123