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>
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
115template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
116TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
117 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
118 const local_matrix_type &lclMatrix,
119 const Teuchos::RCP<Teuchos::ParameterList> &params)
120 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(toTpetra(rowMap), toTpetra(colMap), lclMatrix, params))) {}
121
122template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
123TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(
124 const local_matrix_type &lclMatrix,
125 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
126 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
127 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
128 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
129 const Teuchos::RCP<Teuchos::ParameterList> &params)
130 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(lclMatrix, toTpetra(rowMap), toTpetra(colMap), toTpetra(domainMap), toTpetra(rangeMap), params))) {}
131
132template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
133TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(
134 const local_matrix_type &lclMatrix,
135 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rowMap,
136 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &colMap,
137 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
138 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
139 const Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &importer,
140 const Teuchos::RCP<const Export<LocalOrdinal, GlobalOrdinal, Node>> &exporter,
141 const Teuchos::RCP<Teuchos::ParameterList> &params)
142 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(lclMatrix, toTpetra(rowMap), toTpetra(colMap), toTpetra(domainMap), toTpetra(rangeMap), toTpetra(importer), toTpetra(exporter), params))) {}
143
144template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
145TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::~TpetraCrsMatrix() {}
146
147template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
148void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::insertGlobalValues(GlobalOrdinal globalRow, const ArrayView<const GlobalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
149 XPETRA_MONITOR("TpetraCrsMatrix::insertGlobalValues");
150 mtx_->insertGlobalValues(globalRow, cols, vals);
151}
152
153template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
154void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::insertLocalValues(LocalOrdinal localRow, const ArrayView<const LocalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
155 XPETRA_MONITOR("TpetraCrsMatrix::insertLocalValues");
156 mtx_->insertLocalValues(localRow, cols, vals);
157}
158
159template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
160void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceGlobalValues(GlobalOrdinal globalRow, const ArrayView<const GlobalOrdinal> &cols, const ArrayView<const Scalar> &vals) {
161 XPETRA_MONITOR("TpetraCrsMatrix::replaceGlobalValues");
162 mtx_->replaceGlobalValues(globalRow, cols, vals);
163}
164
165template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
166void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceLocalValues(LocalOrdinal localRow,
167 const ArrayView<const LocalOrdinal> &cols,
168 const ArrayView<const Scalar> &vals) {
169 XPETRA_MONITOR("TpetraCrsMatrix::replaceLocalValues");
170 typedef typename ArrayView<const LocalOrdinal>::size_type size_type;
171 const LocalOrdinal numValid =
172 mtx_->replaceLocalValues(localRow, cols, vals);
173 TEUCHOS_TEST_FOR_EXCEPTION(
174 static_cast<size_type>(numValid) != cols.size(), std::runtime_error,
175 "replaceLocalValues returned " << numValid << " != cols.size() = " << cols.size() << ".");
176}
177
178template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
179void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setAllToScalar(const Scalar &alpha) {
180 XPETRA_MONITOR("TpetraCrsMatrix::setAllToScalar");
181 mtx_->setAllToScalar(alpha);
182}
183
184template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
185void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::scale(const Scalar &alpha) {
186 XPETRA_MONITOR("TpetraCrsMatrix::scale");
187 mtx_->scale(alpha);
188}
189
190template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
191void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::allocateAllValues(size_t numNonZeros, ArrayRCP<size_t> &rowptr, ArrayRCP<LocalOrdinal> &colind, ArrayRCP<Scalar> &values) {
192 XPETRA_MONITOR("TpetraCrsMatrix::allocateAllValues");
193 rowptr.resize(getLocalNumRows() + 1);
194 colind.resize(numNonZeros);
195 values.resize(numNonZeros);
196}
197
198template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
199void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setAllValues(const ArrayRCP<size_t> &rowptr, const ArrayRCP<LocalOrdinal> &colind, const ArrayRCP<Scalar> &values) {
200 XPETRA_MONITOR("TpetraCrsMatrix::setAllValues");
201 mtx_->setAllValues(rowptr, colind, values);
202}
203
204template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
205void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getAllValues(ArrayRCP<const size_t> &rowptr, ArrayRCP<const LocalOrdinal> &colind, ArrayRCP<const Scalar> &values) const {
206 XPETRA_MONITOR("TpetraCrsMatrix::getAllValues");
207 // TODO: Change getAllValues interface to return Kokkos views,
208 // TODO: or add getLocalRowPtrsHost, getLocalIndicesHost, etc., interfaces
209 // TODO: and use them in MueLu
210 rowptr = Kokkos::Compat::persistingView(mtx_->getLocalRowPtrsHost());
211 colind = Kokkos::Compat::persistingView(mtx_->getLocalIndicesHost());
212 values = Teuchos::arcp_reinterpret_cast<const Scalar>(
213 Kokkos::Compat::persistingView(mtx_->getLocalValuesHost(
214 Tpetra::Access::ReadOnly)));
215}
216
217template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
218void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getAllValues(ArrayRCP<Scalar> &values) {
219 XPETRA_MONITOR("TpetraCrsMatrix::getAllValues");
220 // TODO: Change getAllValues interface to return Kokkos view,
221 // TODO: or add getLocalValuesDevice() interfaces
222 values = Teuchos::arcp_reinterpret_cast<Scalar>(
223 Kokkos::Compat::persistingView(mtx_->getLocalValuesDevice(
224 Tpetra::Access::ReadWrite)));
225}
226
227template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
228bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::haveGlobalConstants() const { return mtx_->haveGlobalConstants(); }
229
230template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
231void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::resumeFill(const RCP<ParameterList> &params) {
232 XPETRA_MONITOR("TpetraCrsMatrix::resumeFill");
233 mtx_->resumeFill(params);
234}
235
236template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
237void 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) {
238 XPETRA_MONITOR("TpetraCrsMatrix::fillComplete");
239 mtx_->fillComplete(toTpetra(domainMap), toTpetra(rangeMap), params);
240}
241
242template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
243void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::fillComplete(const RCP<ParameterList> &params) {
244 XPETRA_MONITOR("TpetraCrsMatrix::fillComplete");
245 mtx_->fillComplete(params);
246}
247
248template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
249void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::replaceDomainMapAndImporter(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &newDomainMap, Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &newImporter) {
250 XPETRA_MONITOR("TpetraCrsMatrix::replaceDomainMapAndImporter");
251 XPETRA_DYNAMIC_CAST(const TpetraImportClass, *newImporter, tImporter, "Xpetra::TpetraCrsMatrix::replaceDomainMapAndImporter only accepts Xpetra::TpetraImport.");
252 RCP<const Tpetra::Import<LocalOrdinal, GlobalOrdinal, Node>> myImport = tImporter.getTpetra_Import();
253 mtx_->replaceDomainMapAndImporter(toTpetra(newDomainMap), myImport);
254}
255
256template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
257void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::expertStaticFillComplete(const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &domainMap,
258 const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &rangeMap,
259 const RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>> &importer,
260 const RCP<const Export<LocalOrdinal, GlobalOrdinal, Node>> &exporter,
261 const RCP<ParameterList> &params) {
262 XPETRA_MONITOR("TpetraCrsMatrix::expertStaticFillComplete");
263 RCP<const Tpetra::Import<LocalOrdinal, GlobalOrdinal, Node>> myImport;
264 RCP<const Tpetra::Export<LocalOrdinal, GlobalOrdinal, Node>> myExport;
265
266 if (importer != Teuchos::null) {
267 XPETRA_DYNAMIC_CAST(const TpetraImportClass, *importer, tImporter, "Xpetra::TpetraCrsMatrix::expertStaticFillComplete only accepts Xpetra::TpetraImport.");
268 myImport = tImporter.getTpetra_Import();
269 }
270 if (exporter != Teuchos::null) {
271 XPETRA_DYNAMIC_CAST(const TpetraExportClass, *exporter, tExporter, "Xpetra::TpetraCrsMatrix::expertStaticFillComplete only accepts Xpetra::TpetraExport.");
272 myExport = tExporter.getTpetra_Export();
273 }
274
275 mtx_->expertStaticFillComplete(toTpetra(domainMap), toTpetra(rangeMap), myImport, myExport, params);
276}
277
278template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
283
284template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
289
290template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
295
296template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
297global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumRows() const {
298 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumRows");
299 return mtx_->getGlobalNumRows();
300}
301
302template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
303global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumCols() const {
304 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumCols");
305 return mtx_->getGlobalNumCols();
306}
307
308template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
309size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumRows() const {
310 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumRows");
311 return mtx_->getLocalNumRows();
312}
313
314template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
315size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumCols() const {
316 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumCols");
317 return mtx_->getLocalNumCols();
318}
319
320template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
321global_size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalNumEntries() const {
322 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalNumEntries");
323 return mtx_->getGlobalNumEntries();
324}
325
326template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
327size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalNumEntries() const {
328 XPETRA_MONITOR("TpetraCrsMatrix::getLocalNumEntries");
329 return mtx_->getLocalNumEntries();
330}
331
332template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
333size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getNumEntriesInLocalRow(LocalOrdinal localRow) const {
334 XPETRA_MONITOR("TpetraCrsMatrix::getNumEntriesInLocalRow");
335 return mtx_->getNumEntriesInLocalRow(localRow);
336}
337
338template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
339size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getNumEntriesInGlobalRow(GlobalOrdinal globalRow) const {
340 XPETRA_MONITOR("TpetraCrsMatrix::getNumEntriesInGlobalRow");
341 return mtx_->getNumEntriesInGlobalRow(globalRow);
342}
343
344template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
345size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalMaxNumRowEntries() const {
346 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalMaxNumRowEntries");
347 return mtx_->getGlobalMaxNumRowEntries();
348}
349
350template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
351size_t TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalMaxNumRowEntries() const {
352 XPETRA_MONITOR("TpetraCrsMatrix::getLocalMaxNumRowEntries");
353 return mtx_->getLocalMaxNumRowEntries();
354}
355
356template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
357bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isLocallyIndexed() const {
358 XPETRA_MONITOR("TpetraCrsMatrix::isLocallyIndexed");
359 return mtx_->isLocallyIndexed();
360}
361
362template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
363bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isGloballyIndexed() const {
364 XPETRA_MONITOR("TpetraCrsMatrix::isGloballyIndexed");
365 return mtx_->isGloballyIndexed();
366}
367
368template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
369bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isFillComplete() const {
370 XPETRA_MONITOR("TpetraCrsMatrix::isFillComplete");
371 return mtx_->isFillComplete();
372}
373
374template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
375bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::isFillActive() const {
376 XPETRA_MONITOR("TpetraCrsMatrix::isFillActive");
377 return mtx_->isFillActive();
378}
379
380template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
381typename ScalarTraits<Scalar>::magnitudeType TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getFrobeniusNorm() const {
382 XPETRA_MONITOR("TpetraCrsMatrix::getFrobeniusNorm");
383 return mtx_->getFrobeniusNorm();
384}
385
386template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
387bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::supportsRowViews() const {
388 XPETRA_MONITOR("TpetraCrsMatrix::supportsRowViews");
389 return mtx_->supportsRowViews();
390}
391
392template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
393void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalRowCopy(LocalOrdinal LocalRow, const ArrayView<LocalOrdinal> &Indices, const ArrayView<Scalar> &Values, size_t &NumEntries) const {
394 XPETRA_MONITOR("TpetraCrsMatrix::getLocalRowCopy");
395 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::nonconst_local_inds_host_view_type indices("indices", Indices.size());
396 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type values("values", Values.size());
397
398 mtx_->getLocalRowCopy(LocalRow, indices, values, NumEntries);
399 for (size_t i = 0; i < NumEntries; ++i) {
400 Indices[i] = indices(i);
401 Values[i] = values(i);
402 }
403}
404
405template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
406void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalRowCopy(GlobalOrdinal GlobalRow, const ArrayView<GlobalOrdinal> &Indices, const ArrayView<Scalar> &Values, size_t &NumEntries) const {
407 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalRowCopy");
408 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::nonconst_global_inds_host_view_type indices("indices", Indices.size());
409 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type values("values", Values.size());
410
411 mtx_->getGlobalRowCopy(GlobalRow, indices, values, NumEntries);
412 for (size_t i = 0; i < NumEntries; ++i) {
413 Indices[i] = indices(i);
414 Values[i] = values(i);
415 }
416}
417
418template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
419void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getGlobalRowView(GlobalOrdinal GlobalRow, ArrayView<const GlobalOrdinal> &Indices, ArrayView<const Scalar> &Values) const {
420 XPETRA_MONITOR("TpetraCrsMatrix::getGlobalRowView");
421 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::global_inds_host_view_type indices;
422 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type values;
423
424 mtx_->getGlobalRowView(GlobalRow, indices, values);
425 Indices = ArrayView<const GlobalOrdinal>(indices.data(), indices.extent(0));
426 Values = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(values.data()), values.extent(0));
427}
428
429template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
430void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalRowView(LocalOrdinal LocalRow, ArrayView<const LocalOrdinal> &Indices, ArrayView<const Scalar> &Values) const {
431 XPETRA_MONITOR("TpetraCrsMatrix::getLocalRowView");
432 typename Tpetra::CrsGraph<LocalOrdinal, GlobalOrdinal, Node>::local_inds_host_view_type indices;
433 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type values;
434
435 mtx_->getLocalRowView(LocalRow, indices, values);
436 Indices = ArrayView<const LocalOrdinal>(indices.data(), indices.extent(0));
437 Values = ArrayView<const Scalar>(reinterpret_cast<const Scalar *>(values.data()), values.extent(0));
438}
439
440template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
441void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::apply(const MultiVector &X, MultiVector &Y, Teuchos::ETransp mode, Scalar alpha, Scalar beta) const {
442 XPETRA_MONITOR("TpetraCrsMatrix::apply");
443 mtx_->apply(toTpetra(X), toTpetra(Y), mode, alpha, beta);
444}
445
446template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
447void 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 {
448 XPETRA_MONITOR("TpetraCrsMatrix::apply(region)");
449 RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> regionInterfaceMap = regionInterfaceImporter->getTargetMap();
450 mtx_->localApply(toTpetra(X), toTpetra(Y), mode, alpha, beta);
451 if (sumInterfaceValues) {
452 // preform communication to propagate local interface
453 // values to all the processor that share interfaces.
454 RCP<MultiVector> matvecInterfaceTmp = Xpetra::MultiVectorFactory<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Build(regionInterfaceMap, 1);
455 matvecInterfaceTmp->doImport(Y, *regionInterfaceImporter, INSERT);
456
457 // sum all contributions to interface values
458 // on all ranks
459 ArrayRCP<Scalar> YData = Y.getDataNonConst(0);
460 ArrayRCP<Scalar> interfaceData = matvecInterfaceTmp->getDataNonConst(0);
461 for (LocalOrdinal interfaceIdx = 0; interfaceIdx < static_cast<LocalOrdinal>(interfaceData.size()); ++interfaceIdx) {
462 YData[regionInterfaceLIDs[interfaceIdx]] += interfaceData[interfaceIdx];
463 }
464 }
465}
466
467template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
468const RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getDomainMap() const {
469 XPETRA_MONITOR("TpetraCrsMatrix::getDomainMap");
470 return toXpetra(mtx_->getDomainMap());
471}
472
473template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
478
479template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
480std::string TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::description() const {
481 XPETRA_MONITOR("TpetraCrsMatrix::description");
482 return mtx_->description();
483}
484
485template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
486void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel) const {
487 XPETRA_MONITOR("TpetraCrsMatrix::describe");
488 mtx_->describe(out, verbLevel);
489}
490
491template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
492void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::setObjectLabel(const std::string &objectLabel) {
493 XPETRA_MONITOR("TpetraCrsMatrix::setObjectLabel");
494 Teuchos::LabeledObject::setObjectLabel(objectLabel);
495 mtx_->setObjectLabel(objectLabel);
496}
497
498template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
499TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::TpetraCrsMatrix(const TpetraCrsMatrix &matrix)
500 : mtx_(Teuchos::rcp(new Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(*(matrix.mtx_), Teuchos::Copy))) {}
501
502template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
503void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag) const {
504 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
505 XPETRA_DYNAMIC_CAST(TpetraVectorClass, diag, tDiag, "Xpetra::TpetraCrsMatrix.getLocalDiagCopy() only accept Xpetra::TpetraVector as input arguments.");
506 mtx_->getLocalDiagCopy(*tDiag.getTpetra_Vector());
507}
508
509template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
510void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagOffsets(Teuchos::ArrayRCP<size_t> &offsets) const {
511 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagOffsets");
512 mtx_->getLocalDiagOffsets(offsets);
513}
514
515template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
516void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag, const Teuchos::ArrayView<const size_t> &offsets) const {
517 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
518 mtx_->getLocalDiagCopy(*(toTpetra(diag)), offsets);
519}
520
521template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
522void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getLocalDiagCopy(Vector &diag, const Kokkos::View<const size_t *, typename Node::device_type, Kokkos::MemoryUnmanaged> &offsets) const {
523 XPETRA_MONITOR("TpetraCrsMatrix::getLocalDiagCopy");
524 mtx_->getLocalDiagCopy(*(toTpetra(diag)), offsets);
525}
526
527template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
532
533template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
534void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::leftScale(const Vector &x) {
535 XPETRA_MONITOR("TpetraCrsMatrix::leftScale");
536 mtx_->leftScale(*(toTpetra(x)));
537}
538
539template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
540void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::rightScale(const Vector &x) {
541 XPETRA_MONITOR("TpetraCrsMatrix::rightScale");
542 mtx_->rightScale(*(toTpetra(x)));
543}
544
545template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
550
551template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
552void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doImport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &source,
553 const Import<LocalOrdinal, GlobalOrdinal, Node> &importer, CombineMode CM) {
554 XPETRA_MONITOR("TpetraCrsMatrix::doImport");
555
556 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, source, tSource, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
557 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSource.getTpetra_CrsMatrix();
558 // mtx_->doImport(toTpetraCrsMatrix(source), *tImporter.getTpetra_Import(), toTpetra(CM));
559 mtx_->doImport(*v, toTpetra(importer), toTpetra(CM));
560}
561
563template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
564void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doExport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &dest,
565 const Import<LocalOrdinal, GlobalOrdinal, Node> &importer, CombineMode CM) {
566 XPETRA_MONITOR("TpetraCrsMatrix::doExport");
567
568 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, dest, tDest, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
569 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tDest.getTpetra_CrsMatrix();
570 mtx_->doExport(*v, toTpetra(importer), toTpetra(CM));
571}
572
573template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
574void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doImport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &source,
575 const Export<LocalOrdinal, GlobalOrdinal, Node> &exporter, CombineMode CM) {
576 XPETRA_MONITOR("TpetraCrsMatrix::doImport");
577
578 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, source, tSource, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
579 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tSource.getTpetra_CrsMatrix();
580 mtx_->doImport(*v, toTpetra(exporter), toTpetra(CM));
581}
582
583template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
584void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::doExport(const DistObject<char, LocalOrdinal, GlobalOrdinal, Node> &dest,
585 const Export<LocalOrdinal, GlobalOrdinal, Node> &exporter, CombineMode CM) {
586 XPETRA_MONITOR("TpetraCrsMatrix::doExport");
587
588 XPETRA_DYNAMIC_CAST(const TpetraCrsMatrixClass, dest, tDest, "Xpetra::TpetraCrsMatrix::doImport only accept Xpetra::TpetraCrsMatrix as input arguments."); // TODO: remove and use toTpetra()
589 RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> v = tDest.getTpetra_CrsMatrix();
590 mtx_->doExport(*v, toTpetra(exporter), toTpetra(CM));
591}
592
593template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
594void TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::removeEmptyProcessesInPlace(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>> &newMap) {
595 XPETRA_MONITOR("TpetraCrsMatrix::removeEmptyProcessesInPlace");
596 mtx_->removeEmptyProcessesInPlace(toTpetra(newMap));
597}
598
599template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
600bool TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::hasMatrix() const {
601 return !mtx_.is_null();
602}
603
604template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
607
608template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
609RCP<const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getTpetra_CrsMatrix() const { return mtx_; }
610
612template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
613RCP<Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::getTpetra_CrsMatrixNonConst() const { return mtx_; } // TODO: remove
614
616template <class Scalar, class LocalOrdinal, class GlobalOrdinal, class Node>
622
625// End of TpetrCrsMatrix class definition //
628
629} // namespace Xpetra
630
631#endif // XPETRA_TPETRACRSMATRIX_DEF_HPP
Tpetra_Details_residual.hpp
XPETRA_MONITOR
#define XPETRA_MONITOR(funcName)
Definition Xpetra_ConfigDefs.hpp:94
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
Tpetra::CrsMatrix::getLocalRowView
void getLocalRowView(LocalOrdinal LocalRow, local_inds_host_view_type &indices, values_host_view_type &values) const override
Tpetra::CrsMatrix::getGlobalRowView
void getGlobalRowView(GlobalOrdinal GlobalRow, global_inds_host_view_type &indices, values_host_view_type &values) const override
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:154
Xpetra::TpetraCrsMatrix::supportsRowViews
bool supportsRowViews() const
Returns true if getLocalRowView() and getGlobalRowView() are valid for this class.
Definition Xpetra_TpetraCrsMatrix_def.hpp:387
Xpetra::TpetraCrsMatrix::local_matrix_type
typename Xpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node >::local_matrix_type local_matrix_type
Definition Xpetra_TpetraCrsMatrix_decl.hpp:54
Xpetra::TpetraCrsMatrix::getTpetra_CrsMatrixNonConst
RCP< Tpetra::CrsMatrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetra_CrsMatrixNonConst() const
Get the underlying Tpetra matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:613
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:166
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:381
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:406
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:160
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:503
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:339
Xpetra::TpetraCrsMatrix::scale
void scale(const Scalar &alpha)
Scale the current values of a matrix, this = alpha*this.
Definition Xpetra_TpetraCrsMatrix_def.hpp:185
Xpetra::TpetraCrsMatrix::rightScale
void rightScale(const Vector &x)
Right scale operator with given vector values.
Definition Xpetra_TpetraCrsMatrix_def.hpp:540
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:393
Xpetra::TpetraCrsMatrix::getGlobalNumCols
global_size_t getGlobalNumCols() const
Number of global columns in the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:303
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:237
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:285
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:191
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:617
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:564
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:363
Xpetra::TpetraCrsMatrix::resumeFill
void resumeFill(const RCP< ParameterList > &params=null)
Definition Xpetra_TpetraCrsMatrix_def.hpp:231
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:510
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:148
Xpetra::TpetraCrsMatrix::leftScale
void leftScale(const Vector &x)
Left scale operator with given vector values.
Definition Xpetra_TpetraCrsMatrix_def.hpp:534
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:430
Xpetra::TpetraCrsMatrix::haveGlobalConstants
bool haveGlobalConstants() const
Returns true if globalConstants have been computed; false otherwise.
Definition Xpetra_TpetraCrsMatrix_def.hpp:228
Xpetra::TpetraCrsMatrix::hasMatrix
bool hasMatrix() const
Does this have an underlying matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:600
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:351
Xpetra::TpetraCrsMatrix::description
std::string description() const
A simple one-line description of this object.
Definition Xpetra_TpetraCrsMatrix_def.hpp:480
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:609
Xpetra::TpetraCrsMatrix::getLocalNumRows
size_t getLocalNumRows() const
Returns the number of matrix rows owned on the calling node.
Definition Xpetra_TpetraCrsMatrix_def.hpp:309
Xpetra::TpetraCrsMatrix::setAllToScalar
void setAllToScalar(const Scalar &alpha)
Set all matrix entries equal to scalarThis.
Definition Xpetra_TpetraCrsMatrix_def.hpp:179
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:357
Xpetra::TpetraCrsMatrix::isFillActive
bool isFillActive() const
Returns true if the matrix is in edit mode.
Definition Xpetra_TpetraCrsMatrix_def.hpp:375
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:249
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:199
Xpetra::TpetraCrsMatrix::removeEmptyProcessesInPlace
void removeEmptyProcessesInPlace(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &newMap)
Definition Xpetra_TpetraCrsMatrix_def.hpp:594
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:315
Xpetra::TpetraCrsMatrix::getGlobalNumEntries
global_size_t getGlobalNumEntries() const
Returns the global number of entries in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:321
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:205
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:552
Xpetra::TpetraCrsMatrix::getGlobalNumRows
global_size_t getGlobalNumRows() const
Number of global elements in the row map of this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:297
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:486
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:333
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:419
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:468
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:257
Xpetra::TpetraCrsMatrix::replaceDiag
void replaceDiag(const Vector &diag)
Replace the diagonal entries of the matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:528
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:279
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:345
Xpetra::TpetraCrsMatrix::setObjectLabel
void setObjectLabel(const std::string &objectLabel)
Definition Xpetra_TpetraCrsMatrix_def.hpp:492
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:441
Xpetra::TpetraCrsMatrix::getCrsGraph
RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > getCrsGraph() const
Returns the CrsGraph associated with this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:291
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:474
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:369
Xpetra::TpetraCrsMatrix::getLocalNumEntries
size_t getLocalNumEntries() const
Returns the local number of entries in this matrix.
Definition Xpetra_TpetraCrsMatrix_def.hpp:327
Xpetra::TpetraCrsMatrix::getMap
Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getMap() const
Implements DistObject interface.
Definition Xpetra_TpetraCrsMatrix_def.hpp:546
TEUCHOS_TEST_FOR_EXCEPTION
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
Teuchos::EVerbosityLevel
EVerbosityLevel
Tpetra::Details::residual
void residual(const Operator< SC, LO, GO, NO > &Aop, const MultiVector< SC, LO, GO, NO > &X_in, const MultiVector< SC, LO, GO, NO > &B_in, MultiVector< SC, LO, GO, NO > &R_in)
Tpetra::replaceDiagonalCrsMatrix
LO replaceDiagonalCrsMatrix(::Tpetra::CrsMatrix< SC, LO, GO, NT > &matrix, const ::Tpetra::Vector< SC, LO, GO, NT > &newDiag)
Xpetra::toXpetra
RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toXpetra(RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > graph)
Definition Xpetra_TpetraCrsGraph_decl.hpp:361
Xpetra::global_size_t
size_t global_size_t
Global size_t object.
Definition Xpetra_ConfigDefs.hpp:136
Xpetra::toTpetra
RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toTpetra(const RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > &graph)
Definition Xpetra_TpetraCrsGraph_decl.hpp:374
Xpetra::CombineMode
CombineMode
Xpetra::Combine Mode enumerable type.
Definition Xpetra_ConfigDefs.hpp:174