void Jacobi(
Scalar omega,
const Xpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node> & Dinv,
const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& A,
const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& B,
Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& C,
bool call_FillComplete_on_result = true,
bool doOptimizeStorage = true) {
if(C.getRowMap()->isSameAs(*A.getRowMap()) == false) {
std::string msg = "XpetraExt::MatrixMatrix::Jacobi: row map of C is not same as row map of A";
throw(Xpetra::Exceptions::RuntimeError(msg));
}
else if(C.getRowMap()->isSameAs(*B.getRowMap()) == false) {
std::string msg = "XpetraExt::MatrixMatrix::Jacobi: row map of C is not same as row map of B";
throw(Xpetra::Exceptions::RuntimeError(msg));
}
if (!A.isFillComplete())
throw(Xpetra::Exceptions::RuntimeError("A is not fill-completed"));
if (!B.isFillComplete())
throw(Xpetra::Exceptions::RuntimeError("B is not fill-completed"));
bool haveMultiplyDoFillComplete = call_FillComplete_on_result && doOptimizeStorage;
if (C.getRowMap()->lib() == Xpetra::UseEpetra) {
#ifndef HAVE_XPETRA_EPETRAEXT
throw(Xpetra::Exceptions::RuntimeError("Xpetra::MatrixMatrix::Jacobi requires EpetraExt to be compiled."));
#else
throw(Xpetra::Exceptions::RuntimeError("Xpetra::MatrixMatrix::Jacobi requires you to use an Epetra-compatible data type."));
#endif
} else if (C.getRowMap()->lib() == Xpetra::UseTpetra) {
#ifdef HAVE_XPETRA_TPETRA
const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpA = Xpetra::MatrixMatrix::Op2TpetraCrs(A);
const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpB = Xpetra::MatrixMatrix::Op2TpetraCrs(B);
Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpC = Xpetra::MatrixMatrix::Op2NonConstTpetraCrs(C);
const RCP<Tpetra::Vector<Scalar, LocalOrdinal, GlobalOrdinal, Node> > & tpD = toTpetra(Dinv);
Tpetra::MatrixMatrix::Jacobi(omega,*tpD,tpA,tpB,tpC,haveMultiplyDoFillComplete);
#else
throw(Xpetra::Exceptions::RuntimeError("Xpetra must be compiled with Tpetra."));
#endif
}
if(call_FillComplete_on_result && !haveMultiplyDoFillComplete) {
RCP<Teuchos::ParameterList> params = rcp(new Teuchos::ParameterList());
params->set("Optimize Storage",doOptimizeStorage);
C.fillComplete(B.getDomainMap(),B.getRangeMap(),params);
}
// transfer striding information
RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> > rcpA = Teuchos::rcp_const_cast<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> >(Teuchos::rcpFromRef(A));
RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> > rcpB = Teuchos::rcp_const_cast<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> >(Teuchos::rcpFromRef(B));
C.CreateView("stridedMaps", rcpA, false, rcpB, false); // TODO use references instead of RCPs
} // end Jacobi
void Multiply(
const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& A,
bool transposeA,
const Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& B,
bool transposeB,
Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps>& C,
bool call_FillComplete_on_result = true,
bool doOptimizeStorage = true) {
if(transposeA == false && C.getRowMap()->isSameAs(*A.getRowMap()) == false) {
std::string msg = "XpetraExt::MatrixMatrix::Multiply: row map of C is not same as row map of A";
throw(Xpetra::Exceptions::RuntimeError(msg));
}
else if(transposeA == true && C.getRowMap()->isSameAs(*A.getDomainMap()) == false) {
std::string msg = "XpetraExt::MatrixMatrix::Multiply: row map of C is not same as domain map of A";
throw(Xpetra::Exceptions::RuntimeError(msg));
}
if (!A.isFillComplete())
throw(Xpetra::Exceptions::RuntimeError("A is not fill-completed"));
if (!B.isFillComplete())
throw(Xpetra::Exceptions::RuntimeError("B is not fill-completed"));
bool haveMultiplyDoFillComplete = call_FillComplete_on_result && doOptimizeStorage;
if (C.getRowMap()->lib() == Xpetra::UseEpetra) {
# ifndef HAVE_XPETRA_EPETRAEXT
throw(Xpetra::Exceptions::RuntimeError("Xpetra::MatrixMatrix::Multiply requires EpetraExt to be compiled."));
#else
Epetra_CrsMatrix & epA = Xpetra::MatrixMatrix::Op2NonConstEpetraCrs(A);
Epetra_CrsMatrix & epB = Xpetra::MatrixMatrix::Op2NonConstEpetraCrs(B);
Epetra_CrsMatrix & epC = Xpetra::MatrixMatrix::Op2NonConstEpetraCrs(C);
int i = EpetraExt::MatrixMatrix::Multiply(epA,transposeA,epB,transposeB,epC,haveMultiplyDoFillComplete);
if (i != 0) {
std::ostringstream buf;
buf << i;
std::string msg = "EpetraExt::MatrixMatrix::Multiply return value of " + buf.str();
throw(Exceptions::RuntimeError(msg));
}
#endif
} else if (C.getRowMap()->lib() == Xpetra::UseTpetra) {
#ifdef HAVE_XPETRA_TPETRA
const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpA = Xpetra::MatrixMatrix::Op2TpetraCrs(A);
const Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpB = Xpetra::MatrixMatrix::Op2TpetraCrs(B);
Tpetra::CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> & tpC = Xpetra::MatrixMatrix::Op2NonConstTpetraCrs(C);
//18Feb2013 JJH I'm reenabling the code that allows the matrix matrix multiply to do the fillComplete.
//Previously, Tpetra's matrix matrix multiply did not support fillComplete.
Tpetra::MatrixMatrix::Multiply(tpA,transposeA,tpB,transposeB,tpC,haveMultiplyDoFillComplete);
#else
throw(Xpetra::Exceptions::RuntimeError("Xpetra must be compiled with Tpetra."));
#endif
}
if(call_FillComplete_on_result && !haveMultiplyDoFillComplete) {
RCP<Teuchos::ParameterList> params = rcp(new Teuchos::ParameterList());
params->set("Optimize Storage",doOptimizeStorage);
C.fillComplete((transposeB) ? B.getRangeMap() : B.getDomainMap(),
(transposeA) ? A.getDomainMap() : A.getRangeMap(),
params);
}
// transfer striding information
RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> > rcpA = Teuchos::rcp_const_cast<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> >(Teuchos::rcpFromRef(A));
RCP<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> > rcpB = Teuchos::rcp_const_cast<Xpetra::Matrix<Scalar, LocalOrdinal, GlobalOrdinal, Node, LocalMatOps> >(Teuchos::rcpFromRef(B));
C.CreateView("stridedMaps", rcpA, transposeA, rcpB, transposeB); // TODO use references instead of RCPs
} // end Multiply
