5#ifndef GKO_PUBLIC_CORE_DISTRIBUTED_ROW_GATHERER_HPP_
6#define GKO_PUBLIC_CORE_DISTRIBUTED_ROW_GATHERER_HPP_
9#include <ginkgo/config.hpp>
15#include <ginkgo/core/base/dense_cache.hpp>
16#include <ginkgo/core/base/lin_op.hpp>
17#include <ginkgo/core/base/mpi.hpp>
18#include <ginkgo/core/distributed/base.hpp>
19#include <ginkgo/core/distributed/collective_communicator.hpp>
20#include <ginkgo/core/distributed/index_map.hpp>
24namespace experimental {
54template <
typename LocalIndexType =
int32>
55class RowGatherer final
58 public DistributedBase {
108 std::shared_ptr<const mpi::CollectiveCommunicator>
141 template <
typename GlobalIndexType =
int64,
142 typename = std::enable_if_t<
sizeof(GlobalIndexType) >=
143 sizeof(LocalIndexType)>>
144 static std::unique_ptr<RowGatherer>
create(
145 std::shared_ptr<const Executor> exec,
146 std::shared_ptr<const mpi::CollectiveCommunicator> coll_comm,
149 return std::unique_ptr<RowGatherer>(
150 new RowGatherer(std::move(exec), std::move(coll_comm), imap));
156 static std::unique_ptr<RowGatherer>
create(
173 template <
typename GlobalIndexType>
175 std::shared_ptr<const mpi::CollectiveCommunicator> coll_comm,
185 std::shared_ptr<const mpi::CollectiveCommunicator> coll_comm_;
191 mutable MPI_Request req_listener_{MPI_REQUEST_NULL};
This mixin is used to enable a default PolymorphicObject::copy_from() implementation for objects that...
Definition polymorphic_object.hpp:743
This mixin inherits from (a subclass of) PolymorphicObject and provides a base implementation of a ne...
Definition polymorphic_object.hpp:668
An array is a container which encapsulates fixed-sized arrays, stored on the Executor tied to the arr...
Definition array.hpp:166
The distributed::RowGatherer gathers the rows of distributed::Vector that are located on other proces...
Definition row_gatherer.hpp:58
mpi::request apply_async(ptr_param< const LinOp > b, ptr_param< LinOp > x, array< char > &workspace) const
Asynchronous version of LinOp::apply.
mpi::request apply_async(ptr_param< const LinOp > b, ptr_param< LinOp > x) const
Asynchronous version of LinOp::apply.
const LocalIndexType * get_const_send_idxs() const
Read access to the (local) rows indices.
static std::unique_ptr< RowGatherer > create(std::shared_ptr< const Executor > exec, std::shared_ptr< const mpi::CollectiveCommunicator > coll_comm, const index_map< LocalIndexType, GlobalIndexType > &imap)
Creates a distributed::RowGatherer from a given collective communicator and index map.
Definition row_gatherer.hpp:144
dim< 2 > get_size() const
Returns the size of the row gatherer.
size_type get_num_send_idxs() const
Returns the number of (local) row indices.
std::shared_ptr< const mpi::CollectiveCommunicator > get_collective_communicator() const
Get the used collective communicator.
This class defines mappings between global and local indices.
Definition index_map.hpp:68
A thin wrapper of MPI_Comm that supports most MPI calls.
Definition mpi.hpp:419
The request class is a light, move-only wrapper around the MPI_Request handle.
Definition mpi.hpp:331
This class is used for function parameters in the place of raw pointers.
Definition utils_helper.hpp:41
The distributed namespace.
Definition polymorphic_object.hpp:19
The Ginkgo namespace.
Definition abstract_factory.hpp:20
std::int64_t int64
64-bit signed integral type.
Definition types.hpp:113
std::size_t size_type
Integral type used for allocation quantities.
Definition types.hpp:90
A type representing the dimensions of a multidimensional object.
Definition dim.hpp:26