5#ifndef GKO_PUBLIC_CORE_DISTRIBUTED_INDEX_MAP_HPP_
6#define GKO_PUBLIC_CORE_DISTRIBUTED_INDEX_MAP_HPP_
9#include <ginkgo/core/base/segmented_array.hpp>
10#include <ginkgo/core/distributed/partition.hpp>
14namespace experimental {
67template <
typename LocalIndexType,
typename GlobalIndexType =
int64>
112 std::shared_ptr<const partition_type> partition,
166 std::shared_ptr<const Executor> exec_;
167 std::shared_ptr<const partition_type> partition_;
An array is a container which encapsulates fixed-sized arrays, stored on the Executor tied to the arr...
Definition array.hpp:166
Represents a partition of a range of indices [0, size) into a disjoint set of parts.
Definition partition.hpp:82
The distributed namespace.
Definition precision_dispatch.hpp:309
index_space
Index space classification for the locally stored indices.
Definition index_map.hpp:23
@ combined
both local and non_local indices
Definition index_map.hpp:26
@ non_local
indices that are owned by other processes
Definition index_map.hpp:25
@ local
indices that are locally owned
Definition index_map.hpp:24
int comm_index_type
Index type for enumerating processes in a distributed application.
Definition types.hpp:967
The Ginkgo namespace.
Definition abstract_factory.hpp:20
std::size_t size_type
Integral type used for allocation quantities.
Definition types.hpp:89
index_map(std::shared_ptr< const Executor > exec, std::shared_ptr< const partition_type > partition, comm_index_type rank, const array< GlobalIndexType > &recv_connections)
Creates a new index map.
const array< comm_index_type > & get_remote_target_ids() const
get the rank ids which contain indices accessed by this rank.
size_type get_local_size() const
get size of index_space::local
size_type get_global_size() const
get size of index_space::local
array< LocalIndexType > map_to_local(const array< GlobalIndexType > &global_ids, index_space index_space_v) const
Maps global indices to local indices.
size_type get_non_local_size() const
get size of index_space::non_local
index_map(std::shared_ptr< const Executor > exec)
Creates an empty index map.
const segmented_array< GlobalIndexType > & get_remote_global_idxs() const
get the index set for this rank.
const segmented_array< LocalIndexType > & get_remote_local_idxs() const
get the index set , but mapped to their respective local index space.
std::shared_ptr< const Executor > get_executor() const
get the associated executor.
A minimal interface for a segmented array.
Definition segmented_array.hpp:27