Tpetra parallel linear algebra Version of the Day
Loading...
Searching...
No Matches
Tpetra_Import_decl.hpp
1// @HEADER
2// *****************************************************************************
3// Tpetra: Templated Linear Algebra Services Package
4//
5// Copyright 2008 NTESS and the Tpetra contributors.
6// SPDX-License-Identifier: BSD-3-Clause
7// *****************************************************************************
8// @HEADER
9
10#ifndef TPETRA_IMPORT_DECL_HPP
11#define TPETRA_IMPORT_DECL_HPP
12
13#include "Tpetra_Details_Transfer.hpp"
14#include "Tpetra_Import_fwd.hpp"
15#include "Tpetra_Export_fwd.hpp"
16
17namespace Tpetra {
18
74template <class LocalOrdinal,
75 class GlobalOrdinal,
76 class Node>
77class Import : public ::Tpetra::Details::Transfer<LocalOrdinal, GlobalOrdinal, Node> {
78 private:
79 friend class Export<LocalOrdinal, GlobalOrdinal, Node>;
80 using base_type =
82
83 public:
86
88
89
97 Import(const Teuchos::RCP<const map_type>& source,
98 const Teuchos::RCP<const map_type>& target);
99
110 Import(const Teuchos::RCP<const map_type>& source,
111 const Teuchos::RCP<const map_type>& target,
112 const Teuchos::RCP<Teuchos::FancyOStream>& out);
113
126 Import(const Teuchos::RCP<const map_type>& source,
127 const Teuchos::RCP<const map_type>& target,
128 const Teuchos::RCP<Teuchos::ParameterList>& plist);
129
145 Import(const Teuchos::RCP<const map_type>& source,
146 const Teuchos::RCP<const map_type>& target,
147 const Teuchos::RCP<Teuchos::FancyOStream>& out,
148 const Teuchos::RCP<Teuchos::ParameterList>& plist);
149
163 Import(const Teuchos::RCP<const map_type>& source,
164 const Teuchos::RCP<const map_type>& target,
165 Teuchos::Array<int>& remotePIDs,
166 const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::rcp(new Teuchos::ParameterList));
167
173
181
219 const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::null,
220 const Teuchos::RCP<Teuchos::FancyOStream>& out = Teuchos::null);
221
223 Import(const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>>& source,
224 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>>& target,
225 const Teuchos::ArrayView<int>& remotePIDs,
226 const Teuchos::ArrayView<const LocalOrdinal>& userExportLIDs,
227 const Teuchos::ArrayView<const int>& userExportPIDs,
228 const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::null,
229 const Teuchos::RCP<Teuchos::FancyOStream>& out = Teuchos::null);
230
234
236 virtual ~Import() = default;
237
239
241
250 void
251 findUnionTargetGIDs(Teuchos::Array<GlobalOrdinal>& unionTgtGIDs,
252 Teuchos::Array<std::pair<int, GlobalOrdinal>>& remotePGIDs,
253 typename Teuchos::Array<GlobalOrdinal>::size_type& numSameGIDs,
254 typename Teuchos::Array<GlobalOrdinal>::size_type& numPermuteGIDs,
255 typename Teuchos::Array<GlobalOrdinal>::size_type& numRemoteGIDs,
256 const Teuchos::ArrayView<const GlobalOrdinal>& sameGIDs1,
257 const Teuchos::ArrayView<const GlobalOrdinal>& sameGIDs2,
258 Teuchos::Array<GlobalOrdinal>& permuteGIDs1,
259 Teuchos::Array<GlobalOrdinal>& permuteGIDs2,
260 Teuchos::Array<GlobalOrdinal>& remoteGIDs1,
261 Teuchos::Array<GlobalOrdinal>& remoteGIDs2,
262 Teuchos::Array<int>& remotePIDs1,
263 Teuchos::Array<int>& remotePIDs2) const;
264
298 Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>>
300
317 Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>>
318 setUnion() const;
319
324 Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>>
325 createRemoteOnlyImport(const Teuchos::RCP<const map_type>& remoteTarget) const;
326
328
330
353 virtual void
354 describe(Teuchos::FancyOStream& out,
355 const Teuchos::EVerbosityLevel verbLevel =
356 Teuchos::Describable::verbLevel_default) const;
357
373 virtual void print(std::ostream& os) const;
374
376 private:
378
379
396 void
397 init(const Teuchos::RCP<const map_type>& source,
398 const Teuchos::RCP<const map_type>& target,
399 bool useRemotePIDs,
400 Teuchos::Array<int>& remotePIDs,
401 const Teuchos::RCP<Teuchos::ParameterList>& plist);
402
434 void setupSamePermuteRemote(Teuchos::Array<GlobalOrdinal>& remoteGIDs);
435
464 void
465 setupExport(Teuchos::Array<GlobalOrdinal>& remoteGIDs,
466 bool useRemotePIDs, Teuchos::Array<int>& remotePIDs,
467 const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::null);
469
477 Import(const Teuchos::RCP<const map_type>& source,
478 const Teuchos::RCP<const map_type>& target,
479 const size_t numSameID,
480 Teuchos::Array<LocalOrdinal>& permuteToLIDs,
481 Teuchos::Array<LocalOrdinal>& permuteFromLIDs,
482 Teuchos::Array<LocalOrdinal>& remoteLIDs,
483 Teuchos::Array<LocalOrdinal>& exportLIDs,
484 Teuchos::Array<int>& exportPIDs,
486 const Teuchos::RCP<Teuchos::FancyOStream>& out = Teuchos::null,
487 const Teuchos::RCP<Teuchos::ParameterList>& plist = Teuchos::null);
488
489}; // class Import
490
500template <class LocalOrdinal, class GlobalOrdinal, class Node>
501Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>>
503 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>>& tgt) {
504 if (src == tgt) {
505 return Teuchos::null;
506 }
507#ifdef HAVE_TPETRA_DEBUG
509 src == Teuchos::null || tgt == Teuchos::null, std::runtime_error,
510 "Tpetra::createImport: Neither source nor target Map may be null.");
511#endif // HAVE_TPETRA_DEBUG
513 return Teuchos::rcp(new import_type(src, tgt));
514}
515
526template <class LocalOrdinal, class GlobalOrdinal, class Node>
527Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node>>
529 const Teuchos::RCP<const Map<LocalOrdinal, GlobalOrdinal, Node>>& tgt,
530 const Teuchos::RCP<Teuchos::ParameterList>& plist) {
531 if (src == tgt) {
532 return Teuchos::null;
533 }
534#ifdef HAVE_TPETRA_DEBUG
536 src == Teuchos::null || tgt == Teuchos::null, std::runtime_error,
537 "Tpetra::createImport(): neither source nor target map may be null:"
538 << std::endl
539 << "source: " << src << std::endl
540 << "target: " << tgt
541 << std::endl);
542#endif // HAVE_TPETRA_DEBUG
544 return Teuchos::rcp(new import_type(src, tgt, plist));
545}
546
547} // namespace Tpetra
548
549#endif // TPETRA_IMPORT_DECL_HPP
Forward declaration of Tpetra::Export.
Forward declaration of Tpetra::Import.
Struct that holds views of the contents of a CrsMatrix.
Common base class of Import and Export.
Sets up and executes a communication plan for a Tpetra DistObject.
Communication plan for data redistribution from a (possibly) multiply-owned to a uniquely-owned distr...
Communication plan for data redistribution from a uniquely-owned to a (possibly) multiply-owned distr...
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > setUnion() const
Return the union of this Import this->getSourceMap()
virtual void describe(Teuchos::FancyOStream &out, const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const
Describe this object in a human-readable way to the given output stream.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > createImport(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &src, const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &tgt, const Teuchos::RCP< Teuchos::ParameterList > &plist)
Nonmember constructor for Import that takes a ParameterList.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > createRemoteOnlyImport(const Teuchos::RCP< const map_type > &remoteTarget) const
Returns an importer that contains only the remote entries of this.
Teuchos::RCP< const Import< LocalOrdinal, GlobalOrdinal, Node > > createImport(const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &src, const Teuchos::RCP< const Map< LocalOrdinal, GlobalOrdinal, Node > > &tgt)
Nonmember constructor for Import.
void findUnionTargetGIDs(Teuchos::Array< GlobalOrdinal > &unionTgtGIDs, Teuchos::Array< std::pair< int, GlobalOrdinal > > &remotePGIDs, typename Teuchos::Array< GlobalOrdinal >::size_type &numSameGIDs, typename Teuchos::Array< GlobalOrdinal >::size_type &numPermuteGIDs, typename Teuchos::Array< GlobalOrdinal >::size_type &numRemoteGIDs, const Teuchos::ArrayView< const GlobalOrdinal > &sameGIDs1, const Teuchos::ArrayView< const GlobalOrdinal > &sameGIDs2, Teuchos::Array< GlobalOrdinal > &permuteGIDs1, Teuchos::Array< GlobalOrdinal > &permuteGIDs2, Teuchos::Array< GlobalOrdinal > &remoteGIDs1, Teuchos::Array< GlobalOrdinal > &remoteGIDs2, Teuchos::Array< int > &remotePIDs1, Teuchos::Array< int > &remotePIDs2) const
Find the union of the target IDs from two Import objects.
virtual void print(std::ostream &os) const
Print the Import's data to the given output stream.
virtual ~Import()=default
Destructor.
Import< LocalOrdinal, GlobalOrdinal, Node > & operator=(const Import< LocalOrdinal, GlobalOrdinal, Node > &Source)=default
Assignment operator.
A parallel distribution of indices over processes.
Namespace Tpetra contains the class and methods constituting the Tpetra library.