47 #ifndef XPETRA_BLOCKEDMULTIVECTOR_HPP 48 #define XPETRA_BLOCKEDMULTIVECTOR_HPP 60 #ifndef DOXYGEN_SHOULD_SKIP_THIS 62 template<
class S,
class LO,
class GO,
class N>
class Vector;
63 template<
class S,
class LO,
class GO,
class N>
class BlockedVector;
64 template<
class S,
class LO,
class GO,
class N>
class MapExtractor;
65 template<
class S,
class LO,
class GO,
class N>
class MultiVectorFactory;
68 template <
class Scalar = double,
73 :
public MultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node >
82 #undef XPETRA_BLOCKEDMULTIVECTOR_SHORT 106 vv_.reserve(map->getNumMaps());
109 for (
size_t r = 0; r < map->getNumMaps(); ++r)
126 XPETRA_TEST_FOR_EXCEPTION(bmap->getMap()->getNodeNumElements() != v->getMap()->getNodeNumElements(),
Xpetra::Exceptions::RuntimeError,
"BlockedMultiVector: inconsistent number of local elements of MultiVector and BlockedMap. The BlockedMap has " << bmap->getMap()->getNodeNumElements() <<
" local elements. The vector has " << v->getMap()->getNodeNumElements() <<
".");
127 XPETRA_TEST_FOR_EXCEPTION(bmap->getMap()->getGlobalNumElements() != v->getMap()->getGlobalNumElements(),
Xpetra::Exceptions::RuntimeError,
"BlockedMultiVector: inconsistent number of global elements of MultiVector and BlockedMap. The BlockedMap has " << bmap->getMap()->getGlobalNumElements() <<
" local elements. The vector has " << v->getMap()->getGlobalNumElements() <<
".");
136 vv_.reserve(bmap->getNumMaps());
139 for (
size_t r = 0; r < bmap->getNumMaps(); ++r)
140 vv_.push_back(this->ExtractVector(v, r, bmap->getThyraMode()));
164 vv_.reserve(bmap->getNumMaps());
167 for (
size_t r = 0; r < bmap->getNumMaps(); ++r)
168 vv_.push_back(this->ExtractVector(v, r, bmap->getThyraMode()));
187 std::vector<RCP<const Map> > maps;
188 maps.reserve(mapExtractor->NumMaps());
189 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
190 maps.push_back(mapExtractor->getMap(r,mapExtractor->getThyraMode()));
194 vv_.reserve(mapExtractor->NumMaps());
197 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
198 vv_.push_back(this->ExtractVector(v, r, mapExtractor->getThyraMode()));
217 std::vector<RCP<const Map> > maps;
218 maps.reserve(mapExtractor->NumMaps());
219 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
220 maps.push_back(mapExtractor->getMap(r,mapExtractor->getThyraMode()));
223 vv_.reserve(mapExtractor->NumMaps());
226 for (
size_t r = 0; r < mapExtractor->NumMaps(); ++r)
227 vv_.push_back(this->ExtractVector(v, r, mapExtractor->getThyraMode()));
232 for (
size_t r = 0; r <
vv_.size(); ++r)
233 vv_[r] = Teuchos::null;
234 map_ = Teuchos::null;
278 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
296 ret->setMultiVector(r,subvec,this->
getBlockedMap()->getThyraMode());
310 if(
map_->getNumMaps() == 1) {
311 return vv_[0]->getData(j);
314 return Teuchos::null;
319 if(
map_->getNumMaps() == 1) {
320 return vv_[0]->getDataNonConst(j);
323 return Teuchos::null;
347 virtual void scale(
const Scalar &alpha) {
349 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
359 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
372 if(bA != Teuchos::null) {
377 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
394 if(blmv.
is_null() ==
true && brmv.is_null() ==
false) {
397 lmv->update(alpha, *(brmv->getMultiVector(0)), beta);
399 lmv->update(alpha, *rmv, beta);
414 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
433 if(bA != Teuchos::null && bB != Teuchos::null) {
441 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
443 getMultiVector(r)->update(alpha, *(bA->getMultiVector(r)), beta, *(bB->getMultiVector(r)), gamma);
457 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
461 norms[c] += temp_norms[c];
475 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
479 results[c] += temp_norms[c] * temp_norms[c];
493 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
497 norms[c] = std::max(norms[c],temp_norms[c]);
531 for(
size_t m = 0; m < bmap->getNumMaps(); m++) {
533 bA->getMultiVector(m, bmap->getThyraMode())->
getVector(0);
535 bB->getMultiVector(m, bmap->getThyraMode());
539 thisPart->elementWiseMultiply(scalarAB,*partA,*partB,scalarThis);
555 throw Xpetra::Exceptions::RuntimeError(
"BlockedMultiVector::getLocalLength: routine not implemented. It has no value as one must iterate on the partial vectors.");
562 return map_->getFullMap()->getGlobalNumElements();
572 return std::string(
"BlockedMultiVector");
578 for(
size_t r = 0; r <
map_->getNumMaps(); r++)
593 std::vector<Teuchos::RCP<const Map> > subMaps (1, map);
602 for(
size_t r = 0; r < map_->getNumMaps(); r++)
603 getMultiVector(r)->replaceMap(bmap->getMap(r,map_->getThyraMode()));
633 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
639 virtual void randomize(
bool bUseXpetraImplementation =
false) {
640 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
651 #ifdef HAVE_XPETRA_KOKKOS_REFACTOR 652 typedef typename Kokkos::Details::ArithTraits<Scalar>::val_type impl_scalar_type;
653 typedef Kokkos::DualView<impl_scalar_type**, Kokkos::LayoutStride,
654 typename node_type::execution_space,
655 Kokkos::MemoryUnmanaged> dual_view_type;
656 typedef typename dual_view_type::host_mirror_space host_execution_space;
657 typedef typename dual_view_type::t_dev::execution_space dev_execution_space;
664 template<
class TargetDeviceType>
665 typename Kokkos::Impl::if_c<
666 Kokkos::Impl::is_same<
667 typename dev_execution_space::memory_space,
668 typename TargetDeviceType::memory_space>::value,
669 typename dual_view_type::t_dev_um,
670 typename dual_view_type::t_host_um>::type
671 getLocalView ()
const {
672 if(Kokkos::Impl::is_same<
673 typename host_execution_space::memory_space,
674 typename TargetDeviceType::memory_space
676 return getHostLocalView();
678 return getDeviceLocalView();
682 virtual typename dual_view_type::t_host_um getHostLocalView ()
const {
683 typename dual_view_type::t_host_um test;
686 virtual typename dual_view_type::t_dev_um getDeviceLocalView()
const {
687 typename dual_view_type::t_dev_um test;
704 TEUCHOS_TEST_FOR_EXCEPTION(r >
map_->getNumMaps(), std::out_of_range,
"Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " <<
map_->getNumMaps() <<
" partial blocks.");
710 XPETRA_MONITOR(
"BlockedMultiVector::getMultiVector(r,bThyraMode)");
711 TEUCHOS_TEST_FOR_EXCEPTION(r >
map_->getNumMaps(), std::out_of_range,
"Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " <<
map_->getNumMaps() <<
" partial blocks.");
723 XPETRA_TEST_FOR_EXCEPTION(r >=
map_->getNumMaps(), std::out_of_range,
"Error, r = " << r <<
" is too big. The BlockedMultiVector only contains " <<
map_->getNumMaps() <<
" partial blocks.");
737 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
764 if(bRhs == Teuchos::null)
768 vv_ = std::vector<Teuchos::RCP<MultiVector> >(
map_->getNumMaps());
769 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
779 if(bsrc.is_null() ==
true && bvv.is_null() ==
true) {
781 }
else if (bsrc.is_null() ==
true && bvv.is_null() ==
false) {
784 }
else if (bsrc.is_null() ==
false && bvv.is_null() ==
true) {
787 if(bsrc->getBlockedMap()->getNumMaps() > 1) {
788 throw Xpetra::Exceptions::RuntimeError(
"BlockedMultiVector::assign: source vector is of type BlockedMultiVector (with more than 1 blocks) and target is a MultiVector.");
812 XPETRA_TEST_FOR_EXCEPTION(block >=
map_->getNumMaps(), std::out_of_range,
"ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " <<
map_->getNumMaps() <<
" partial blocks.");
814 "ExtractVector: map_->getmap(" << block <<
",false) is null");
816 if(bfull.is_null() ==
true) {
826 rcpNonConstFull->replaceMap(
map_->getImporter(block)->getSourceMap());
829 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
830 if(bThyraMode ==
true)
831 vv->replaceMap(
map_->getMap(block,
true));
832 rcpNonConstFull->replaceMap(oldThyMapFull);
838 "ExtractVector: Number of blocks in map extractor is " <<
map_->getNumMaps() <<
" but should be " << bfull->getBlockedMap()->getNumMaps() <<
" (number of blocks in BlockedMultiVector)");
839 return bfull->getMultiVector(block,bThyraMode);
843 XPETRA_TEST_FOR_EXCEPTION(block >=
map_->getNumMaps(), std::out_of_range,
"ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " <<
map_->getNumMaps() <<
" partial blocks.");
845 "ExtractVector: map_->getmap(" << block <<
",false) is null");
847 if(bfull.is_null() ==
true) {
856 full->replaceMap(
map_->getImporter(block)->getSourceMap());
859 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
860 if(bThyraMode ==
true)
861 vv->replaceMap(
map_->getMap(block,
true));
862 full->replaceMap(oldThyMapFull);
868 "ExtractVector: Number of blocks in map extractor is " <<
map_->getNumMaps() <<
" but should be " << bfull->getBlockedMap()->getNumMaps() <<
" (number of blocks in BlockedMultiVector)");
869 return bfull->getMultiVector(block,bThyraMode);
873 XPETRA_TEST_FOR_EXCEPTION(block >=
map_->getNumMaps(), std::out_of_range,
"ExtractVector: Error, block = " << block <<
" is too big. The BlockedMultiVector only contains " <<
map_->getNumMaps() <<
" partial blocks.");
875 "ExtractVector: maps_[" << block <<
"] is null");
880 XPETRA_TEST_FOR_EXCEPTION(block >=
map_->getNumMaps(), std::out_of_range,
"ExtractVector: Error, block = " << block <<
" is too big. The MapExtractor only contains " <<
map_->getNumMaps() <<
" partial blocks.");
882 "ExtractVector: map_->getmap(" << block <<
",false) is null");
884 "MapExtractor::InsertVector: InsertVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
905 rcpNonConstPartial->replaceMap(
map_->getMap(block,
false));
913 rcpNonConstPartial->replaceMap(oldThyMapPartial);
925 "InsertVector: maps_[" << block <<
"] is null");
926 full->setMultiVector(block, partial, bThyraMode);
935 "InsertVector: maps_[" << block <<
"] is null");
936 bfull->setMultiVector(block, partial, bThyraMode);
942 std::vector<Teuchos::RCP<MultiVector> >
vv_;
948 #define XPETRA_BLOCKEDMULTIVECTOR_SHORT 949 #endif // XPETRA_BLOCKEDMULTIVECTOR_HPP virtual void doImport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &source, const Export &exporter, CombineMode CM)
Import (using an Exporter).
void ExtractVector(RCP< const MultiVector > &full, size_t block, RCP< MultiVector > &partial) const
virtual void setSeed(unsigned int seed)
Set seed for Random function.
virtual std::string description() const
A simple one-line description of this object.
LocalOrdinal local_ordinal_type
Teuchos::RCP< MultiVector > getMultiVector(size_t r, bool bThyraMode) const
return partial multivector associated with block row r
virtual void norm2(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const
Teuchos::RCP< const Xpetra::BlockedMap< LocalOrdinal, GlobalOrdinal, Node > > getBlockedMap() const
Access function for the underlying Map this DistObject was constructed with.
virtual void norm1(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const
Compute 1-norm of each vector in multi-vector.
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Print the object with the given verbosity level to a FancyOStream.
virtual void doExport(const DistObject< Packet, LocalOrdinal, GlobalOrdinal, Node > &source, const Export< LocalOrdinal, GlobalOrdinal, Node > &exporter, CombineMode CM)=0
Export data into this object using an Export object ("forward mode").
virtual void dot(const MultiVector &A, const Teuchos::ArrayView< Scalar > &dots) const
Compute dot product of each corresponding pair of vectors, dots[i] = this[i].dot(A[i]).
GlobalOrdinal global_ordinal_type
std::vector< Teuchos::RCP< MultiVector > > vv_
array containing RCPs of the partial vectors
virtual void multiply(Teuchos::ETransp transA, Teuchos::ETransp transB, const Scalar &alpha, const MultiVector &A, const MultiVector &B, const Scalar &beta)
Matrix-matrix multiplication: this = beta*this + alpha*op(A)*op(B).
#define TEUCHOS_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
virtual void Xpetra_randomize()
Set multi-vector values to random numbers. XPetra implementation.
BlockedMultiVector(const Teuchos::RCP< const BlockedMap > &map, size_t NumVectors, bool zeroOut=true)
Constructor.
virtual void doExport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &dest, const Import &importer, CombineMode CM)
Export.
virtual global_size_t getGlobalLength() const
Global number of rows in the multivector.
virtual void doImport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &source, const Import &importer, CombineMode CM)
Import.
virtual void update(const Scalar &alpha, const MultiVector &A, const Scalar &beta, const MultiVector &B, const Scalar &gamma)
Update multi-vector with scaled values of A and B, this = gamma*this + alpha*A + beta*B.
Exception throws to report errors in the internal logical of the program.
virtual void reciprocal(const MultiVector &A)
Put element-wise reciprocal values of input Multi-vector in target, this(i,j) = 1/A(i,j).
virtual void scale(const Scalar &alpha)
Scale the current values of a multi-vector, this = alpha*this.
void InsertVector(RCP< MultiVector > partial, size_t block, RCP< MultiVector > full, bool bThyraMode=false) const
void InsertVector(const MultiVector &partial, size_t block, MultiVector &full, bool bThyraMode=false) const
Teuchos::RCP< MultiVector > getMultiVector(size_t r) const
return partial multivector associated with block row r
virtual void replaceMap(const RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &map)=0
virtual void update(const Scalar &alpha, const MultiVector &A, const Scalar &beta)
Update multi-vector values with scaled values of A, this = beta*this + alpha*A.
virtual void sumIntoLocalValue(LocalOrdinal myRow, size_t vectorIndex, const Scalar &value)
Add value to existing value, using local (row) index.
virtual Teuchos::ArrayRCP< Scalar > getDataNonConst(size_t j)
View of the local values in a particular vector of this multivector.
Teuchos::RCP< const Map > getMap() const
Access function for the underlying Map this DistObject was constructed with.
BlockedMultiVector(Teuchos::RCP< const Xpetra::BlockedMap< LocalOrdinal, GlobalOrdinal, Node > > bmap, Teuchos::RCP< MultiVector > v)
RCP< MultiVector > ExtractVector(RCP< const MultiVector > &full, size_t block, bool bThyraMode=false) const
virtual void doImport(const DistObject< Packet, LocalOrdinal, GlobalOrdinal, Node > &source, const Import< LocalOrdinal, GlobalOrdinal, Node > &importer, CombineMode CM)=0
Import data into this object using an Import object ("forward mode").
virtual ~BlockedMultiVector()
Destructor.
size_t numVectors_
number of vectors (columns in multi vector)
void ExtractVector(RCP< MultiVector > &full, size_t block, RCP< MultiVector > &partial) const
TEUCHOS_DEPRECATED RCP< T > rcp(T *p, Dealloc_T dealloc, bool owns_mem)
GlobalOrdinal global_ordinal_type
virtual void abs(const MultiVector &A)
Put element-wise absolute values of input Multi-vector in target: A = abs(this).
LocalOrdinal local_ordinal_type
virtual void Xpetra_randomize()
Set multi-vector values to random numbers. XPetra implementation.
RCP< MultiVector > ExtractVector(RCP< MultiVector > &full, size_t block, bool bThyraMode=false) const
virtual void elementWiseMultiply(Scalar scalarAB, const Xpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > &A, const MultiVector &B, Scalar scalarThis)
Element-wise multiply of a Vector A with a MultiVector B.
virtual void replaceMap(const RCP< const Map > &map)
virtual Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > getMap() const =0
The Map describing the parallel distribution of this object.
virtual size_t getLocalLength() const
Local number of rows on the calling process.
virtual void assign(const MultiVector &rhs)
Implementation of the assignment operator (operator=); does a deep copy.
virtual void normInf(const Teuchos::ArrayView< typename Teuchos::ScalarTraits< Scalar >::magnitudeType > &norms) const
Compute Inf-norm of each vector in multi-vector.
virtual Teuchos::RCP< const Xpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getVector(size_t j) const
Return a Vector which is a const view of column j.
void setMultiVector(size_t r, Teuchos::RCP< const MultiVector > v, bool bThyraMode)
set partial multivector associated with block row r
void InsertVector(RCP< const MultiVector > partial, size_t block, RCP< MultiVector > full, bool bThyraMode=false) const
BlockedMultiVector(Teuchos::RCP< const Xpetra::MapExtractor< Scalar, LocalOrdinal, GlobalOrdinal, Node > > mapExtractor, Teuchos::RCP< MultiVector > v)
virtual void randomize(bool bUseXpetraImplementation=false)
size_t global_size_t
Global size_t object.
Teuchos::RCP< MultiVector > Merge() const
merge BlockedMultiVector blocks to a single MultiVector
static const EVerbosityLevel verbLevel_default
virtual size_t getNumVectors() const
Number of columns in the multivector.
BlockedMultiVector(Teuchos::RCP< const Xpetra::BlockedMap< LocalOrdinal, GlobalOrdinal, Node > > bmap, Teuchos::RCP< const MultiVector > v)
virtual void scale(Teuchos::ArrayView< const Scalar > alpha)
Scale the current values of a multi-vector, this[j] = alpha[j]*this[j].
void ExtractVector(const MultiVector &full, size_t block, MultiVector &partial) const
#define XPETRA_TEST_FOR_EXCEPTION(throw_exception_test, Exception, msg)
Teuchos::RCP< const BlockedMap > map_
blocked map containing the sub block maps (either thyra or xpetra mode)
virtual void replaceLocalValue(LocalOrdinal myRow, size_t vectorIndex, const Scalar &value)
Replace value, using local (row) index.
virtual void putScalar(const Scalar &value)
Set all values in the multivector with the given value.
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.
CombineMode
Xpetra::Combine Mode enumerable type.
virtual void doExport(const DistObject< Scalar, LocalOrdinal, GlobalOrdinal, Node > &dest, const Export &exporter, CombineMode CM)
Export (using an Importer).
#define XPETRA_MONITOR(funcName)
virtual void meanValue(const Teuchos::ArrayView< Scalar > &means) const
Compute mean (average) value of each vector in multi-vector. The outcome of this routine is undefined...
BlockedMultiVector< Scalar, LocalOrdinal, GlobalOrdinal, Node > & operator=(const MultiVector &rhs)
Assignment operator: Does a deep copy.
virtual Teuchos::ArrayRCP< const Scalar > getData(size_t j) const
Const view of the local values in a particular vector of this multivector.
virtual void replaceGlobalValue(GlobalOrdinal globalRow, size_t vectorIndex, const Scalar &value)
Replace value, using global (row) index.
BlockedMultiVector(Teuchos::RCP< const Xpetra::MapExtractor< Scalar, LocalOrdinal, GlobalOrdinal, Node > > mapExtractor, Teuchos::RCP< const MultiVector > v)
virtual void sumIntoGlobalValue(GlobalOrdinal globalRow, size_t vectorIndex, const Scalar &value)
Add value to existing value, using global (row) index.
virtual Teuchos::RCP< Xpetra::Vector< Scalar, LocalOrdinal, GlobalOrdinal, Node > > getVectorNonConst(size_t j)
Return a Vector which is a nonconst view of column j.