42 #ifndef THYRA_TPETRA_VECTOR_HPP
43 #define THYRA_TPETRA_VECTOR_HPP
46 #include "Thyra_TpetraVector_decl.hpp"
47 #include "Thyra_TpetraMultiVector.hpp"
55 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
60 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
66 initializeImpl(tpetraVectorSpace, tpetraVector);
70 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
76 initializeImpl(tpetraVectorSpace, tpetraVector);
80 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
84 return tpetraVector_.getNonconstObj();
88 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
97 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
101 if (domainSpace_.is_null()) {
102 domainSpace_ = tpetraVectorSpace<Scalar>(
103 Tpetra::createLocalMapWithNode<LocalOrdinal,GlobalOrdinal,Node>(
105 tpetraVector_.getConstObj()->getMap()->getComm()
116 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
120 return tpetraVectorSpace_;
127 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
131 *localValues = tpetraVector_.getNonconstObj()->get1dViewNonConst();
135 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
139 *localValues = tpetraVector_->get1dView();
146 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
154 if (!tpetraVector_.getNonconstObj()->isDistributed()) {
155 auto comm = tpetraVector_.getNonconstObj()->getMap()->getComm();
156 if (tpetraVector_.getConstObj()->getMap()->getComm()->getRank() == 0)
157 tpetraVector_.getNonconstObj()->randomize(l, u);
160 tpetraVector_.getNonconstObj()->reduce();
162 tpetraVector_.getNonconstObj()->randomize(l, u);
167 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
172 auto tx = this->getConstTpetraVector(Teuchos::rcpFromRef(x));
177 tpetraVector_.getNonconstObj()->abs(*tx);
184 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
189 auto tx = this->getConstTpetraVector(Teuchos::rcpFromRef(x));
194 tpetraVector_.getNonconstObj()->reciprocal(*tx);
201 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
206 auto tx = this->getConstTpetraVector(Teuchos::rcpFromRef(x));
212 tpetraVector_.getNonconstObj()->elementWiseMultiply(
213 ST::one(), *tx, *tpetraVector_.getConstObj(), ST::zero());
220 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
226 auto tx = this->getConstTpetraVector(Teuchos::rcpFromRef(x));
234 = Tpetra::createVector<Scalar>(tx->getMap());
235 temp->elementWiseMultiply(
236 ST::one(), *tx, *tpetraVector_.getConstObj(), ST::zero());
237 return ST::magnitude(ST::squareroot(tpetraVector_.getConstObj()->dot(*temp)));
244 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
257 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
268 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
280 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
293 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
296 tpetraVector_.getNonconstObj()->putScalar(alpha);
300 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
304 auto tmv = this->getConstTpetraMultiVector(Teuchos::rcpFromRef(mv));
309 tpetraVector_.getNonconstObj()->assign(*tmv);
316 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
319 tpetraVector_.getNonconstObj()->scale(alpha);
323 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
329 auto tmv = this->getConstTpetraMultiVector(Teuchos::rcpFromRef(mv));
335 tpetraVector_.getNonconstObj()->update(alpha, *tmv, ST::one());
342 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
356 bool allCastsSuccessful =
true;
358 auto mvIter = mv.begin();
359 auto tmvIter = tmvs.begin();
360 for (; mvIter != mv.end(); ++mvIter, ++tmvIter) {
361 tmv = this->getConstTpetraMultiVector(Teuchos::rcpFromPtr(*mvIter));
365 allCastsSuccessful =
false;
373 auto len = mv.size();
375 tpetraVector_.getNonconstObj()->scale(beta);
376 }
else if (len == 1 && allCastsSuccessful) {
377 tpetraVector_.getNonconstObj()->update(alpha[0], *tmvs[0], beta);
378 }
else if (len == 2 && allCastsSuccessful) {
379 tpetraVector_.getNonconstObj()->update(alpha[0], *tmvs[0], alpha[1], *tmvs[1], beta);
380 }
else if (allCastsSuccessful) {
382 auto tmvIter = tmvs.begin();
383 auto alphaIter = alpha.
begin();
388 for (; tmvIter != tmvs.end(); ++tmvIter) {
389 if (tmvIter->getRawPtr() == tpetraVector_.getConstObj().getRawPtr()) {
397 tmvIter = tmvs.begin();
401 if ((tmvs.size() % 2) == 0) {
402 tpetraVector_.getNonconstObj()->scale(beta);
404 tpetraVector_.getNonconstObj()->update(*alphaIter, *(*tmvIter), beta);
408 for (; tmvIter != tmvs.end(); tmvIter+=2, alphaIter+=2) {
409 tpetraVector_.getNonconstObj()->update(
410 *alphaIter, *(*tmvIter), *(alphaIter+1), *(*(tmvIter+1)), ST::one());
418 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
424 auto tmv = this->getConstTpetraMultiVector(Teuchos::rcpFromRef(mv));
429 tpetraVector_.getConstObj()->dot(*tmv, prods);
436 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
441 tpetraVector_.getConstObj()->norm1(norms);
445 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
450 tpetraVector_.getConstObj()->norm2(norms);
454 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
459 tpetraVector_.getConstObj()->normInf(norms);
463 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
483 "Error, conjugation without transposition is not allowed for complex scalar types!");
501 Y_tpetra->multiply(trans,
Teuchos::NO_TRANS, alpha, *tpetraVector_.getConstObj(), *X_tpetra, beta);
513 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
514 template<
class TpetraVector_t>
524 tpetraVectorSpace_ = tpetraVectorSpace;
525 tpetraVector_.initialize(tpetraVector);
526 this->updateSpmdSpace();
530 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
531 RCP<Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> >
532 TpetraVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>::
533 getTpetraMultiVector(
const RCP<MultiVectorBase<Scalar> >& mv)
const
535 using Teuchos::rcp_dynamic_cast;
536 typedef TpetraMultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> TMV;
537 typedef TpetraVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> TV;
539 RCP<TMV> tmv = rcp_dynamic_cast<TMV>(mv);
541 return tmv->getTpetraMultiVector();
544 RCP<TV> tv = rcp_dynamic_cast<TV>(mv);
546 return tv->getTpetraVector();
549 return Teuchos::null;
553 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
554 RCP<const Tpetra::MultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> >
555 TpetraVector<Scalar,LocalOrdinal,GlobalOrdinal,Node>::
556 getConstTpetraMultiVector(
const RCP<
const MultiVectorBase<Scalar> >& mv)
const
558 using Teuchos::rcp_dynamic_cast;
559 typedef TpetraMultiVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> TMV;
560 typedef TpetraVector<Scalar,LocalOrdinal,GlobalOrdinal,Node> TV;
562 RCP<const TMV> tmv = rcp_dynamic_cast<
const TMV>(mv);
564 return tmv->getConstTpetraMultiVector();
567 RCP<const TV> tv = rcp_dynamic_cast<
const TV>(mv);
569 return tv->getConstTpetraVector();
572 return Teuchos::null;
576 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
577 RCP<Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node> >
581 typedef TpetraVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> TV;
582 RCP<TV> tv = Teuchos::rcp_dynamic_cast<TV>(v);
584 return tv->getTpetraVector();
586 return Teuchos::null;
591 template <
class Scalar,
class LocalOrdinal,
class GlobalOrdinal,
class Node>
592 RCP<const Tpetra::Vector<Scalar,LocalOrdinal,GlobalOrdinal,Node> >
596 typedef TpetraVector<Scalar, LocalOrdinal, GlobalOrdinal, Node> TV;
597 RCP<const TV> tv = Teuchos::rcp_dynamic_cast<
const TV>(v);
599 return tv->getConstTpetraVector();
601 return Teuchos::null;
609 #endif // THYRA_TPETRA_VECTOR_HPP
void constInitialize(const RCP< const TpetraVectorSpace< Scalar, LocalOrdinal, GlobalOrdinal, Node > > &tpetraVectorSpace, const RCP< const Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > &tpetraVector)
Initialize.
void applyOpImpl(const RTOpPack::RTOpT< Scalar > &op, const ArrayView< const Ptr< const VectorBase< Scalar > > > &vecs, const ArrayView< const Ptr< VectorBase< Scalar > > > &targ_vecs, const Ptr< RTOpPack::ReductTarget > &reduct_obj, const Ordinal global_offset) const
Calls applyOpImplWithComm(null,op,...).
void applyImpl(const EOpTransp M_trans, const MultiVectorBase< Scalar > &X, const Ptr< MultiVectorBase< Scalar > > &Y, const Scalar alpha, const Scalar beta) const
Concrete implementation of an SPMD vector space for Tpetra.
void commitNonconstDetachedVectorViewImpl(RTOpPack::SubVectorView< Scalar > *sub_vec)
EOpTransp
Enumeration for determining how a linear operator is applied. `*.
void acquireDetachedVectorViewImpl(const Range1D &rng, RTOpPack::ConstSubVectorView< Scalar > *sub_vec) const
Implemented through this->getLocalData()
void initialize(const RCP< const TpetraVectorSpace< Scalar, LocalOrdinal, GlobalOrdinal, Node > > &tpetraVectorSpace, const RCP< Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > &tpetraVector)
Initialize.
TpetraVector()
Construct to uninitialized.
virtual void assignMultiVecImpl(const MultiVectorBase< Scalar > &mv)
Default implementation of assign(MV) using RTOps.
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
virtual void norms1Impl(const ArrayView< typename ScalarTraits< Scalar >::magnitudeType > &norms) const
Use the non-transposed operator.
virtual void applyOpImpl(const RTOpPack::RTOpT< Scalar > &op, const ArrayView< const Ptr< const VectorBase< Scalar > > > &vecs, const ArrayView< const Ptr< VectorBase< Scalar > > > &targ_vecs, const Ptr< RTOpPack::ReductTarget > &reduct_obj, const Ordinal global_offset) const
virtual void eleWiseScaleImpl(const VectorBase< Scalar > &x)
Default implementation of ele_wise_scale using RTOps.
virtual void reciprocalImpl(const VectorBase< Scalar > &x)
Default implementation of reciprocal using RTOps.
virtual void linearCombinationImpl(const ArrayView< const Scalar > &alpha, const ArrayView< const Ptr< const MultiVectorBase< Scalar > > > &mv, const Scalar &beta)
Default implementation of linear_combination using RTOps.
virtual void norms2Impl(const ArrayView< typename ScalarTraits< Scalar >::magnitudeType > &norms) const
Use the transposed operator with complex-conjugate clements (same as TRANS for real scalar types)...
void getLocalVectorDataImpl(const Ptr< ArrayRCP< const Scalar > > &localValues) const
virtual void assignMultiVecImpl(const MultiVectorBase< Scalar > &mv)
virtual void absImpl(const VectorBase< Scalar > &x)
Default implementation of abs using RTOps.
Use the non-transposed operator with complex-conjugate elements (same as NOTRANS for real scalar type...
void acquireNonconstDetachedVectorViewImpl(const Range1D &rng, RTOpPack::SubVectorView< Scalar > *sub_vec)
Use the transposed operator.
RCP< const Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getConstTpetraVector() const
Get the embedded non-const Tpetra::Vector.
virtual void reciprocalImpl(const VectorBase< Scalar > &x)
void acquireDetachedVectorViewImpl(const Range1D &rng, RTOpPack::ConstSubVectorView< Scalar > *sub_vec) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
virtual Teuchos::ScalarTraits< Scalar >::magnitudeType norm2WeightedImpl(const VectorBase< Scalar > &x) const
Teuchos::Ordinal Ordinal
Type for the dimension of a vector space. `*.
Interface for a collection of column vectors called a multi-vector.
virtual void linearCombinationImpl(const ArrayView< const Scalar > &alpha, const ArrayView< const Ptr< const MultiVectorBase< Scalar > > > &mv, const Scalar &beta)
virtual void absImpl(const VectorBase< Scalar > &x)
void getNonconstLocalVectorDataImpl(const Ptr< ArrayRCP< Scalar > > &localValues)
Abstract interface for finite-dimensional dense vectors.
RCP< const SpmdVectorSpaceBase< Scalar > > spmdSpaceImpl() const
Concrete Thyra::SpmdVectorBase using Tpetra::Vector.
virtual void dotsImpl(const MultiVectorBase< Scalar > &mv, const ArrayView< Scalar > &prods) const
RCP< Tpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getTpetraVector()
Get the embedded non-const Tpetra::Vector.
void applyImpl(const EOpTransp M_trans, const MultiVectorBase< Scalar > &X, const Ptr< MultiVectorBase< Scalar > > &Y, const Scalar alpha, const Scalar beta) const
. Applies vector or its adjoint (transpose) as a linear operator.
void commitNonconstDetachedVectorViewImpl(RTOpPack::SubVectorView< Scalar > *sub_vec)
Implemented through this->commitLocalData()
bool nonnull(const boost::shared_ptr< T > &p)
virtual void assignImpl(Scalar alpha)
virtual void eleWiseScaleImpl(const VectorBase< Scalar > &x)
virtual void normsInfImpl(const ArrayView< typename ScalarTraits< Scalar >::magnitudeType > &norms) const
RCP< const VectorSpaceBase< Scalar > > domain() const
virtual void randomizeImpl(Scalar l, Scalar u)
virtual Teuchos::ScalarTraits< Scalar >::magnitudeType norm2WeightedImpl(const VectorBase< Scalar > &x) const
Default implementation of norm_2 (weighted) using RTOps.
virtual void updateImpl(Scalar alpha, const MultiVectorBase< Scalar > &mv)
virtual void scaleImpl(Scalar alpha)
#define TEUCHOS_ASSERT(assertion_test)
#define TEUCHOS_ASSERT_EQUALITY(val1, val2)
virtual void dotsImpl(const MultiVectorBase< Scalar > &mv, const ArrayView< Scalar > &prods) const
Default implementation of dots using RTOps.
virtual void updateImpl(Scalar alpha, const MultiVectorBase< Scalar > &mv)
Default implementation of update using RTOps.
void acquireNonconstDetachedVectorViewImpl(const Range1D &rng, RTOpPack::SubVectorView< Scalar > *sub_vec)
Implemented through this->getLocalData()