47 #ifndef MUELU_AGGREGATES_KOKKOS_DECL_HPP
48 #define MUELU_AGGREGATES_KOKKOS_DECL_HPP
52 #include <Kokkos_StaticCrsGraph.hpp>
53 #include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
63 #include "MueLu_LWGraph_kokkos.hpp"
64 #include "MueLu_IndexManager_kokkos.hpp"
66 #define MUELU_UNAGGREGATED -1
69 #define MUELU_UNASSIGNED -1
102 template <
class LocalOrdinal,
class GlobalOrdinal,
class Node>
105 template <
class LocalOrdinal,
class GlobalOrdinal,
class DeviceType>
115 using node_type = Tpetra::KokkosCompat::KokkosDeviceWrapperNode<DeviceType>;
117 using range_type = Kokkos::RangePolicy<local_ordinal_type, execution_space>;
118 using LO_view = Kokkos::View<local_ordinal_type*, device_type>;
125 #undef MUELU_AGGREGATES_KOKKOS_SHORT
132 using colors_view_type = Kokkos::View<
typename local_graph_type::entries_type::data_type,
133 typename local_graph_type::device_type::memory_space>;
203 return numAggregates_;
208 aggregatesIncludeGhosts_ = flag;
216 return aggregatesIncludeGhosts_;
243 KOKKOS_INLINE_FUNCTION
250 KOKKOS_INLINE_FUNCTION
264 typename aggregates_sizes_type::const_type ComputeAggregateSizes(
bool forceRecompute =
false)
const;
266 local_graph_type GetGraph()
const;
272 void ComputeNodesInAggregate(LO_view & aggPtr, LO_view & aggNodes, LO_view & unaggregated)
const;
276 GO GetNumGlobalAggregatesComputeIfNeeded();
282 std::string description()
const;
335 #define MUELU_AGGREGATES_KOKKOS_SHORT
336 #endif // MUELU_AGGREGATES_KOKKOS_DECL_HPP
MueLu::DefaultLocalOrdinal LocalOrdinal
KOKKOS_INLINE_FUNCTION void AggregatesCrossProcessors(const bool &flag)
Record whether aggregates include DOFs from other processes.
Lightweight MueLu representation of a compressed row storage graph.
KOKKOS_INLINE_FUNCTION bool AggregatesCrossProcessors() const
Return false if and only if no aggregates include DOFs from other processes.
virtual ~Aggregates_kokkos()
Destructor.
void SetGraphColors(colors_view_type graphColors)
Set a distance 2 coloring of the underlying graph. The coloring is computed and set during Phase1 of ...
void SetNumGlobalAggregates(GO nGlobalAggregates)
Set number of global aggregates on current processor.
colors_view_type graphColors_
LocalOrdinal local_ordinal_type
typename LWGraph_kokkos::local_graph_type local_graph_type
GlobalOrdinal global_ordinal_type
Kokkos::View< bool *, device_type > isRoot_
const RCP< LOMultiVector > & GetVertex2AggId() const
Returns constant vector that maps local node IDs to local aggregates IDs.
LO numAggregates_
Number of aggregates on this processor.
local_graph_type graph_
Aggregates represented as Kokkos graph type.
typename DeviceType::execution_space execution_space
MueLu::DefaultGlobalOrdinal GlobalOrdinal
RCP< IndexManager_kokkos > & GetIndexManager()
Get the index manager used by structured aggregation algorithms. This has to be done by the aggregati...
Kokkos::View< local_ordinal_type *, device_type > LO_view
GO numGlobalAggregates_
Number of global aggregates.
RCP< IndexManager_kokkos > geoData_
Kokkos::View< LocalOrdinal *, device_type > aggregates_sizes_type
RCP< LOMultiVector > vertex2AggId_
KOKKOS_INLINE_FUNCTION void SetIsRoot(LO i, bool value=true)
Set root node information.
const RCP< LOVector > & GetProcWinner() const
Returns constant vector that maps local node IDs to owning processor IDs.
void SetNumAggregates(LO nAggregates)
Set number of local aggregates on current processor.
Tpetra::KokkosCompat::KokkosDeviceWrapperNode< DeviceType > node_type
Base class for MueLu classes.
void SetGraphNumColors(const LO graphNumColors)
Set the number of colors needed by the distance 2 coloring.
RCP< LOVector > & GetProcWinnerNonConst()
Returns nonconsant vector that maps local node IDs to owning processor IDs.
RCP< LOMultiVector > & GetVertex2AggIdNonConst()
Returns a nonconstant vector that maps local node IDs to local aggregates IDs.
LO GetGraphNumColors()
Get the number of colors needed by the distance 2 coloring.
KOKKOS_INLINE_FUNCTION LO GetNumAggregates() const
Kokkos::View< typename local_graph_type::entries_type::data_type, typename local_graph_type::device_type::memory_space > colors_view_type
Kokkos::RangePolicy< local_ordinal_type, execution_space > range_type
aggregates_sizes_type aggregateSizes_
Array of sizes of each local aggregate.
KOKKOS_INLINE_FUNCTION bool IsRoot(LO i) const
Returns true if node with given local node id is marked to be a root node.
colors_view_type & GetGraphColors()
Get a distance 2 coloring of the underlying graph. The coloring is computed and set during Phase1 of ...
RCP< LOVector > procWinner_
void SetIndexManager(RCP< IndexManager_kokkos > &geoData)
Set the index manager used by structured aggregation algorithms. This has to be done by the aggregati...
bool aggregatesIncludeGhosts_
Set to false iff aggregates do not include any DOFs belong to other processes.