53 #ifndef AMESOS2_EPETRA_MULTIVEC_ADAPTER_DEF_HPP
54 #define AMESOS2_EPETRA_MULTIVEC_ADAPTER_DEF_HPP
56 #include <Teuchos_as.hpp>
58 #include <Epetra_SerialComm.h>
60 #include <Epetra_MpiComm.h>
62 #include <Epetra_LocalMap.h>
63 #include <Epetra_Import.h>
64 #include <Epetra_Export.h>
73 , mv_map_(adapter.mv_map_)
79 mv_map_ = Teuchos::rcpFromRef(mv_->Map());
83 Teuchos::RCP<Epetra_MultiVector>
86 Teuchos::RCP<Epetra_MultiVector> Y (
new Epetra_MultiVector(*mv_map_, mv_->NumVectors(),
false));
94 return !mv_->DistributedGlobal();
99 return mv_->DistributedGlobal();
103 const Teuchos::RCP<const Teuchos::Comm<int> >
112 return Teuchos::as<size_t>(mv_->MyLength());
118 return Teuchos::as<size_t>(mv_->NumVectors());
125 return Teuchos::as<global_size_t>(mv_->GlobalLength());
131 return Teuchos::as<size_t>(mv_->NumVectors());
137 return Teuchos::as<size_t>(mv_->Stride());
143 return mv_->ConstantStride();
147 Teuchos::RCP<const Tpetra::Vector<MultiVecAdapter<Epetra_MultiVector>::scalar_t,
155 using Teuchos::ArrayRCP;
156 using Tpetra::MultiVector;
159 typedef local_ordinal_t lot;
160 typedef global_ordinal_t got;
163 RCP<MultiVector<st,lot,got,nt> > vector = rcp(
new MultiVector<st,lot,got,nt>(this->getMap(),1));
166 ArrayRCP<st> it = vector->getDataNonConst(0);
167 double* vector_data = mv_->operator[](Teuchos::as<int>(j));
168 Tpetra::global_size_t
size = vector->getGlobalLength();
170 for( Tpetra::global_size_t i = 0; i <
size; ++i ){
171 *it = vector_data[i];
174 return vector->getVector(j);
179 Teuchos::RCP<Tpetra::Vector<MultiVecAdapter<Epetra_MultiVector>::scalar_t,
187 using Teuchos::ArrayRCP;
188 using Tpetra::MultiVector;
191 typedef local_ordinal_t lot;
192 typedef global_ordinal_t got;
195 RCP<MultiVector<st,lot,got,nt> > vector = rcp(
new MultiVector<st,lot,got,nt>(this->getMap(),1));
198 ArrayRCP<st> it = vector->getDataNonConst(0);
199 double* vector_data = mv_->operator[](Teuchos::as<int>(j));
200 Tpetra::global_size_t
size = vector->getGlobalLength();
202 for( Tpetra::global_size_t i = 0; i <
size; ++i ){
203 *it = vector_data[i];
206 return vector->getVectorNonConst(j);
212 TEUCHOS_TEST_FOR_EXCEPTION( this->getGlobalNumVectors() != 1,
213 std::invalid_argument,
214 "Amesos2_EpetraMultiVectorAdapter: getMVPointer_impl should only be called for case with a single vector and single MPI process" );
216 double* vector_data = mv_->operator[](Teuchos::as<int>(0));
230 using Teuchos::rcpFromPtr;
233 const size_t num_vecs = getGlobalNumVectors();
235 #ifdef HAVE_AMESOS2_DEBUG
236 const size_t requested_vector_length = distribution_map->getLocalNumElements();
237 TEUCHOS_TEST_FOR_EXCEPTION( lda < requested_vector_length,
238 std::invalid_argument,
239 "Given stride is not large enough for local vector length" );
240 TEUCHOS_TEST_FOR_EXCEPTION( as<size_t>(av.size()) < (num_vecs-1) * lda + requested_vector_length,
241 std::invalid_argument,
242 "MultiVector storage not large enough given leading dimension "
243 "and number of vectors" );
247 if ( num_vecs == 1 && mv_->Comm().MyPID() == 0 && mv_->Comm().NumProc() == 1 ) {
248 mv_->ExtractCopy(av.getRawPtr(), lda);
251 Epetra_Map e_dist_map
255 node_t>(*distribution_map);
257 multivec_t redist_mv(e_dist_map, as<int>(num_vecs));
258 const Epetra_Import importer(e_dist_map, *mv_map_);
259 redist_mv.Import(*mv_, importer, Insert);
262 redist_mv.ExtractCopy(av.getRawPtr(), lda);
268 Kokkos::View<scalar_t**, Kokkos::LayoutLeft, Kokkos::HostSpace> & host_view,
276 using Teuchos::rcpFromPtr;
279 const size_t num_vecs = getGlobalNumVectors();
281 #ifdef HAVE_AMESOS2_DEBUG
282 const size_t requested_vector_length = distribution_map->getLocalNumElements();
283 TEUCHOS_TEST_FOR_EXCEPTION( lda < requested_vector_length,
284 std::invalid_argument,
285 "Given stride is not large enough for local vector length" );
289 host_view = Kokkos::View<scalar_t**, Kokkos::LayoutLeft, Kokkos::HostSpace>(
290 Kokkos::ViewAllocateWithoutInitializing(
"get1dCopy_kokkos_view"),
291 distribution_map->getLocalNumElements(), num_vecs);
294 if ( num_vecs == 1 && this->mv_->Comm().MyPID() == 0 && this->mv_->Comm().NumProc() == 1 ) {
295 mv_->ExtractCopy(host_view.data(), lda);
298 Epetra_Map e_dist_map
302 node_t>(*distribution_map);
304 multivec_t redist_mv(e_dist_map, as<int>(num_vecs));
305 const Epetra_Import importer(e_dist_map, *mv_map_);
306 redist_mv.Import(*mv_, importer, Insert);
312 redist_mv.ExtractCopy(host_view.data(), lda);
316 Teuchos::ArrayRCP<MultiVecAdapter<Epetra_MultiVector>::scalar_t>
368 return Teuchos::null;
382 using Teuchos::rcpFromPtr;
385 const size_t num_vecs = getGlobalNumVectors();
387 double* data_ptr =
const_cast<double*
>(new_data.getRawPtr());
390 if ( num_vecs == 1 && mv_->Comm().MyPID() == 0 && mv_->Comm().NumProc() == 1 ) {
396 auto vector = mv_->Pointers();
397 for (
size_t i = 0; i < lda; ++i ) {
398 vector[0][i] = data_ptr[i];
402 const Epetra_BlockMap e_source_map
403 = *Util::tpetra_map_to_epetra_map<local_ordinal_t,global_ordinal_t,global_size_t,node_t>(*source_map);
404 const multivec_t source_mv(Copy, e_source_map, data_ptr, as<int>(lda), as<int>(num_vecs));
405 const Epetra_Import importer(*mv_map_, e_source_map);
407 mv_->Import(source_mv, importer, Insert);
413 Kokkos::View<scalar_t**, Kokkos::LayoutLeft, Kokkos::HostSpace> & host_new_data,
421 using Teuchos::rcpFromPtr;
424 const size_t num_vecs = getGlobalNumVectors();
426 double* data_ptr = host_new_data.data();
429 if ( num_vecs == 1 && mv_->Comm().MyPID() == 0 && mv_->Comm().NumProc() == 1 ) {
430 auto vector = mv_->Pointers();
431 for (
size_t i = 0; i < lda; ++i ) {
432 vector[0][i] = data_ptr[i];
436 const Epetra_BlockMap e_source_map
437 = *Util::tpetra_map_to_epetra_map<local_ordinal_t,global_ordinal_t,global_size_t,node_t>(*source_map);
438 const multivec_t source_mv(Copy, e_source_map, data_ptr, as<int>(lda), as<int>(num_vecs));
439 const Epetra_Import importer(*mv_map_, e_source_map);
441 mv_->Import(source_mv, importer, Insert);
447 std::ostringstream oss;
448 oss <<
"Amesos2 adapter wrapping: Epetra_MultiVector";
454 Teuchos::FancyOStream& os,
455 const Teuchos::EVerbosityLevel verbLevel)
const
458 if(verbLevel != Teuchos::VERB_NONE)
460 os <<
"TODO: implement! ";
465 Teuchos::RCP<const Tpetra::Map<MultiVecAdapter<Epetra_MultiVector>::local_ordinal_t,
470 return Util::epetra_map_to_tpetra_map<local_ordinal_t,global_ordinal_t,global_size_t,node_t>(*mv_map_);
475 =
"Amesos2 adapter for Epetra_MultiVector";
480 #endif // AMESOS2_EPETRA_MULTIVEC_ADAPTER_DEF_HPP
const int size
Definition: klu2_simple.cpp:74
Utility functions for Amesos2.
const RCP< const Teuchos::Comm< int > > to_teuchos_comm(RCP< const Epetra_Comm > c)
Transform an Epetra_Comm object into a Teuchos::Comm object.
Amesos2::MultiVecAdapter specialization for the Epetra_MultiVector class.
RCP< Epetra_Map > tpetra_map_to_epetra_map(const Tpetra::Map< LO, GO, Node > &map)
Transform a Tpetra::Map object into an Epetra_Map.
Definition: Amesos2_Util.hpp:759
EDistribution
Definition: Amesos2_TypeDecl.hpp:123
A templated MultiVector class adapter for Amesos2.
Definition: Amesos2_MultiVecAdapter_decl.hpp:176