39#include "Tpetra_Distributor.hpp"
43#include "Tpetra_Details_makeValidVerboseStream.hpp"
44#include "Teuchos_StandardParameterEntryValidators.hpp"
45#include "Teuchos_VerboseObjectParameterListHelpers.hpp"
55 const bool tpetraDistributorDebugDefault =
false;
58 Teuchos::Array<std::string>
69 Distributor (
const Teuchos::RCP<
const Teuchos::Comm<int> >& comm,
70 const Teuchos::RCP<Teuchos::FancyOStream>& ,
71 const Teuchos::RCP<Teuchos::ParameterList>&
plist)
78 Distributor (
const Teuchos::RCP<
const Teuchos::Comm<int> >& comm)
83 Distributor (
const Teuchos::RCP<
const Teuchos::Comm<int> >& comm,
84 const Teuchos::RCP<Teuchos::FancyOStream>&
out)
89 Distributor (
const Teuchos::RCP<
const Teuchos::Comm<int> >& comm,
90 const Teuchos::RCP<Teuchos::ParameterList>&
plist)
99 , reverseDistributor_ (
distributor.reverseDistributor_)
101 using Teuchos::ParameterList;
107 Teuchos::parameterList (*
rhsList);
112 using Teuchos::ParameterList;
113 using Teuchos::parameterList;
116 std::swap (plan_,
rhs.plan_);
117 std::swap (actor_,
rhs.actor_);
118 std::swap (verbose_,
rhs.verbose_);
119 std::swap (reverseDistributor_,
rhs.reverseDistributor_);
140 Distributor::getVerbose()
146 std::unique_ptr<std::string>
148 createPrefix(
const char methodName[])
const
151 plan_.getComm().getRawPtr(),
"Distributor", methodName);
158 using ::Tpetra::Details::Behavior;
159 using Teuchos::FancyOStream;
160 using Teuchos::getIntegralValue;
161 using Teuchos::includesVerbLevel;
162 using Teuchos::ParameterList;
163 using Teuchos::parameterList;
167 if (!
plist.is_null()) {
182 Teuchos::RCP<const Teuchos::ParameterList>
185 using Teuchos::Array;
186 using Teuchos::ParameterList;
187 using Teuchos::parameterList;
189 using Teuchos::setStringToIntegralParameter;
204 plist->set (
"Debug", debug,
"Whether to print copious debugging output on "
206 plist->set (
"Timer Label",
"",
"Label for Time Monitor output");
214 Teuchos::setupVerboseObjectSublist (&*
plist);
215 return Teuchos::rcp_const_cast<const ParameterList> (
plist);
220 {
return plan_.getTotalReceiveLength(); }
223 {
return plan_.getNumReceives(); }
226 {
return plan_.hasSelfMessage(); }
229 {
return plan_.getNumSends(); }
232 {
return plan_.getMaxSendLength(); }
235 {
return plan_.getProcsFrom(); }
238 {
return plan_.getLengthsFrom(); }
241 {
return plan_.getProcsTo(); }
244 {
return plan_.getLengthsTo(); }
246 Teuchos::RCP<Distributor>
248 if (reverseDistributor_.is_null () &&
create) {
249 createReverseDistributor ();
252 (reverseDistributor_.is_null () &&
create, std::logic_error,
"The reverse "
253 "Distributor is null after createReverseDistributor returned. "
254 "Please report this bug to the Tpetra developers.");
255 return reverseDistributor_;
260 Distributor::createReverseDistributor()
const
262 reverseDistributor_ = Teuchos::rcp(
new Distributor(plan_.getComm()));
263 reverseDistributor_->plan_ = *plan_.getReversePlan();
264 reverseDistributor_->verbose_ = verbose_;
280 reverseDistributor_->reverseDistributor_ = Teuchos::null;
286 actor_.doWaits(plan_);
291 if (! reverseDistributor_.is_null()) {
292 reverseDistributor_->doWaits();
297 std::ostringstream
out;
299 out <<
"\"Tpetra::Distributor\": {";
302 out <<
"Label: " << label <<
", ";
304 out <<
"How initialized: "
308 << DistributorSendTypeEnumToString (plan_.getSendType())
309 <<
", Debug: " << (verbose_ ?
"true" :
"false")
316 localDescribeToString (
const Teuchos::EVerbosityLevel
vl)
const
318 using Teuchos::toString;
319 using Teuchos::VERB_HIGH;
320 using Teuchos::VERB_EXTREME;
324 if (
vl <= Teuchos::VERB_LOW || plan_.getComm().is_null ()) {
325 return std::string ();
328 auto outStringP = Teuchos::rcp (
new std::ostringstream ());
329 auto outp = Teuchos::getFancyOStream (outStringP);
330 Teuchos::FancyOStream& out = *outp;
332 const int myRank = plan_.getComm()->getRank ();
333 const int numProcs = plan_.getComm()->getSize ();
334 out <<
"Process " << myRank <<
" of " << numProcs <<
":" << endl;
335 Teuchos::OSTab tab1 (out);
339 if (vl == VERB_HIGH || vl == VERB_EXTREME) {
340 out <<
"procsTo: " << toString (plan_.getProcsTo()) << endl;
341 out <<
"lengthsTo: " << toString (plan_.getLengthsTo()) << endl;
344 if (vl == VERB_EXTREME) {
345 out <<
"startsTo: " << toString (plan_.getStartsTo()) << endl;
346 out <<
"indicesTo: " << toString (plan_.getIndicesTo()) << endl;
348 if (vl == VERB_HIGH || vl == VERB_EXTREME) {
351 out <<
"lengthsFrom: " << toString (plan_.getLengthsFrom()) << endl;
352 out <<
"procsFrom: " << toString (plan_.getProcsFrom()) << endl;
356 return outStringP->str ();
362 const Teuchos::EVerbosityLevel
verbLevel)
const
365 using Teuchos::VERB_DEFAULT;
366 using Teuchos::VERB_NONE;
367 using Teuchos::VERB_LOW;
368 using Teuchos::VERB_MEDIUM;
369 using Teuchos::VERB_HIGH;
370 using Teuchos::VERB_EXTREME;
371 const Teuchos::EVerbosityLevel
vl =
382 if (plan_.getComm().is_null ()) {
385 const int myRank = plan_.getComm()->getRank ();
386 const int numProcs = plan_.getComm()->getSize ();
395 Teuchos::RCP<Teuchos::OSTab>
tab0,
tab1;
401 tab0 = Teuchos::rcp (
new Teuchos::OSTab (
out));
404 out <<
"\"Tpetra::Distributor\":" <<
endl;
405 tab1 = Teuchos::rcp (
new Teuchos::OSTab (
out));
409 out <<
"Label: " << label <<
endl;
412 <<
"How initialized: "
418 out <<
"\"Send type\": "
419 << DistributorSendTypeEnumToString (plan_.getSendType()) <<
endl
420 <<
"\"Debug\": " << (verbose_ ?
"true" :
"false") <<
endl;
426 const std::string
lclStr = this->localDescribeToString (
vl);
430 out <<
"Reverse Distributor:";
431 if (reverseDistributor_.is_null ()) {
436 reverseDistributor_->describe (
out,
vl);
Declaration of Tpetra::Details::Behavior, a class that describes Tpetra's behavior.
Declaration of a function that prints strings from each process.
Stand-alone utility functions and macros.
Struct that holds views of the contents of a CrsMatrix.
static bool verbose()
Whether Tpetra is in verbose mode.
Sets up and executes a communication plan for a Tpetra DistObject.
size_t getMaxSendLength() const
Maximum number of values this process will send to another single process.
Teuchos::RCP< Distributor > getReverse(bool create=true) const
A reverse communication plan Distributor.
Teuchos::ArrayView< const int > getProcsTo() const
Ranks of the processes to which this process will send values.
size_t getNumReceives() const
The number of processes from which we will receive data.
void setParameterList(const Teuchos::RCP< Teuchos::ParameterList > &plist)
Set Distributor parameters.
size_t getTotalReceiveLength() const
Total number of values this process will receive from other processes.
bool hasSelfMessage() const
Whether the calling process will send or receive messages to itself.
void swap(Distributor &rhs)
Swap the contents of rhs with those of *this.
Teuchos::ArrayView< const size_t > getLengthsTo() const
Number of values this process will send to each process.
Teuchos::ArrayView< const int > getProcsFrom() const
Ranks of the processes sending values to this process.
Distributor(const Teuchos::RCP< const Teuchos::Comm< int > > &comm)
Construct using the specified communicator and default parameters.
std::string description() const
Return a one-line description of this object.
size_t createFromSends(const Teuchos::ArrayView< const int > &exportProcIDs)
Set up Distributor using list of process ranks to which this process will send.
void createFromSendsAndRecvs(const Teuchos::ArrayView< const int > &exportProcIDs, const Teuchos::ArrayView< const int > &remoteProcIDs)
Set up Distributor using list of process ranks to which to send, and list of process ranks from which...
Teuchos::RCP< const Teuchos::ParameterList > getValidParameters() const
List of valid Distributor parameters.
Teuchos::ArrayView< const size_t > getLengthsFrom() const
Number of values this process will receive from each process.
size_t getNumSends() const
The number of processes to which we will send data.
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.
std::unique_ptr< std::string > createPrefix(const int myRank, const char prefix[])
Create string prefix for each line of verbose output.
std::string DistributorHowInitializedEnumToString(EDistributorHowInitialized how)
Convert an EDistributorHowInitialized enum value to a string.
void gathervPrint(std::ostream &out, const std::string &s, const Teuchos::Comm< int > &comm)
On Process 0 in the given communicator, print strings from each process in that communicator,...
Namespace Tpetra contains the class and methods constituting the Tpetra library.
Teuchos::Array< std::string > distributorSendTypes()
Valid values for Distributor's "Send type" parameter.