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> >(
"AmalgamatedPartition", Teuchos::null,
"(advanced) Factory generating the AmalgamatedPartition (e.g. an IsorropiaInterface)");
30 return validParamList;
34 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
36 Input(currentLevel,
"A");
37 Input(currentLevel,
"AmalgamatedPartition");
40 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
44 RCP<Matrix> A = Get< RCP<Matrix> > (level,
"A");
45 RCP<Xpetra::Vector<GO, LO, GO, NO> > amalgPartition = Get< RCP<Xpetra::Vector<GO, LO, GO, NO> > >(level,
"AmalgamatedPartition");
48 RCP<const Teuchos::Comm< int > > comm = A->getRowMap()->getComm();
50 ArrayRCP<GO> amalgPartitionData = amalgPartition->getDataNonConst(0);
52 RCP<const Map> rowMap = A->getRowMap();
53 RCP<const Map> nodeMap = amalgPartition->getMap();
59 LO nStridedOffset = 0;
60 LO stridedblocksize = blockdim;
64 if(A->IsView(
"stridedMaps") &&
65 Teuchos::rcp_dynamic_cast<
const StridedMap>(A->getRowMap(
"stridedMaps")) != Teuchos::null) {
66 Xpetra::viewLabel_t oldView = A->SwitchToView(
"stridedMaps");
67 RCP<const StridedMap> strMap = Teuchos::rcp_dynamic_cast<
const StridedMap>(A->getRowMap());
68 TEUCHOS_TEST_FOR_EXCEPTION(strMap == Teuchos::null,
Exceptions::BadCast,
"MueLu::RepartitionInterface::Build: cast to strided row map failed.");
69 blockdim = strMap->getFixedBlockSize();
70 offset = strMap->getOffset();
71 blockid = strMap->getStridedBlockId();
73 std::vector<size_t> stridingInfo = strMap->getStridingData();
74 for (
size_t j = 0; j < Teuchos::as<size_t>(blockid); j++)
75 nStridedOffset += stridingInfo[j];
76 stridedblocksize = Teuchos::as<LocalOrdinal>(stridingInfo[blockid]);
79 stridedblocksize = blockdim;
81 oldView = A->SwitchToView(oldView);
82 GetOStream(
Statistics0, -1) <<
"RepartitionInterface::Build():" <<
" found blockdim=" << blockdim <<
" from strided maps (blockid=" << blockid <<
", strided block size=" << stridedblocksize <<
"). offset=" << offset << std::endl;
83 }
else GetOStream(
Statistics0, -1) <<
"RepartitionInterface::Build(): no striding information available. Use blockdim=1 with offset=0" << std::endl;
86 RCP<Xpetra::Vector<GO, LO, GO, NO> > decomposition = Xpetra::VectorFactory<GO, LO, GO, NO>::Build(rowMap,
false);
87 ArrayRCP<GO> decompEntries = decomposition->getDataNonConst(0);
89 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!");
96 for(
size_t i = 0; i < nodeMap->getNodeNumElements(); i++) {
108 for (LO j = 0; j < stridedblocksize; j++) {
114 decompEntries[i*stridedblocksize + j] = amalgPartitionData[i];
120 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.
RCP< const ParameterList > GetValidParameterList() const
Return a const parameter list of valid parameters that setParameterList() will accept.
Class that holds all level-specific information.
void Build(Level &level) const
Build an object with this factory.
Exception throws to report errors in the internal logical of the program.
void DeclareInput(Level &level) const
Specifies the data that this class needs, and the factories that generate that data.