5#ifndef GKO_PUBLIC_CORE_DISTRIBUTED_ASSEMBLY_HPP_
6#define GKO_PUBLIC_CORE_DISTRIBUTED_ASSEMBLY_HPP_
9#include <ginkgo/config.hpp>
15#include <ginkgo/core/base/device_matrix_data.hpp>
16#include <ginkgo/core/base/mpi.hpp>
17#include <ginkgo/core/base/range.hpp>
21namespace experimental {
22namespace distributed {
25template <
typename LocalIndexType,
typename GlobalIndexType>
44template <
typename ValueType,
typename LocalIndexType,
typename GlobalIndexType>
This type is a device-side equivalent to matrix_data.
Definition device_matrix_data.hpp:36
Represents a partition of a range of indices [0, size) into a disjoint set of parts.
Definition vector.hpp:36
A thin wrapper of MPI_Comm that supports most MPI calls.
Definition mpi.hpp:408
This class is used for function parameters in the place of raw pointers.
Definition utils_helper.hpp:41
device_matrix_data< ValueType, GlobalIndexType > assemble_rows_from_neighbors(mpi::communicator comm, const device_matrix_data< ValueType, GlobalIndexType > &input, ptr_param< const Partition< LocalIndexType, GlobalIndexType > > partition)
Assembles device_matrix_data entries owned by this MPI rank from other ranks and communicates entries...
The Ginkgo namespace.
Definition abstract_factory.hpp:20