Merge Pull Request #7156 from cgcgcg/Trilinos/tpetraDistParams

Automatically Merged using Trilinos Pull Request AutoTester PR Title: Tpetra Distributor: add method to get reverse distributor PR Author: cgcgcg
trilinos · Apr 13, 2020 · be49977 · be49977
2 parents 9a9f81c + e68542f
commit be49977
Show file tree

Hide file tree

Showing 4 changed files with 17 additions and 1 deletion.
diff --git a/packages/tpetra/core/src/Tpetra_Distributor.cpp b/packages/tpetra/core/src/Tpetra_Distributor.cpp
@@ -385,6 +385,12 @@ namespace Tpetra {
     }
   }
 
+  Teuchos::RCP<Distributor>
+  Distributor::
+  getReverseDistributor() {
+    return reverseDistributor_;
+  }
+
   Teuchos::RCP<const Teuchos::ParameterList>
   Distributor::getValidParameters () const
   {

diff --git a/packages/tpetra/core/src/Tpetra_Distributor.hpp b/packages/tpetra/core/src/Tpetra_Distributor.hpp
@@ -265,6 +265,8 @@ namespace Tpetra {
     /// parameters and their default values.
     void setParameterList (const Teuchos::RCP<Teuchos::ParameterList>& plist);
 
+    Teuchos::RCP<Distributor> getReverseDistributor();
+
     /// \brief List of valid Distributor parameters.
     ///
     /// Please see the class documentation for a list of all accepted

diff --git a/packages/xpetra/src/Export/Xpetra_TpetraExport_def.hpp b/packages/xpetra/src/Export/Xpetra_TpetraExport_def.hpp
@@ -201,6 +201,9 @@ TpetraExport<LocalOrdinal, GlobalOrdinal, Node>::
 setDistributorParameters(const Teuchos::RCP<Teuchos::ParameterList> params) const {
   XPETRA_MONITOR("TpetraExport::setDistributorParameters");
   export_->getDistributor().setParameterList(params);
+  auto revDistor = export_->getDistributor().getReverseDistributor();
+  if (!revDistor.is_null())
+    revDistor->setParameterList(params);
 }
 
 

diff --git a/packages/xpetra/src/Import/Xpetra_TpetraImport_def.hpp b/packages/xpetra/src/Import/Xpetra_TpetraImport_def.hpp
@@ -95,7 +95,12 @@ size_t TpetraImport<LocalOrdinal,GlobalOrdinal,Node>::getNumRemoteIDs() const
 
 template<class LocalOrdinal, class GlobalOrdinal, class Node>
 void TpetraImport<LocalOrdinal,GlobalOrdinal,Node>::setDistributorParameters(const Teuchos::RCP<Teuchos::ParameterList> params) const
-{ XPETRA_MONITOR("TpetraImport::setDistributorParameters"); import_->getDistributor().setParameterList(params); }
+{ XPETRA_MONITOR("TpetraImport::setDistributorParameters");
+  import_->getDistributor().setParameterList(params);
+  auto revDistor = import_->getDistributor().getReverseDistributor();
+  if (!revDistor.is_null())
+    revDistor->setParameterList(params);
+}
 
 template<class LocalOrdinal, class GlobalOrdinal, class Node>
 ArrayView< const LocalOrdinal > TpetraImport<LocalOrdinal,GlobalOrdinal,Node>::getRemoteLIDs() const