116 RCP<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsOp = rcp_dynamic_cast<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
117 if (crsOp == Teuchos::null)
119 const RCP<const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>>& tmp_ECrsMtx = rcp_dynamic_cast<const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>>(crsOp->getCrsMatrix());
120 if (tmp_ECrsMtx == Teuchos::null)
121 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
122 return tmp_ECrsMtx->getEpetra_CrsMatrix();
127 RCP<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsOp = rcp_dynamic_cast<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
128 if (crsOp == Teuchos::null)
130 const RCP<const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>>& tmp_ECrsMtx = rcp_dynamic_cast<const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>>(crsOp->getCrsMatrix());
131 if (tmp_ECrsMtx == Teuchos::null)
132 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
133 return tmp_ECrsMtx->getEpetra_CrsMatrixNonConst();
139 const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>& crsOp =
dynamic_cast<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>&
>(Op);
141 const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>& tmp_ECrsMtx =
dynamic_cast<const Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>&
>(*crsOp.getCrsMatrix());
142 return *tmp_ECrsMtx.getEpetra_CrsMatrix();
143 }
catch (std::bad_cast&) {
144 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
146 }
catch (std::bad_cast&) {
154 Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>& crsOp =
dynamic_cast<Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>&
>(Op);
156 Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>& tmp_ECrsMtx =
dynamic_cast<Xpetra::EpetraCrsMatrixT<GlobalOrdinal, Node>&
>(*crsOp.getCrsMatrix());
157 return *tmp_ECrsMtx.getEpetra_CrsMatrixNonConst();
158 }
catch (std::bad_cast&) {
159 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::EpetraCrsMatrix failed");
161 }
catch (std::bad_cast&) {
177 RCP<const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> mat = rcp_dynamic_cast<const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
178 RCP<const Xpetra::TpetraRowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> rmat = rcp_dynamic_cast<const Xpetra::TpetraRowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
179 if (!mat.is_null()) {
180 RCP<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsOp = rcp_dynamic_cast<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(mat);
181 if (crsOp == Teuchos::null)
184 RCP<const Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsMat = crsOp->getCrsMatrix();
185 const RCP<const Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tmp_Crs = rcp_dynamic_cast<const Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(crsMat);
186 RCP<const Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tmp_BlockCrs;
187 if (!tmp_Crs.is_null()) {
188 return tmp_Crs->getTpetra_CrsMatrixNonConst();
190 tmp_BlockCrs = rcp_dynamic_cast<const Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(crsMat);
191 if (tmp_BlockCrs.is_null())
192 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix and Xpetra::TpetraBlockCrsMatrix failed");
193 return tmp_BlockCrs->getTpetra_BlockCrsMatrixNonConst();
195 }
else if (!rmat.is_null()) {
196 return rmat->getTpetra_RowMatrix();
198 RCP<const Xpetra::TpetraOperator<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tpOp = rcp_dynamic_cast<const Xpetra::TpetraOperator<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op,
true);
199 RCP<const Tpetra::Operator<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tOp = tpOp->getOperatorConst();
200 RCP<const Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tRow = rcp_dynamic_cast<const Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(tOp,
true);
207 RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> mat = rcp_dynamic_cast<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
208 RCP<Xpetra::TpetraRowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> rmat = rcp_dynamic_cast<Xpetra::TpetraRowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op);
209 if (!mat.is_null()) {
210 RCP<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsOp = rcp_dynamic_cast<const Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(mat);
211 if (crsOp == Teuchos::null)
214 RCP<const Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> crsMat = crsOp->getCrsMatrix();
215 const RCP<const Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tmp_Crs = rcp_dynamic_cast<const Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(crsMat);
216 RCP<const Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tmp_BlockCrs;
217 if (!tmp_Crs.is_null()) {
218 return tmp_Crs->getTpetra_CrsMatrixNonConst();
220 tmp_BlockCrs = rcp_dynamic_cast<const Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(crsMat);
221 if (tmp_BlockCrs.is_null())
222 throw Exceptions::BadCast(
"Cast from Xpetra::CrsMatrix to Xpetra::TpetraCrsMatrix and Xpetra::TpetraBlockCrsMatrix failed");
223 return tmp_BlockCrs->getTpetra_BlockCrsMatrixNonConst();
225 }
else if (!rmat.is_null()) {
226 return rmat->getTpetra_RowMatrixNonConst();
228 RCP<Xpetra::TpetraOperator<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tpOp = rcp_dynamic_cast<Xpetra::TpetraOperator<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(Op,
true);
229 RCP<Tpetra::Operator<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tOp = tpOp->getOperator();
230 RCP<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> tRow = rcp_dynamic_cast<Tpetra::RowMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(tOp,
true);
238 bool doOptimizeStorage) {
239 Scalar one = Teuchos::ScalarTraits<Scalar>::one();
240 Teuchos::ArrayRCP<Scalar> sv(scalingVector.size());
242 for (
int i = 0; i < scalingVector.size(); ++i)
243 sv[i] = one / scalingVector[i];
245 for (
int i = 0; i < scalingVector.size(); ++i)
246 sv[i] = scalingVector[i];
249 switch (Op.getRowMap()->lib()) {
250 case Xpetra::UseTpetra:
251 MyOldScaleMatrix_Tpetra(Op, sv, doFillComplete, doOptimizeStorage);
254 case Xpetra::UseEpetra:
255 MyOldScaleMatrix_Epetra(Op, sv, doFillComplete, doOptimizeStorage);
271 bool doOptimizeStorage) {
273 Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& tpOp = toTpetra(Op);
275 const RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> rowMap = tpOp.getRowMap();
276 const RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> domainMap = tpOp.getDomainMap();
277 const RCP<const Tpetra::Map<LocalOrdinal, GlobalOrdinal, Node>> rangeMap = tpOp.getRangeMap();
279 size_t maxRowSize = tpOp.getLocalMaxNumRowEntries();
280 if (maxRowSize == Teuchos::as<size_t>(-1))
283 if (tpOp.isFillComplete())
286 if (Op.isLocallyIndexed() ==
true) {
287 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::local_inds_host_view_type cols;
288 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type vals;
290 for (
size_t i = 0; i < rowMap->getLocalNumElements(); ++i) {
291 tpOp.getLocalRowView(i, cols, vals);
293 size_t nnz = tpOp.getNumEntriesInLocalRow(i);
294 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type scaledVals(
"ScaledVals", nnz);
296 for (
size_t j = 0; j < nnz; ++j) {
297 scaledVals[j] = scalingVector[i] * vals[j];
301 tpOp.replaceLocalValues(i, cols, scaledVals);
306 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::global_inds_host_view_type cols;
307 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::values_host_view_type vals;
309 for (
size_t i = 0; i < rowMap->getLocalNumElements(); ++i) {
311 tpOp.getGlobalRowView(gid, cols, vals);
312 size_t nnz = tpOp.getNumEntriesInGlobalRow(gid);
313 typename Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>::nonconst_values_host_view_type scaledVals(
"ScaledVals", nnz);
316 for (
size_t j = 0; j < nnz; ++j)
317 scaledVals[j] = scalingVector[i] * vals[j];
320 tpOp.replaceGlobalValues(gid, cols, scaledVals);
325 if (doFillComplete) {
326 if (domainMap == Teuchos::null || rangeMap == Teuchos::null)
327 throw Exceptions::RuntimeError(
"In Utilities::Scaling: cannot fillComplete because the domain and/or range map hasn't been defined");
329 RCP<Teuchos::ParameterList> params = rcp(
new Teuchos::ParameterList());
330 params->set(
"Optimize Storage", doOptimizeStorage);
331 params->set(
"No Nonlocal Changes",
true);
332 Op.fillComplete(Op.getDomainMap(), Op.getRangeMap(), params);
342 Transpose(Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Op,
bool ,
const std::string& label,
const Teuchos::RCP<Teuchos::ParameterList>& params) {
343#if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_EPETRAEXT)
344 std::string TorE =
"epetra";
346 std::string TorE =
"tpetra";
349#if defined(HAVE_MUELU_EPETRA) && defined(HAVE_MUELU_EPETRAEXT)
358 if (TorE ==
"tpetra") {
359 using Helpers = Xpetra::Helpers<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
361 if (Helpers::isTpetraCrs(Op)) {
362 const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& tpetraOp = toTpetra(Op);
364 RCP<Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> A;
365 Tpetra::RowMatrixTransposer<Scalar, LocalOrdinal, GlobalOrdinal, Node> transposer(rcpFromRef(tpetraOp), label);
368 using Teuchos::ParameterList;
370 RCP<ParameterList> transposeParams = params.is_null() ? rcp(
new ParameterList) : rcp(
new ParameterList(*params));
371 transposeParams->set(
"sort",
false);
372 A = transposer.createTranspose(transposeParams);
375 RCP<Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> AA = rcp(
new Xpetra::TpetraCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(A));
376 RCP<Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> AAA = rcp_implicit_cast<Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(AA);
377 RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> AAAA = rcp(
new Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>(AAA));
378 if (!AAAA->isFillComplete())
379 AAAA->fillComplete(Op.getRangeMap(), Op.getDomainMap());
381 if (Op.IsView(
"stridedMaps"))
382 AAAA->CreateView(
"stridedMaps", Teuchos::rcpFromRef(Op),
true );
385 }
else if (Helpers::isTpetraBlockCrs(Op)) {
386 using XMatrix = Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
387 using XCrsMatrix = Xpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
388 using XCrsMatrixWrap = Xpetra::CrsMatrixWrap<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
389 using BCRS = Tpetra::BlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>;
391 const BCRS& tpetraOp = toTpetraBlock(Op);
395 Tpetra::BlockCrsMatrixTransposer<Scalar, LocalOrdinal, GlobalOrdinal, Node> transposer(rcpFromRef(tpetraOp), label);
397 using Teuchos::ParameterList;
399 RCP<ParameterList> transposeParams = params.is_null() ? rcp(
new ParameterList) : rcp(
new ParameterList(*params));
400 transposeParams->set(
"sort",
false);
401 At = transposer.createTranspose(transposeParams);
404 RCP<Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>> AA = rcp(
new Xpetra::TpetraBlockCrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>(At));
405 RCP<XCrsMatrix> AAA = rcp_implicit_cast<XCrsMatrix>(AA);
406 RCP<XMatrix> AAAA = rcp(
new XCrsMatrixWrap(AAA));
408 if (Op.IsView(
"stridedMaps"))
409 AAAA->CreateView(
"stridedMaps", Teuchos::rcpFromRef(Op),
true );
418 std::cout <<
"Utilities::Transpose() not implemented for Epetra" << std::endl;
419 return Teuchos::null;
427 RCP<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>> Xscalar;
428#if defined(HAVE_XPETRA_TPETRA) && (defined(HAVE_TPETRA_INST_COMPLEX_DOUBLE) || defined(HAVE_TPETRA_INST_COMPLEX_FLOAT))
429 using range_type = Kokkos::RangePolicy<LocalOrdinal, typename Node::execution_space>;
432 if ((
typeid(
Scalar).name() ==
typeid(std::complex<double>).name()) ||
433 (
typeid(
Scalar).name() ==
typeid(std::complex<float>).name())) {
434 size_t numVecs = X->getNumVectors();
435 Xscalar = Xpetra::MultiVectorFactory<Scalar, LocalOrdinal, GlobalOrdinal, Node>::Build(X->getMap(), numVecs);
436 auto XVec = X->getDeviceLocalView(Xpetra::Access::ReadOnly);
437 auto XVecScalar = Xscalar->getDeviceLocalView(Xpetra::Access::ReadWrite);
439 Kokkos::parallel_for(
440 "MueLu:Utils::RealValuedToScalarMultiVector", range_type(0, X->getLocalLength()),
441 KOKKOS_LAMBDA(
const size_t i) {
442 for (
size_t j = 0; j < numVecs; j++)
443 XVecScalar(i, j) = XVec(i, j);
447 Xscalar = rcp_dynamic_cast<Xpetra::MultiVector<Scalar, LocalOrdinal, GlobalOrdinal, Node>>(X);
455 RCP<Xpetra::MultiVector<typename Teuchos::ScalarTraits<Scalar>::magnitudeType,
LocalOrdinal,
GlobalOrdinal,
Node>> coordinates = Teuchos::null;
458 if (paramList.isParameter(
"Coordinates") ==
false)
464#if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_FLOAT)
465 typedef Tpetra::MultiVector<float, LocalOrdinal, GlobalOrdinal, Node> tfMV;
466 RCP<tfMV> floatCoords = Teuchos::null;
472#if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_DOUBLE)
474 RCP<tdMV> doubleCoords = Teuchos::null;
475 if (paramList.isType<RCP<tdMV>>(
"Coordinates")) {
477 doubleCoords = paramList.get<RCP<tdMV>>(
"Coordinates");
478 paramList.remove(
"Coordinates");
480#if !defined(HAVE_TPETRA_EXPLICIT_INSTANTIATION) || defined(HAVE_TPETRA_INST_FLOAT)
481 else if (paramList.isType<RCP<tfMV>>(
"Coordinates")) {
483 floatCoords = paramList.get<RCP<tfMV>>(
"Coordinates");
484 paramList.remove(
"Coordinates");
485 doubleCoords = rcp(
new tdMV(floatCoords->getMap(), floatCoords->getNumVectors()));
486 deep_copy(*doubleCoords, *floatCoords);
490 if (doubleCoords != Teuchos::null) {
492 coordinates = Teuchos::rcp_dynamic_cast<Xpetra::MultiVector<typename Teuchos::ScalarTraits<Scalar>::magnitudeType,
LocalOrdinal,
GlobalOrdinal,
Node>>(MueLu::TpetraMultiVector_To_XpetraMultiVector<typename Teuchos::ScalarTraits<Scalar>::magnitudeType,
LocalOrdinal,
GlobalOrdinal,
Node>(doubleCoords));
493 TEUCHOS_TEST_FOR_EXCEPT(Teuchos::is_null(coordinates));
494 TEUCHOS_TEST_FOR_EXCEPT(doubleCoords->getNumVectors() != coordinates->getNumVectors());
499 throw Exceptions::RuntimeError(
"ExtractCoordinatesFromParameterList: The coordinates vector in parameter list is expected to be a Tpetra multivector with SC=double or float.");
503 if (paramList.isType<
decltype(coordinates)>(
"Coordinates")) {
504 coordinates = paramList.get<
decltype(coordinates)>(
"Coordinates");
static RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > Transpose(Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Op, bool optimizeTranspose=false, const std::string &label=std::string(), const Teuchos::RCP< Teuchos::ParameterList > ¶ms=Teuchos::null)