Ginkgo Generated from develop branch based on develop. Ginkgo version 1.10.0
A numerical linear algebra library targeting many-core architectures
 
Loading...
Searching...
No Matches
vector_cache.hpp
1// SPDX-FileCopyrightText: 2017 - 2025 The Ginkgo authors
2//
3// SPDX-License-Identifier: BSD-3-Clause
4
5#ifndef GKO_PUBLIC_CORE_DISTRIBUTED_VECTOR_CACHE_HPP_
6#define GKO_PUBLIC_CORE_DISTRIBUTED_VECTOR_CACHE_HPP_
7
8
9#include <memory>
10
11#include <ginkgo/config.hpp>
12#include <ginkgo/core/base/executor.hpp>
13#include <ginkgo/core/base/mpi.hpp>
14#include <ginkgo/core/distributed/vector.hpp>
15
16
17#if GINKGO_BUILD_MPI
18
19
20namespace gko {
21namespace experimental {
22namespace distributed {
23namespace detail {
24
25
35template <typename ValueType>
36class VectorCache {
37public:
38 VectorCache() = default;
39 ~VectorCache() = default;
40 VectorCache(const VectorCache&) {}
41 VectorCache(VectorCache&&) noexcept {}
42 VectorCache& operator=(const VectorCache&) { return *this; }
43 VectorCache& operator=(VectorCache&&) noexcept { return *this; }
44 mutable std::unique_ptr<Vector<ValueType>> vec{};
45
46
61 void init_from(const Vector<ValueType>* template_vec) const;
62
75 void init(std::shared_ptr<const Executor> exec,
76 gko::experimental::mpi::communicator comm, dim<2> global_size,
77 dim<2> local_size) const;
78
84 Vector<ValueType>& operator*() const { return *vec; }
85
90 Vector<ValueType>* operator->() const { return vec.get(); }
91
96 Vector<ValueType>* get() const { return vec.get(); }
97};
98
99
100} // namespace detail
101} // namespace distributed
102} // namespace experimental
103} // namespace gko
104
105
106#endif // GINKGO_BUILD_MPI
107#endif // GKO_PUBLIC_CORE_DISTRIBUTED_VECTOR_CACHE_HPP_
The distributed namespace.
Definition polymorphic_object.hpp:19
The Ginkgo namespace.
Definition abstract_factory.hpp:20