8 #ifndef MUELU_REPARTITIONINTERFACE_DEF_HPP_ 9 #define MUELU_REPARTITIONINTERFACE_DEF_HPP_ 16 #include "MueLu_Graph.hpp" 17 #include "MueLu_AmalgamationFactory.hpp" 18 #include "MueLu_AmalgamationInfo.hpp" 19 #include "MueLu_Utilities.hpp" 24 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
26 RCP<ParameterList> validParamList = rcp(
new ParameterList());
27 validParamList->set< RCP<const FactoryBase> >(
"A", Teuchos::null,
"Factory of the matrix A");
28 validParamList->set< RCP<const FactoryBase> >(
"number of partitions", Teuchos::null,
"Instance of RepartitionHeuristicFactory.");
29 validParamList->set< RCP<const FactoryBase> >(
"AmalgamatedPartition", Teuchos::null,
"(advanced) Factory generating the AmalgamatedPartition (e.g. an IsorropiaInterface)");
31 return validParamList;
35 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
37 Input(currentLevel,
"A");
38 Input(currentLevel,
"number of partitions");
39 Input(currentLevel,
"AmalgamatedPartition");
42 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
46 RCP<Matrix> A = Get< RCP<Matrix> > (level,
"A");
47 RCP<Xpetra::Vector<GO, LO, GO, NO> > amalgPartition = Get< RCP<Xpetra::Vector<GO, LO, GO, NO> > >(level,
"AmalgamatedPartition");
48 int numParts = Get<int>(level,
"number of partitions");
50 RCP<const Map> rowMap = A->getRowMap();
53 RCP<const Teuchos::Comm< int > > comm = A->getRowMap()->getComm();
56 if (numParts == 1 || numParts == -1) {
59 Set(level,
"Partition", decomposition);
61 }
else if (numParts == -1) {
63 RCP<Xpetra::Vector<GO,LO,GO,NO> > decomposition = Teuchos::null;
65 Set(level,
"Partition", decomposition);
69 ArrayRCP<GO> amalgPartitionData = amalgPartition->getDataNonConst(0);
70 RCP<const Map> nodeMap = amalgPartition->getMap();
76 LO nStridedOffset = 0;
77 LO stridedblocksize = blockdim;
81 if(A->IsView(
"stridedMaps") &&
82 Teuchos::rcp_dynamic_cast<
const StridedMap>(A->getRowMap(
"stridedMaps")) != Teuchos::null) {
84 RCP<const StridedMap> strMap = Teuchos::rcp_dynamic_cast<
const StridedMap>(A->getRowMap());
85 TEUCHOS_TEST_FOR_EXCEPTION(strMap == Teuchos::null,
Exceptions::BadCast,
"MueLu::RepartitionInterface::Build: cast to strided row map failed.");
86 blockdim = strMap->getFixedBlockSize();
87 offset = strMap->getOffset();
88 blockid = strMap->getStridedBlockId();
90 std::vector<size_t> stridingInfo = strMap->getStridingData();
91 for (
size_t j = 0; j < Teuchos::as<size_t>(blockid); j++)
92 nStridedOffset += stridingInfo[j];
93 stridedblocksize = Teuchos::as<LocalOrdinal>(stridingInfo[blockid]);
96 stridedblocksize = blockdim;
98 oldView = A->SwitchToView(oldView);
100 }
else GetOStream(
Statistics0, -1) <<
"RepartitionInterface::Build(): no striding information available. Use blockdim=1 with offset=0" << std::endl;
104 ArrayRCP<GO> decompEntries = decomposition->getDataNonConst(0);
106 TEUCHOS_TEST_FOR_EXCEPTION(Teuchos::as<int>(nodeMap->getNodeNumElements())*stridedblocksize != Teuchos::as<int>(rowMap->getNodeNumElements()),
Exceptions::RuntimeError,
"Inconsistency between nodeMap and dofMap: we are supporting block maps only. No support for general strided maps, yet!");
113 for(
size_t i = 0; i < nodeMap->getNodeNumElements(); i++) {
125 for (
LO j = 0; j < stridedblocksize; j++) {
131 decompEntries[i*stridedblocksize + j] = amalgPartitionData[i];
137 Set(level,
"Partition", decomposition);
Exception indicating invalid cast attempted.
Timer to be used in factories. Similar to Monitor but with additional timers.
Namespace for MueLu classes and methods.
Print statistics that do not involve significant additional computation.
Class that holds all level-specific information.
static RCP< Vector > Build(const Teuchos::RCP< const Map > &map, bool zeroOut=true)
void Build(Level &level) const
Build an object with this factory.
void DeclareInput(Level &level) const
Specifies the data that this class needs, and the factories that generate that data.
Exception throws to report errors in the internal logical of the program.
RCP< const ParameterList > GetValidParameterList() const
Return a const parameter list of valid parameters that setParameterList() will accept.