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()));
241 vv_.resize(vin.size());
242 for(
size_t i=0; i<
vv_.size(); i++)
248 for (
size_t r = 0; r <
vv_.size(); ++r)
249 vv_[r] = Teuchos::null;
250 map_ = Teuchos::null;
294 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
312 ret->setMultiVector(r,subvec,this->
getBlockedMap()->getThyraMode());
326 if(
map_->getNumMaps() == 1) {
327 return vv_[0]->getData(j);
335 if(
map_->getNumMaps() == 1) {
336 return vv_[0]->getDataNonConst(j);
363 virtual void scale(
const Scalar &alpha) {
365 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
375 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
388 if(bA != Teuchos::null) {
393 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
413 lmv->update(alpha, *(brmv->getMultiVector(0)), beta);
415 lmv->update(alpha, *rmv, beta);
430 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
449 if(bA != Teuchos::null && bB != Teuchos::null) {
457 for(
size_t r = 0; r <
map_->getNumMaps(); r++) {
459 getMultiVector(r)->update(alpha, *(bA->getMultiVector(r)), beta, *(bB->getMultiVector(r)), gamma);
473 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
477 norms[c] += temp_norms[c];
491 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
495 results[c] += temp_norms[c] * temp_norms[c];
509 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
513 norms[c] = std::max(norms[c],temp_norms[c]);
547 for(
size_t m = 0; m < bmap->getNumMaps(); m++) {
549 bA->getMultiVector(m, bmap->getThyraMode())->
getVector(0);
551 bB->getMultiVector(m, bmap->getThyraMode());
555 thisPart->elementWiseMultiply(scalarAB,*partA,*partB,scalarThis);
572 return map_->getFullMap()->getNodeNumElements();
578 return map_->getFullMap()->getGlobalNumElements();
584 if(!Vb)
return false;
585 for (
size_t r = 0; r <
map_->getNumMaps(); ++r) {
588 if((a==Teuchos::null && b != Teuchos::null) ||
589 (a!=Teuchos::null && b == Teuchos::null))
591 if(a!=Teuchos::null && b !=Teuchos::null && !a->isSameSize(*b))
604 return std::string(
"BlockedMultiVector");
610 for(
size_t r = 0; r <
map_->getNumMaps(); r++)
625 std::vector<Teuchos::RCP<const Map> > subMaps (1, map);
634 for(
size_t r = 0; r <
map_->getNumMaps(); r++)
665 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
671 virtual void randomize(
bool bUseXpetraImplementation =
false) {
672 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
683 #ifdef HAVE_XPETRA_KOKKOS_REFACTOR
684 typedef typename Kokkos::Details::ArithTraits<Scalar>::val_type impl_scalar_type;
685 typedef Kokkos::DualView<impl_scalar_type**, Kokkos::LayoutStride,
686 typename node_type::execution_space,
687 Kokkos::MemoryUnmanaged> dual_view_type;
688 typedef typename dual_view_type::host_mirror_space host_execution_space;
689 typedef typename dual_view_type::t_dev::execution_space dev_execution_space;
696 template<
class TargetDeviceType>
697 typename Kokkos::Impl::if_c<
698 Kokkos::Impl::is_same<
699 typename dev_execution_space::memory_space,
700 typename TargetDeviceType::memory_space>::value,
701 typename dual_view_type::t_dev_um,
702 typename dual_view_type::t_host_um>::type
703 getLocalView ()
const {
704 if(Kokkos::Impl::is_same<
705 typename host_execution_space::memory_space,
706 typename TargetDeviceType::memory_space
708 return getHostLocalView();
710 return getDeviceLocalView();
714 virtual typename dual_view_type::t_host_um getHostLocalView ()
const {
715 typename dual_view_type::t_host_um test;
718 virtual typename dual_view_type::t_dev_um getDeviceLocalView()
const {
719 typename dual_view_type::t_dev_um test;
736 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.");
742 XPETRA_MONITOR(
"BlockedMultiVector::getMultiVector(r,bThyraMode)");
743 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.");
755 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.");
769 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
796 if(bRhs == Teuchos::null)
800 vv_ = std::vector<Teuchos::RCP<MultiVector> >(
map_->getNumMaps());
801 for(
size_t r = 0; r <
map_->getNumMaps(); ++r) {
811 if(bsrc.is_null() ==
true && bvv.
is_null() ==
true) {
813 }
else if (bsrc.is_null() ==
true && bvv.
is_null() ==
false) {
816 }
else if (bsrc.is_null() ==
false && bvv.
is_null() ==
true) {
819 if(bsrc->getBlockedMap()->getNumMaps() > 1) {
820 throw Xpetra::Exceptions::RuntimeError(
"BlockedMultiVector::assign: source vector is of type BlockedMultiVector (with more than 1 blocks) and target is a MultiVector.");
844 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.");
846 "ExtractVector: map_->getmap(" << block <<
",false) is null");
848 if(bfull.is_null() ==
true) {
857 RCP<MultiVector> rcpNonConstFull = Teuchos::rcp_const_cast<MultiVector>(full);
858 rcpNonConstFull->replaceMap(
map_->getImporter(block)->getSourceMap());
861 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
862 if(bThyraMode ==
true)
863 vv->replaceMap(
map_->getMap(block,
true));
864 rcpNonConstFull->replaceMap(oldThyMapFull);
870 "ExtractVector: Number of blocks in map extractor is " <<
map_->getNumMaps() <<
" but should be " << bfull->getBlockedMap()->getNumMaps() <<
" (number of blocks in BlockedMultiVector)");
871 return bfull->getMultiVector(block,bThyraMode);
875 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.");
877 "ExtractVector: map_->getmap(" << block <<
",false) is null");
879 if(bfull.is_null() ==
true) {
888 full->replaceMap(
map_->getImporter(block)->getSourceMap());
891 "MapExtractor::ExtractVector: ExtractVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
892 if(bThyraMode ==
true)
893 vv->replaceMap(
map_->getMap(block,
true));
894 full->replaceMap(oldThyMapFull);
900 "ExtractVector: Number of blocks in map extractor is " <<
map_->getNumMaps() <<
" but should be " << bfull->getBlockedMap()->getNumMaps() <<
" (number of blocks in BlockedMultiVector)");
901 return bfull->getMultiVector(block,bThyraMode);
905 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.");
907 "ExtractVector: maps_[" << block <<
"] is null");
912 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.");
914 "ExtractVector: map_->getmap(" << block <<
",false) is null");
916 "MapExtractor::InsertVector: InsertVector in Thyra-style numbering only possible if MapExtractor has been created using Thyra-style numbered submaps.");
926 RCP<MultiVector> rcpNonConstPartial = Teuchos::rcp_const_cast<MultiVector>(rcpPartial);
937 rcpNonConstPartial->replaceMap(
map_->getMap(block,
false));
945 rcpNonConstPartial->replaceMap(oldThyMapPartial);
957 "InsertVector: maps_[" << block <<
"] is null");
958 full->setMultiVector(block, partial, bThyraMode);
967 "InsertVector: maps_[" << block <<
"] is null");
968 bfull->setMultiVector(block, partial, bThyraMode);
974 std::vector<Teuchos::RCP<MultiVector> >
vv_;
980 #define XPETRA_BLOCKEDMULTIVECTOR_SHORT
981 #endif // XPETRA_BLOCKEDMULTIVECTOR_HPP