46#ifndef XPETRA_TPETRAHALFPRECISIONOPEARTOR_HPP
47#define XPETRA_TPETRAHALFPRECISIONOPEARTOR_HPP
53#include <Tpetra_CrsMatrix.hpp>
55#include <Xpetra_MultiVector.hpp>
56#include <Xpetra_CrsMatrixWrap.hpp>
57#include <Xpetra_MultiVectorFactory.hpp>
61 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
62 RCP<Xpetra::Matrix<typename Teuchos::ScalarTraits<Scalar>::halfPrecision, LocalOrdinal, GlobalOrdinal, Node> >
64#if defined(HAVE_TPETRA_INST_DOUBLE) && defined(HAVE_TPETRA_INST_FLOAT)
68 RCP<XpCrs> xpCrs = Teuchos::rcp_dynamic_cast<Xpetra::CrsMatrixWrap<Scalar,LocalOrdinal,GlobalOrdinal,Node> >(A,
true)->getCrsMatrix();
69 auto tpCrs = Teuchos::rcp_dynamic_cast<Xpetra::TpetraCrsMatrix<Scalar,LocalOrdinal,GlobalOrdinal,Node> >(xpCrs,
true)->getTpetra_CrsMatrix();
70 auto newTpCrs = tpCrs->template convert<HalfScalar>();
77 "Xpetra::convertToHalfPrecision only available for Tpetra with SC=double and SC=float enabled");
82 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
83 RCP<Xpetra::MultiVector<typename Teuchos::ScalarTraits<Scalar>::halfPrecision, LocalOrdinal, GlobalOrdinal, Node> >
85#if defined(HAVE_TPETRA_INST_DOUBLE) && defined(HAVE_TPETRA_INST_FLOAT)
89 auto newTpX = tpX.template convert<HalfScalar>();
95 "Xpetra::convertToHalfPrecision only available for Tpetra with SC=double and SC=float enabled");
102 template <
class Scalar,
130 return Op_->getDomainMap();
135 return Op_->getRangeMap();
149 Tpetra::deep_copy(*Teuchos::rcp_dynamic_cast<tMVHalf>(
X_)->getTpetra_MultiVector(),
151 Op_->apply(*
X_, *
Y_, mode, Teuchos::as<HalfScalar>(alpha), Teuchos::as<HalfScalar>(beta));
153 *Teuchos::rcp_dynamic_cast<tMVHalf>(
Y_)->getTpetra_MultiVector());
164 R.
update(STS::one(),B,STS::zero());
Concrete implementation of Xpetra::Matrix.
Exception throws to report errors in the internal logical of the program.
Xpetra-specific matrix class.
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.
virtual void update(const Scalar &alpha, const MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const Scalar &beta)=0
Update multi-vector values with scaled values of A, this = beta*this + alpha*A.
Wraps an existing halfer precision Xpetra::Operator as a Xpetra::Operator.
void SetHalfPrecisionOperator(const RCP< Xpetra::Operator< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > &op)
Teuchos::ScalarTraits< Scalar >::halfPrecision HalfScalar
RCP< Xpetra::MultiVector< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > X_
RCP< Xpetra::MultiVector< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > Y_
void residual(const Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &X, const Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &B, Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &R) const
Compute a residual R = B - (*this) * X.
Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getRangeMap() const
Returns the Tpetra::Map object associated with the range of this TpetraOperator.
void apply(const Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &X, Xpetra::MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &Y, Teuchos::ETransp mode=Teuchos::NO_TRANS, Scalar alpha=Teuchos::ScalarTraits< Scalar >::one(), Scalar beta=Teuchos::ScalarTraits< Scalar >::one()) const
Returns in Y the result of a Xpetra::TpetraOperator applied to a Xpetra::MultiVector X.
RCP< Xpetra::Operator< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > Op_
virtual ~TpetraHalfPrecisionOperator()
Destructor.
TpetraHalfPrecisionOperator(const RCP< Xpetra::Operator< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > &op)
Constructor.
RCP< Xpetra::Operator< HalfScalar, LocalOrdinal, GlobalOrdinal, Node > > GetHalfPrecisionOperator() const
Direct access to the underlying TpetraOperator.
void Allocate(int numVecs)
Teuchos::RCP< const Xpetra::Map< LocalOrdinal, GlobalOrdinal, Node > > getDomainMap() const
Returns the Tpetra::Map object associated with the domain of this TpetraOperator.
bool hasTransposeApply() const
Indicates whether this TpetraOperator supports applying the adjoint TpetraOperator.
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
#define TEUCHOS_UNREACHABLE_RETURN(dummyReturnVal)
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
RCP< const Tpetra::CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > toTpetra(const RCP< const CrsGraph< LocalOrdinal, GlobalOrdinal, Node > > &graph)
RCP< Xpetra::Matrix< typename Teuchos::ScalarTraits< Scalar >::halfPrecision, LocalOrdinal, GlobalOrdinal, Node > > convertToHalfPrecision(RCP< Xpetra::Matrix< Scalar, LocalOrdinal, GlobalOrdinal, Node > > &A)