Xpetra Version of the Day
Loading...
Searching...
No Matches
Xpetra_TpetraCrsMatrix_def.hpp
Go to the documentation of this file.
1// @HEADER
2// *****************************************************************************
3// Xpetra: A linear algebra interface package
4//
5// Copyright 2012 NTESS and the Xpetra contributors.
6// SPDX-License-Identifier: BSD-3-Clause
7// *****************************************************************************
8// @HEADER
9
10#ifndef XPETRA_TPETRACRSMATRIX_DEF_HPP
11#define XPETRA_TPETRACRSMATRIX_DEF_HPP
12
13#include <Xpetra_MultiVectorFactory.hpp>
14#include "Xpetra_TpetraCrsMatrix_decl.hpp"
15#include "Tpetra_Details_residual.hpp"
16
17namespace Xpetra {
18
19template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
20TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap, size_t maxNumEntriesPerRow, const Teuchos::RCP<Teuchos::ParameterList> &params)
21 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), maxNumEntriesPerRow, params))) {}
22
23template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
24TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap, const ArrayRCP<const size_t> &NumEntriesPerRowToAlloc, const Teuchos::RCP<Teuchos::ParameterList> &params)
25 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), NumEntriesPerRowToAlloc(), params))) {}
26
27template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
28TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap, const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap, size_t maxNumEntriesPerRow, const Teuchos::RCP<Teuchos::ParameterList> &params)
29 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), toTpetra(colMap), maxNumEntriesPerRow, params))) {}
30
31template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
32TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap, const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap, const ArrayRCP<const size_t> &NumEntriesPerRowToAlloc, const Teuchos::RCP<Teuchos::ParameterList> &params)
33 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), toTpetra(colMap), NumEntriesPerRowToAlloc(), params))) {}
34
35template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
36TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const CrsGraph<LocalOrdinal, GlobalOrdinal, Node>> &graph, const Teuchos::RCP<Teuchos::ParameterList> &params)
37 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(graph), params))) {}
38
39template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
42
43template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
44TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const CrsMatrix> &sourceMatrix,
45 const Import<LocalOrdinal, GlobalOrdinal, Node> &importer,
46 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
47 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
48 const Teuchos::RCP<Teuchos::ParameterList> &params) {
49 typedef Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> MyTpetraCrsMatrix;
50 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, *sourceMatrix, tSourceMatrix, "Xpetra::TpetraCrsMatrix constructor only accepts Xpetra::TpetraCrsMatrix as the input argument."); // TODO: remove and use toTpetra()
51 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSourceMatrix.getTpetra_CrsMatrix();
52
53 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myDomainMap = domainMap != Teuchos::null ? toTpetra(domainMap) : Teuchos::null;
54 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myRangeMap = rangeMap != Teuchos::null ? toTpetra(rangeMap) : Teuchos::null;
55 mtx_ = Tpetra::importAndFillCompleteCrsMatrix<MyTpetraCrsMatrix>(tSourceMatrix.getTpetra_CrsMatrix(), toTpetra(importer), myDomainMap, myRangeMap, params);
56 bool restrictComm = false;
57 if (!params.is_null()) restrictComm = params->get("Restrict Communicator", restrictComm);
58 if (restrictComm && mtx_->getRowMap().is_null()) mtx_ = Teuchos::null;
59}
60
61template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
62TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const CrsMatrix> &sourceMatrix,
63 const Export<LocalOrdinal, GlobalOrdinal, Node> &exporter,
64 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
65 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
66 const Teuchos::RCP<Teuchos::ParameterList> &params) {
67 typedef Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> MyTpetraCrsMatrix;
68 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, *sourceMatrix, tSourceMatrix, "Xpetra::TpetraCrsMatrix constructor only accepts Xpetra::TpetraCrsMatrix as the input argument."); // TODO: remove and use toTpetra()
69 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSourceMatrix.getTpetra_CrsMatrix();
70
71 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myDomainMap = domainMap != Teuchos::null ? toTpetra(domainMap) : Teuchos::null;
72 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myRangeMap = rangeMap != Teuchos::null ? toTpetra(rangeMap) : Teuchos::null;
73 mtx_ = Tpetra::exportAndFillCompleteCrsMatrix<MyTpetraCrsMatrix>(tSourceMatrix.getTpetra_CrsMatrix(), toTpetra(exporter), myDomainMap, myRangeMap, params);
74}
75
76template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
77TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const CrsMatrix> &sourceMatrix,
78 const Import<LocalOrdinal, GlobalOrdinal, Node> &RowImporter,
79 const Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> DomainImporter,
80 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
81 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
82 const Teuchos::RCP<Teuchos::ParameterList> &params) {
83 typedef Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> MyTpetraCrsMatrix;
84 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, *sourceMatrix, tSourceMatrix, "Xpetra::TpetraCrsMatrix constructor only accepts Xpetra::TpetraCrsMatrix as the input argument."); // TODO: remove and use toTpetra()
85 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSourceMatrix.getTpetra_CrsMatrix();
86
87 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myDomainMap = domainMap != Teuchos::null ? toTpetra(domainMap) : Teuchos::null;
88 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myRangeMap = rangeMap != Teuchos::null ? toTpetra(rangeMap) : Teuchos::null;
89
90 mtx_ = Tpetra::importAndFillCompleteCrsMatrix<MyTpetraCrsMatrix>(tSourceMatrix.getTpetra_CrsMatrix(), toTpetra(RowImporter), toTpetra(*DomainImporter), myDomainMap, myRangeMap, params);
91 bool restrictComm = false;
92 if (!params.is_null()) restrictComm = params->get("Restrict Communicator", restrictComm);
93 if (restrictComm && mtx_->getRowMap().is_null()) mtx_ = Teuchos::null;
94}
95
96template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
97TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const CrsMatrix> &sourceMatrix,
98 const Export<LocalOrdinal, GlobalOrdinal, Node> &RowExporter,
99 const Teuchos::RCP<const Export<LocalOrdinal, GlobalOrdinal, Node>> DomainExporter,
100 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
101 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
102 const Teuchos::RCP<Teuchos::ParameterList> &params) {
103 typedef Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node> MyTpetraCrsMatrix;
104 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, *sourceMatrix, tSourceMatrix, "Xpetra::TpetraCrsMatrix constructor only accepts Xpetra::TpetraCrsMatrix as the input argument."); // TODO: remove and use toTpetra()
105 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSourceMatrix.getTpetra_CrsMatrix();
106
107 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myDomainMap = domainMap != Teuchos::null ? toTpetra(domainMap) : Teuchos::null;
108 RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> myRangeMap = rangeMap != Teuchos::null ? toTpetra(rangeMap) : Teuchos::null;
109
110 mtx_ = Tpetra::exportAndFillCompleteCrsMatrix<MyTpetraCrsMatrix>(tSourceMatrix.getTpetra_CrsMatrix(), toTpetra(RowExporter), toTpetra(*DomainExporter), myDomainMap, myRangeMap, params);
111}
112
114
115#ifdef HAVE_XPETRA_TPETRA
116template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
117TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
118 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
119 const local_matrix_type &lclMatrix,
120 const Teuchos::RCP<Teuchos::ParameterList> &params)
121 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), toTpetra(colMap), lclMatrix, params))) {}
122
123template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
124TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(
125 const local_matrix_type &lclMatrix,
126 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
127 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
128 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
129 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
130 const Teuchos::RCP<Teuchos::ParameterList> &params)
131 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(lclMatrix, toTpetra(rowMap), toTpetra(colMap), toTpetra(domainMap), toTpetra(rangeMap), params))) {}
132
133template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
134TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(
135 const local_matrix_type &lclMatrix,
136 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
137 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
138 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
139 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
140 const Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &importer,
141 const Teuchos::RCP<const Export<LocalOrdinal, GlobalOrdinal, Node>> &exporter,
142 const Teuchos::RCP<Teuchos::ParameterList> &params)
143 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(lclMatrix, toTpetra(rowMap), toTpetra(colMap), toTpetra(domainMap), toTpetra(rangeMap), toTpetra(importer), toTpetra(exporter), params))) {}
144#endif
145
146template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
147TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::~TpetraCrsMatrix() {}
148
149template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
150void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::insertGlobalValues(GlobalOrdinal globalRow, const ArrayView<const GlobalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
151 XPETRA_MONITOR("TpetraCrsMatrix::insertGlobalValues");
152 mtx_->insertGlobalValues(globalRow, cols, vals);
153}
154
155template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
156void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::insertLocalValues(LocalOrdinal localRow, const ArrayView<const LocalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
157 XPETRA_MONITOR("TpetraCrsMatrix::insertLocalValues");
158 mtx_->insertLocalValues(localRow, cols, vals);
159}
160
161template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
162void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceGlobalValues(GlobalOrdinal globalRow, const ArrayView<const GlobalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
163 XPETRA_MONITOR("TpetraCrsMatrix::replaceGlobalValues");
164 mtx_->replaceGlobalValues(globalRow, cols, vals);
165}
166
167template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
168void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceLocalValues(LocalOrdinal localRow,
169 const ArrayView<const LocalOrdinal> &cols,
170 const ArrayView<const Scalar> &vals) {
171 XPETRA_MONITOR("TpetraCrsMatrix::replaceLocalValues");
172 typedef typename ArrayView<const LocalOrdinal>::size_type size_type;
173 const LocalOrdinal numValid =
174 mtx_->replaceLocalValues(localRow, cols, vals);
175 TEUCHOS_TEST_FOR_EXCEPTION(
176 static_cast<size_type>(numValid) != cols.size(), std::runtime_error,
177 "replaceLocalValues returned " << numValid << " != cols.size() = " << cols.size() << ".");
178}
179
180template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
181void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setAllToScalar(const Scalar &alpha) {
182 XPETRA_MONITOR("TpetraCrsMatrix::setAllToScalar");
183 mtx_->setAllToScalar(alpha);
184}
185
186template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
187void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::scale(const Scalar &alpha) {
188 XPETRA_MONITOR("TpetraCrsMatrix::scale");
189 mtx_->scale(alpha);
190}
191
192template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
193void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::allocateAllValues(size_t numNonZeros, ArrayRCP<size_t> &rowptr, ArrayRCP<LocalOrdinal> &colind, ArrayRCP<Scalar> &values) {
194 XPETRA_MONITOR("TpetraCrsMatrix::allocateAllValues");
195 rowptr.resize(getLocalNumRows() + 1);
196 colind.resize(numNonZeros);
197 values.resize(numNonZeros);
198}
199
200template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
201void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setAllValues(const ArrayRCP<size_t> &rowptr, const ArrayRCP<LocalOrdinal> &colind, const ArrayRCP<Scalar> &values) {
202 XPETRA_MONITOR("TpetraCrsMatrix::setAllValues");
203 mtx_->setAllValues(rowptr, colind, values);
204}
205
206template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
207void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getAllValues(ArrayRCP<const size_t> &rowptr, ArrayRCP<const LocalOrdinal> &colind, ArrayRCP<const Scalar> &values) const {
208 XPETRA_MONITOR("TpetraCrsMatrix::getAllValues");
209 // TODO: Change getAllValues interface to return Kokkos views,
210 // TODO: or add getLocalRowPtrsHost, getLocalIndicesHost, etc., interfaces
211 // TODO: and use them in MueLu
212 rowptr = Kokkos::Compat::persistingView(mtx_->getLocalRowPtrsHost());
213 colind = Kokkos::Compat::persistingView(mtx_->getLocalIndicesHost());
214 values = Teuchos::arcp_reinterpret_cast<const Scalar>(
215 Kokkos::Compat::persistingView(mtx_->getLocalValuesHost(
216 Tpetra::Access::ReadOnly)));
217}
218
219template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
220void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getAllValues(ArrayRCP<Scalar> &values) {
221 XPETRA_MONITOR("TpetraCrsMatrix::getAllValues");
222 // TODO: Change getAllValues interface to return Kokkos view,
223 // TODO: or add getLocalValuesDevice() interfaces
224 values = Teuchos::arcp_reinterpret_cast<Scalar>(
225 Kokkos::Compat::persistingView(mtx_->getLocalValuesDevice(
226 Tpetra::Access::ReadWrite)));
227}
228
229template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
230bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::haveGlobalConstants() const { return mtx_->haveGlobalConstants(); }
231
232template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
233void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::resumeFill(const RCP<ParameterList> &params) {
234 XPETRA_MONITOR("TpetraCrsMatrix::resumeFill");
235 mtx_->resumeFill(params);
236}
237
238template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
239void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::fillComplete(const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap, const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap, const RCP<ParameterList> &params) {
240 XPETRA_MONITOR("TpetraCrsMatrix::fillComplete");
241 mtx_->fillComplete(toTpetra(domainMap), toTpetra(rangeMap), params);
242}
243
244template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
245void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::fillComplete(const RCP<ParameterList> &params) {
246 XPETRA_MONITOR("TpetraCrsMatrix::fillComplete");
247 mtx_->fillComplete(params);
248}
249
250template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
251void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceDomainMapAndImporter(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &newDomainMap, Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &newImporter) {
252 XPETRA_MONITOR("TpetraCrsMatrix::replaceDomainMapAndImporter");
253 XPETRA_DYNAMIC_CAST(const TpetraImportClass, *newImporter, tImporter, "Xpetra::TpetraCrsMatrix::replaceDomainMapAndImporter only accepts Xpetra::TpetraImport.");
254 RCP<const Tpetra::Import<LocalOrdinal, GlobalOrdinal, Node>> myImport = tImporter.getTpetra_Import();
255 mtx_->replaceDomainMapAndImporter(toTpetra(newDomainMap), myImport);
256}
257
258template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
259void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::expertStaticFillComplete(const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
260 const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
261 const RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &importer,
262 const RCP<const Export<LocalOrdinal, GlobalOrdinal, Node>> &exporter,
263 const RCP<ParameterList> &params) {
264 XPETRA_MONITOR("TpetraCrsMatrix::expertStaticFillComplete");
265 RCP<const Tpetra::Import<LocalOrdinal, GlobalOrdinal, Node>> myImport;
266 RCP<const Tpetra::Export<LocalOrdinal, GlobalOrdinal, Node>> myExport;
267
268 if (importer != Teuchos::null) {
269 XPETRA_DYNAMIC_CAST(const TpetraImportClass, *importer, tImporter, "Xpetra::TpetraCrsMatrix::expertStaticFillComplete only accepts Xpetra::TpetraImport.");
270 myImport = tImporter.getTpetra_Import();
271 }
272 if (exporter != Teuchos::null) {
273 XPETRA_DYNAMIC_CAST(const TpetraExportClass, *exporter, tExporter, "Xpetra::TpetraCrsMatrix::expertStaticFillComplete only accepts Xpetra::TpetraExport.");
274 myExport = tExporter.getTpetra_Export();
275 }
276
277 mtx_->expertStaticFillComplete(toTpetra(domainMap), toTpetra(rangeMap), myImport, myExport, params);
278}
279
280template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
285
286template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
291
292template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
297
298template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
299global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumRows() const {
300 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumRows");
301 return mtx_->getGlobalNumRows();
302}
303
304template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
305global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumCols() const {
306 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumCols");
307 return mtx_->getGlobalNumCols();
308}
309
310template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
311size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumRows() const {
312 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumRows");
313 return mtx_->getLocalNumRows();
314}
315
316template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
317size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumCols() const {
318 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumCols");
319 return mtx_->getLocalNumCols();
320}
321
322template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
323global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumEntries() const {
324 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumEntries");
325 return mtx_->getGlobalNumEntries();
326}
327
328template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
329size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumEntries() const {
330 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumEntries");
331 return mtx_->getLocalNumEntries();
332}
333
334template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
335size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getNumEntriesInLocalRow(LocalOrdinal localRow) const {
336 XPETRA_MONITOR("TpetraCrsMatrix::getNumEntriesInLocalRow");
337 return mtx_->getNumEntriesInLocalRow(localRow);
338}
339
340template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
341size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const {
342 XPETRA_MONITOR("TpetraCrsMatrix::getNumEntriesInGlobalRow");
343 return mtx_->getNumEntriesInGlobalRow(globalRow);
344}
345
346template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
347size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalMaxNumRowEntries() const {
348 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalMaxNumRowEntries");
349 return mtx_->getGlobalMaxNumRowEntries();
350}
351
352template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
353size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalMaxNumRowEntries() const {
354 XPETRA_MONITOR("TpetraCrsMatrix::getLocalMaxNumRowEntries");
355 return mtx_->getLocalMaxNumRowEntries();
356}
357
358template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
359bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isLocallyIndexed() const {
360 XPETRA_MONITOR("TpetraCrsMatrix::isLocallyIndexed");
361 return mtx_->isLocallyIndexed();
362}
363
364template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
365bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isGloballyIndexed() const {
366 XPETRA_MONITOR("TpetraCrsMatrix::isGloballyIndexed");
367 return mtx_->isGloballyIndexed();
368}
369
370template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
371bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isFillComplete() const {
372 XPETRA_MONITOR("TpetraCrsMatrix::isFillComplete");
373 return mtx_->isFillComplete();
374}
375
376template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
377bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isFillActive() const {
378 XPETRA_MONITOR("TpetraCrsMatrix::isFillActive");
379 return mtx_->isFillActive();
380}
381
382template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
383typename ScalarTraits<Scalar>::magnitudeType TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getFrobeniusNorm() const {
384 XPETRA_MONITOR("TpetraCrsMatrix::getFrobeniusNorm");
385 return mtx_->getFrobeniusNorm();
386}
387
388template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
389bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::supportsRowViews() const {
390 XPETRA_MONITOR("TpetraCrsMatrix::supportsRowViews");
391 return mtx_->supportsRowViews();
392}
393
394template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
395void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalRowCopy(LocalOrdinal LocalRow, const ArrayView<LocalOrdinal> &Indices, const ArrayView<Scalar> &Values, size_t &NumEntries) const {
396 XPETRA_MONITOR("TpetraCrsMatrix::getLocalRowCopy");
397 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::nonconst_local_inds_host_view_type indices("indices", Indices.size());
398 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type values("values", Values.size());
399
400 mtx_->getLocalRowCopy(LocalRow, indices, values, NumEntries);
401 for (size_t i = 0; i < NumEntries; ++i) {
402 Indices[i] = indices(i);
403 Values[i] = values(i);
404 }
405}
406
407template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
408void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalRowCopy(GlobalOrdinal GlobalRow, const ArrayView<GlobalOrdinal> &Indices, const ArrayView<Scalar> &Values, size_t &NumEntries) const {
409 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalRowCopy");
410 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::nonconst_global_inds_host_view_type indices("indices", Indices.size());
411 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type values("values", Values.size());
412
413 mtx_->getGlobalRowCopy(GlobalRow, indices, values, NumEntries);
414 for (size_t i = 0; i < NumEntries; ++i) {
415 Indices[i] = indices(i);
416 Values[i] = values(i);
417 }
418}
419
420template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
421void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView<const GlobalOrdinal> &Indices, ArrayView<const Scalar> &Values) const {
422 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalRowView");
423 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::global_inds_host_view_type indices;
424 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type values;
425
426 mtx_->getGlobalRowView(GlobalRow, indices, values);
427 Indices = ArrayView<const GlobalOrdinal>(indices.data(), indices.extent(0));
428 Values = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(values.data()), values.extent(0));
429}
430
431template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
432void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalRowView(LocalOrdinal LocalRow, ArrayView<const LocalOrdinal> &Indices, ArrayView<const Scalar> &Values) const {
433 XPETRA_MONITOR("TpetraCrsMatrix::getLocalRowView");
434 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::local_inds_host_view_type indices;
435 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type values;
436
437 mtx_->getLocalRowView(LocalRow, indices, values);
438 Indices = ArrayView<const LocalOrdinal>(indices.data(), indices.extent(0));
439 Values = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(values.data()), values.extent(0));
440}
441
442template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
443void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::apply(const MultiVector &X, MultiVector &Y, Teuchos::ETransp mode, Scalar alpha, Scalar beta) const {
444 XPETRA_MONITOR("TpetraCrsMatrix::apply");
445 mtx_->apply(toTpetra(X), toTpetra(Y), mode, alpha, beta);
446}
447
448template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
449void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::apply(const MultiVector &X, MultiVector &Y, Teuchos::ETransp mode, Scalar alpha, Scalar beta, bool sumInterfaceValues, const RCP<Import<LocalOrdinal, GlobalOrdinal, Node>> &regionInterfaceImporter, const Teuchos::ArrayRCP<LocalOrdinal> &regionInterfaceLIDs) const {
450 XPETRA_MONITOR("TpetraCrsMatrix::apply(region)");
451 RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> regionInterfaceMap = regionInterfaceImporter->getTargetMap();
452 mtx_->localApply(toTpetra(X), toTpetra(Y), mode, alpha, beta);
453 if (sumInterfaceValues) {
454 // preform communication to propagate local interface
455 // values to all the processor that share interfaces.
456 RCP<MultiVector> matvecInterfaceTmp = Xpetra::MultiVectorFactory<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Build(regionInterfaceMap, 1);
457 matvecInterfaceTmp->doImport(Y, *regionInterfaceImporter, INSERT);
458
459 // sum all contributions to interface values
460 // on all ranks
461 ArrayRCP<Scalar> YData = Y.getDataNonConst(0);
462 ArrayRCP<Scalar> interfaceData = matvecInterfaceTmp->getDataNonConst(0);
463 for (LocalOrdinal interfaceIdx = 0; interfaceIdx < static_cast<LocalOrdinal>(interfaceData.size()); ++interfaceIdx) {
464 YData[regionInterfaceLIDs[interfaceIdx]] += interfaceData[interfaceIdx];
465 }
466 }
467}
468
469template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
470const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getDomainMap() const {
471 XPETRA_MONITOR("TpetraCrsMatrix::getDomainMap");
472 return toXpetra(mtx_->getDomainMap());
473}
474
475template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
480
481template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
482std::string TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::description() const {
483 XPETRA_MONITOR("TpetraCrsMatrix::description");
484 return mtx_->description();
485}
486
487template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
488void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel) const {
489 XPETRA_MONITOR("TpetraCrsMatrix::describe");
490 mtx_->describe(out, verbLevel);
491}
492
493template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
494void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setObjectLabel(const std::string &objectLabel) {
495 XPETRA_MONITOR("TpetraCrsMatrix::setObjectLabel");
496 Teuchos::LabeledObject::setObjectLabel(objectLabel);
497 mtx_->setObjectLabel(objectLabel);
498}
499
500template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
501TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const TpetraCrsMatrix &matrix)
502 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(*(matrix.mtx_), Teuchos::Copy))) {}
503
504template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
505void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag) const {
506 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
507 XPETRA_DYNAMIC_CAST(TpetraVectorClass, diag, tDiag, "Xpetra::TpetraCrsMatrix.getLocalDiagCopy() only accept Xpetra::TpetraVector as input arguments.");
508 mtx_->getLocalDiagCopy(*tDiag.getTpetra_Vector());
509}
510
511template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
512void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagOffsets(Teuchos::ArrayRCP<size_t> &offsets) const {
513 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagOffsets");
514 mtx_->getLocalDiagOffsets(offsets);
515}
516
517template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
518void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag, const Teuchos::ArrayView<const size_t> &offsets) const {
519 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
520 mtx_->getLocalDiagCopy(*(toTpetra(diag)), offsets);
521}
522
523template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
524void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag, const Kokkos::View<const size_t *, typename Node::device_type, Kokkos::MemoryUnmanaged> &offsets) const {
525 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
526 mtx_->getLocalDiagCopy(*(toTpetra(diag)), offsets);
527}
528
529template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
530void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceDiag(const Vector &diag) {
531 XPETRA_MONITOR("TpetraCrsMatrix::replaceDiag");
532 Tpetra::replaceDiagonalCrsMatrix(*mtx_, *(toTpetra(diag)));
533}
534
535template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
536void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::leftScale(const Vector &x) {
537 XPETRA_MONITOR("TpetraCrsMatrix::leftScale");
538 mtx_->leftScale(*(toTpetra(x)));
539}
540
541template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
542void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::rightScale(const Vector &x) {
543 XPETRA_MONITOR("TpetraCrsMatrix::rightScale");
544 mtx_->rightScale(*(toTpetra(x)));
545}
546
547template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
552
553template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
554void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doImport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &source,
555 const Import<LocalOrdinal, GlobalOrdinal, Node> &importer, CombineMode CM) {
556 XPETRA_MONITOR("TpetraCrsMatrix::doImport");
557
558 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, source, tSource, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
559 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSource.getTpetra_CrsMatrix();
560 // mtx_->doImport(toTpetraCrsMatrix(source), *tImporter.getTpetra_Import(), toTpetra(CM));
561 mtx_->doImport(*v, toTpetra(importer), toTpetra(CM));
562}
563
565template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
566void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doExport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &dest,
567 const Import<LocalOrdinal, GlobalOrdinal, Node> &importer, CombineMode CM) {
568 XPETRA_MONITOR("TpetraCrsMatrix::doExport");
569
570 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, dest, tDest, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
571 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tDest.getTpetra_CrsMatrix();
572 mtx_->doExport(*v, toTpetra(importer), toTpetra(CM));
573}
574
575template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
576void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doImport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &source,
577 const Export<LocalOrdinal, GlobalOrdinal, Node> &exporter, CombineMode CM) {
578 XPETRA_MONITOR("TpetraCrsMatrix::doImport");
579
580 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, source, tSource, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
581 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSource.getTpetra_CrsMatrix();
582 mtx_->doImport(*v, toTpetra(exporter), toTpetra(CM));
583}
584
585template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
586void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doExport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &dest,
587 const Export<LocalOrdinal, GlobalOrdinal, Node> &exporter, CombineMode CM) {
588 XPETRA_MONITOR("TpetraCrsMatrix::doExport");
589
590 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, dest, tDest, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
591 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tDest.getTpetra_CrsMatrix();
592 mtx_->doExport(*v, toTpetra(exporter), toTpetra(CM));
593}
594
595template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
596void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::removeEmptyProcessesInPlace(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &newMap) {
597 XPETRA_MONITOR("TpetraCrsMatrix::removeEmptyProcessesInPlace");
598 mtx_->removeEmptyProcessesInPlace(toTpetra(newMap));
599}
600
601template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
602bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::hasMatrix() const {
603 return !mtx_.is_null();
604}
605
606template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
607TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> &mtx)
608 : mtx_(mtx) {}
609
610template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
611RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getTpetra_CrsMatrix() const { return mtx_; }
612
614template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
615RCP<Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getTpetra_CrsMatrixNonConst() const { return mtx_; } // TODO: remove
616
618template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
619void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::residual(const MultiVector &X,
620 const MultiVector &B,
621 MultiVector &R) const {
622 Tpetra::Details::residual(*mtx_, toTpetra(X), toTpetra(B), toTpetra(R));
623}
624
627// End of TpetrCrsMatrix class definition //
630
631} // namespace Xpetra
632
633#endif // XPETRA_TPETRACRSMATRIX_DEF_HPP
XPETRA_MONITOR
#define XPETRA_MONITOR(funcName)
Definition Xpetra_ConfigDefs.hpp:96
XPETRA_DYNAMIC_CAST
#define XPETRA_DYNAMIC_CAST(type, obj, newObj, exceptionMsg)
Definition Xpetra_Exceptions.hpp:22
Xpetra_TpetraCrsMatrix_decl.hpp
Teuchos::ArrayRCP::size
size_type size() const
Teuchos::ArrayRCP::resize
void resize(const size_type n, const T &val=T())
Teuchos::ArrayView::size
size_type size() const
Teuchos::LabeledObject::setObjectLabel
virtual void setObjectLabel(const std::string &objectLabel)
Teuchos::RCP::is_null
bool is_null() const
Teuchos::RCP::get
T * get() const
Teuchos::basic_FancyOStream
Xpetra::MultiVectorFactory::Build
static Teuchos::RCP< MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Build(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &map, size_t NumVectors, bool zeroOut=true)
Constructor specifying the number of non-zeros for all rows.
Definition Xpetra_MultiVectorFactory_def.hpp:24
Xpetra::MultiVector::getDataNonConst
virtual Teuchos::ArrayRCP< Scalar > getDataNonConst(size_t j)=0
View of the local values in a particular vector of this multivector.
Xpetra::TpetraCrsMatrix::insertLocalValues
void insertLocalValues(LocalOrdinal localRow, const ArrayView< const LocalOrdinal > &cols, const ArrayView< const Scalar > &vals)
Insert matrix entries, using local IDs.
Definition Xpetra_TpetraCrsMatrix_def.hpp:156
Xpetra::TpetraCrsMatrix::supportsRowViews
bool supportsRowViews() const
Returns true if getLocalRowView() and getGlobalRowView() are valid for this class.
Definition Xpetra_TpetraCrsMatrix_def.hpp:389
Xpetra::TpetraCrsMatrix::getTpetra_CrsMatrixNonConst
RCP< Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsMatrixNonConst() const
Get the underlying Tpetra matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:615
Xpetra::TpetraCrsMatrix::replaceLocalValues
void replaceLocalValues(LocalOrdinal localRow, const ArrayView< const LocalOrdinal > &cols, const ArrayView< const Scalar > &vals)
Replace matrix entries, using local IDs.
Definition Xpetra_TpetraCrsMatrix_def.hpp:168
Xpetra::TpetraCrsMatrix::TpetraCrsMatrix
TpetraCrsMatrix(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &rowMap, size_t maxNumEntriesPerRow, const Teuchos::RCP< Teuchos::ParameterList > &params=Teuchos::null)
Constructor specifying fixed number of entries for each row.
Definition Xpetra_TpetraCrsMatrix_def.hpp:20
Xpetra::TpetraCrsMatrix::getFrobeniusNorm
ScalarTraits< Scalar >::magnitudeType getFrobeniusNorm() const
Returns the Frobenius norm of the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:383
Xpetra::TpetraCrsMatrix::getGlobalRowCopy
void getGlobalRowCopy(GlobalOrdinal GlobalRow, const ArrayView< GlobalOrdinal > &indices, const ArrayView< Scalar > &values, size_t &numEntries) const
Extract a list of entries in a specified global row of this matrix. Put into pre-allocated storage.
Definition Xpetra_TpetraCrsMatrix_def.hpp:408
Xpetra::TpetraCrsMatrix::local_matrix_type
Xpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node >::local_matrix_type local_matrix_type
Definition Xpetra_TpetraCrsMatrix_decl.hpp:54
Xpetra::TpetraCrsMatrix::replaceGlobalValues
void replaceGlobalValues(GlobalOrdinal globalRow, const ArrayView< const GlobalOrdinal > &cols, const ArrayView< const Scalar > &vals)
Replace matrix entries, using global IDs.
Definition Xpetra_TpetraCrsMatrix_def.hpp:162
Xpetra::TpetraCrsMatrix::getLocalDiagCopy
void getLocalDiagCopy(Vector &diag) const
Get a copy of the diagonal entries owned by this node, with local row indices.
Definition Xpetra_TpetraCrsMatrix_def.hpp:505
Xpetra::TpetraCrsMatrix::getNumEntriesInGlobalRow
size_t getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const
Returns the current number of entries in the (locally owned) global row.
Definition Xpetra_TpetraCrsMatrix_def.hpp:341
Xpetra::TpetraCrsMatrix::scale
void scale(const Scalar &alpha)
Scale the current values of a matrix, this = alpha*this.
Definition Xpetra_TpetraCrsMatrix_def.hpp:187
Xpetra::TpetraCrsMatrix::rightScale
void rightScale(const Vector &x)
Right scale operator with given vector values.
Definition Xpetra_TpetraCrsMatrix_def.hpp:542
Xpetra::TpetraCrsMatrix::getLocalRowCopy
void getLocalRowCopy(LocalOrdinal LocalRow, const ArrayView< LocalOrdinal > &Indices, const ArrayView< Scalar > &Values, size_t &NumEntries) const
Extract a list of entries in a specified local row of the matrix. Put into storage allocated by calli...
Definition Xpetra_TpetraCrsMatrix_def.hpp:395
Xpetra::TpetraCrsMatrix::getGlobalNumCols
global_size_t getGlobalNumCols() const
Number of global columns in the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:305
Xpetra::TpetraCrsMatrix::fillComplete
void fillComplete(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &domainMap, const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &rangeMap, const RCP< ParameterList > &params=null)
Signal that data entry is complete, specifying domain and range maps.
Definition Xpetra_TpetraCrsMatrix_def.hpp:239
Xpetra::TpetraCrsMatrix::getColMap
const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getColMap() const
Returns the Map that describes the column distribution in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:287
Xpetra::TpetraCrsMatrix::allocateAllValues
void allocateAllValues(size_t numNonZeros, ArrayRCP< size_t > &rowptr, ArrayRCP< LocalOrdinal > &colind, ArrayRCP< Scalar > &values)
Allocates and returns ArrayRCPs of the Crs arrays — This is an Xpetra-only routine.
Definition Xpetra_TpetraCrsMatrix_def.hpp:193
Xpetra::TpetraCrsMatrix::residual
void residual(const MultiVector &X, const MultiVector &B, MultiVector &R) const
Compute a residual R = B - (*this) * X.
Definition Xpetra_TpetraCrsMatrix_def.hpp:619
Xpetra::TpetraCrsMatrix::doExport
void doExport(const DistObject< char, LocalOrdinal, GlobalOrdinal, Node > &dest, const Import< LocalOrdinal, GlobalOrdinal, Node > &importer, CombineMode CM)
Export.
Definition Xpetra_TpetraCrsMatrix_def.hpp:566
Xpetra::TpetraCrsMatrix::isGloballyIndexed
bool isGloballyIndexed() const
If matrix indices are in the global range, this function returns true. Otherwise, this function retur...
Definition Xpetra_TpetraCrsMatrix_def.hpp:365
Xpetra::TpetraCrsMatrix::resumeFill
void resumeFill(const RCP< ParameterList > &params=null)
Definition Xpetra_TpetraCrsMatrix_def.hpp:233
Xpetra::TpetraCrsMatrix::getLocalDiagOffsets
void getLocalDiagOffsets(Teuchos::ArrayRCP< size_t > &offsets) const
Get offsets of the diagonal entries in the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:512
Xpetra::TpetraCrsMatrix::insertGlobalValues
void insertGlobalValues(GlobalOrdinal globalRow, const ArrayView< const GlobalOrdinal > &cols, const ArrayView< const Scalar > &vals)
Insert matrix entries, using global IDs.
Definition Xpetra_TpetraCrsMatrix_def.hpp:150
Xpetra::TpetraCrsMatrix::leftScale
void leftScale(const Vector &x)
Left scale operator with given vector values.
Definition Xpetra_TpetraCrsMatrix_def.hpp:536
Xpetra::TpetraCrsMatrix::getLocalRowView
void getLocalRowView(LocalOrdinal LocalRow, ArrayView< const LocalOrdinal > &indices, ArrayView< const Scalar > &values) const
Extract a const, non-persisting view of local indices in a specified row of the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:432
Xpetra::TpetraCrsMatrix::haveGlobalConstants
bool haveGlobalConstants() const
Returns true if globalConstants have been computed; false otherwise.
Definition Xpetra_TpetraCrsMatrix_def.hpp:230
Xpetra::TpetraCrsMatrix::hasMatrix
bool hasMatrix() const
Does this have an underlying matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:602
Xpetra::TpetraCrsMatrix::getLocalMaxNumRowEntries
size_t getLocalMaxNumRowEntries() const
Returns the maximum number of entries across all rows/columns on this node.
Definition Xpetra_TpetraCrsMatrix_def.hpp:353
Xpetra::TpetraCrsMatrix::description
std::string description() const
A simple one-line description of this object.
Definition Xpetra_TpetraCrsMatrix_def.hpp:482
Xpetra::TpetraCrsMatrix::getTpetra_CrsMatrix
RCP< const Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsMatrix() const
Get the underlying Tpetra matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:611
Xpetra::TpetraCrsMatrix::getLocalNumRows
size_t getLocalNumRows() const
Returns the number of matrix rows owned on the calling node.
Definition Xpetra_TpetraCrsMatrix_def.hpp:311
Xpetra::TpetraCrsMatrix::setAllToScalar
void setAllToScalar(const Scalar &alpha)
Set all matrix entries equal to scalarThis.
Definition Xpetra_TpetraCrsMatrix_def.hpp:181
Xpetra::TpetraCrsMatrix::isLocallyIndexed
bool isLocallyIndexed() const
If matrix indices are in the local range, this function returns true. Otherwise, this function return...
Definition Xpetra_TpetraCrsMatrix_def.hpp:359
Xpetra::TpetraCrsMatrix::isFillActive
bool isFillActive() const
Returns true if the matrix is in edit mode.
Definition Xpetra_TpetraCrsMatrix_def.hpp:377
Xpetra::TpetraCrsMatrix::replaceDomainMapAndImporter
void replaceDomainMapAndImporter(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &newDomainMap, Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > &newImporter)
Replaces the current domainMap and importer with the user-specified objects.
Definition Xpetra_TpetraCrsMatrix_def.hpp:251
Xpetra::TpetraCrsMatrix::setAllValues
void setAllValues(const ArrayRCP< size_t > &rowptr, const ArrayRCP< LocalOrdinal > &colind, const ArrayRCP< Scalar > &values)
Sets the 1D pointer arrays of the graph.
Definition Xpetra_TpetraCrsMatrix_def.hpp:201
Xpetra::TpetraCrsMatrix::removeEmptyProcessesInPlace
void removeEmptyProcessesInPlace(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &newMap)
Definition Xpetra_TpetraCrsMatrix_def.hpp:596
Xpetra::TpetraCrsMatrix::getLocalNumCols
size_t getLocalNumCols() const
Returns the number of columns connected to the locally owned rows of this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:317
Xpetra::TpetraCrsMatrix::getGlobalNumEntries
global_size_t getGlobalNumEntries() const
Returns the global number of entries in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:323
Xpetra::TpetraCrsMatrix::getAllValues
void getAllValues(ArrayRCP< const size_t > &rowptr, ArrayRCP< const LocalOrdinal > &colind, ArrayRCP< const Scalar > &values) const
Gets the 1D pointer arrays of the graph.
Definition Xpetra_TpetraCrsMatrix_def.hpp:207
Xpetra::TpetraCrsMatrix::doImport
void doImport(const DistObject< char, LocalOrdinal, GlobalOrdinal, Node > &source, const Import< LocalOrdinal, GlobalOrdinal, Node > &importer, CombineMode CM)
Import.
Definition Xpetra_TpetraCrsMatrix_def.hpp:554
Xpetra::TpetraCrsMatrix::getGlobalNumRows
global_size_t getGlobalNumRows() const
Number of global elements in the row map of this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:299
Xpetra::TpetraCrsMatrix::describe
void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Print the object with some verbosity level to an FancyOStream object.
Definition Xpetra_TpetraCrsMatrix_def.hpp:488
Xpetra::TpetraCrsMatrix::getNumEntriesInLocalRow
size_t getNumEntriesInLocalRow(LocalOrdinal localRow) const
Returns the current number of entries on this node in the specified local row.
Definition Xpetra_TpetraCrsMatrix_def.hpp:335
Xpetra::TpetraCrsMatrix::getGlobalRowView
void getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView< const GlobalOrdinal > &indices, ArrayView< const Scalar > &values) const
Extract a const, non-persisting view of global indices in a specified row of the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:421
Xpetra::TpetraCrsMatrix::getDomainMap
const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getDomainMap() const
Returns the Map associated with the domain of this operator. This will be null until fillComplete() i...
Definition Xpetra_TpetraCrsMatrix_def.hpp:470
Xpetra::TpetraCrsMatrix::expertStaticFillComplete
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=Teuchos::null, const RCP< const Export< LocalOrdinal, GlobalOrdinal, Node > > &exporter=Teuchos::null, const RCP< ParameterList > &params=Teuchos::null)
Expert static fill complete.
Definition Xpetra_TpetraCrsMatrix_def.hpp:259
Xpetra::TpetraCrsMatrix::replaceDiag
void replaceDiag(const Vector &diag)
Replace the diagonal entries of the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:530
Xpetra::TpetraCrsMatrix::getRowMap
const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getRowMap() const
Returns the Map that describes the row distribution in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:281
Xpetra::TpetraCrsMatrix::getGlobalMaxNumRowEntries
size_t getGlobalMaxNumRowEntries() const
Returns the maximum number of entries across all rows/columns on all nodes.
Definition Xpetra_TpetraCrsMatrix_def.hpp:347
Xpetra::TpetraCrsMatrix::setObjectLabel
void setObjectLabel(const std::string &objectLabel)
Definition Xpetra_TpetraCrsMatrix_def.hpp:494
Xpetra::TpetraCrsMatrix::apply
void apply(const MultiVector &X, MultiVector &Y, Teuchos::ETransp mode=Teuchos::NO_TRANS, Scalar alpha=ScalarTraits< Scalar >::one(), Scalar beta=ScalarTraits< Scalar >::zero()) const
Computes the sparse matrix-multivector multiplication.
Definition Xpetra_TpetraCrsMatrix_def.hpp:443
Xpetra::TpetraCrsMatrix::getCrsGraph
RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > getCrsGraph() const
Returns the CrsGraph associated with this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:293
Xpetra::TpetraCrsMatrix::getRangeMap
const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getRangeMap() const
Returns the Map associated with the range of this operator, which must be compatible with Y....
Definition Xpetra_TpetraCrsMatrix_def.hpp:476
Xpetra::TpetraCrsMatrix::isFillComplete
bool isFillComplete() const
Returns true if the matrix is in compute mode, i.e. if fillComplete() has been called.
Definition Xpetra_TpetraCrsMatrix_def.hpp:371
Xpetra::TpetraCrsMatrix::getLocalNumEntries
size_t getLocalNumEntries() const
Returns the local number of entries in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:329
Xpetra::TpetraCrsMatrix::getMap
Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getMap() const
Implements DistObject interface.
Definition Xpetra_TpetraCrsMatrix_def.hpp:548
TEUCHOS_TEST_FOR_EXCEPTION
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
Teuchos::EVerbosityLevel
EVerbosityLevel
Xpetra::global_size_t
size_t global_size_t
Global size_t object.
Definition Xpetra_ConfigDefs.hpp:138
Xpetra::toXpetra
RCP< const CrsGraph< int, GlobalOrdinal, Node > > toXpetra(const Epetra_CrsGraph &g)
Definition Xpetra_EpetraCrsGraph.cpp:29
Xpetra::toTpetra
RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toTpetra(const RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > &graph)
Definition Xpetra_TpetraCrsGraph_decl.hpp:376
Xpetra::CombineMode
CombineMode
Xpetra::Combine Mode enumerable type.
Definition Xpetra_ConfigDefs.hpp:176