 |
Ginkgo
Generated from pipelines/2017069469 branch based on develop. Ginkgo version 1.11.0
A numerical linear algebra library targeting many-core architectures
|
5 #ifndef GKO_PUBLIC_CORE_DISTRIBUTED_NEIGHBORHOOD_COMMUNICATOR_HPP_
6 #define GKO_PUBLIC_CORE_DISTRIBUTED_NEIGHBORHOOD_COMMUNICATOR_HPP_
9 #include <ginkgo/config.hpp>
15 #include <ginkgo/core/base/mpi.hpp>
16 #include <ginkgo/core/distributed/collective_communicator.hpp>
17 #include <ginkgo/core/distributed/index_map.hpp>
21 namespace experimental {
66 template <
typename LocalIndexType,
typename GlobalIndexType>
80 [[nodiscard]] std::unique_ptr<CollectiveCommunicator>
create_inverse()
114 request i_all_to_all_v_impl(std::shared_ptr<const Executor> exec,
115 const void* send_buffer, MPI_Datatype send_type,
117 MPI_Datatype recv_type)
const override;
122 std::vector<comm_index_type> send_sizes_;
123 std::vector<comm_index_type> send_offsets_;
124 std::vector<comm_index_type> recv_sizes_;
125 std::vector<comm_index_type> recv_offsets_;
135 #endif // GKO_PUBLIC_CORE_DISTRIBUTED_NEIGHBORHOOD_COMMUNICATOR_HPP_
std::unique_ptr< CollectiveCommunicator > create_with_same_type(communicator base, index_map_ptr imap) const override
Creates a new CollectiveCommunicator with the same dynamic type.
comm_index_type get_send_size() const override
Get the number of elements sent by this process within this communication pattern.
request i_all_to_all_v(std::shared_ptr< const Executor > exec, const SendType *send_buffer, RecvType *recv_buffer) const
Non-blocking all-to-all communication.
Definition: collective_communicator.hpp:126
The request class is a light, move-only wrapper around the MPI_Request handle.
Definition: mpi.hpp:331
Interface for a collective communicator.
Definition: collective_communicator.hpp:31
comm_index_type get_recv_size() const override
Get the number of elements received by this process within this communication pattern.
The Ginkgo namespace.
Definition: abstract_factory.hpp:20
A thin wrapper of MPI_Comm that supports most MPI calls.
Definition: mpi.hpp:419
int comm_index_type
Index type for enumerating processes in a distributed application.
Definition: types.hpp:1060
std::unique_ptr< CollectiveCommunicator > create_inverse() const override
Creates the inverse NeighborhoodCommunicator by switching sources and destinations.
A CollectiveCommunicator that uses a neighborhood topology.
Definition: neighborhood_communicator.hpp:33
std::variant< const distributed::index_map< int32, int32 > *, const distributed::index_map< int32, int64 > *, const distributed::index_map< int64, int64 > * > index_map_ptr
All allowed index_map types (as const *)
Definition: collective_communicator.hpp:39
friend bool operator==(const NeighborhoodCommunicator &a, const NeighborhoodCommunicator &b)
Compares two communicators for equality locally.
This class defines mappings between global and local indices.
Definition: index_map.hpp:68
friend bool operator!=(const NeighborhoodCommunicator &a, const NeighborhoodCommunicator &b)
Compares two communicators for inequality.