From 370117ac9f0922ed3701ae0540d5a6640a8859be Mon Sep 17 00:00:00 2001
From: Marcel Koch <marcel.koch@kit.edu>
Date: Tue, 28 May 2024 16:31:09 +0200
Subject: [PATCH] [imap] create variant type of supported index maps

---
 .../ginkgo/core/distributed/index_map_fwd.hpp | 30 +++++++++++++++++++
 include/ginkgo/ginkgo.hpp                     |  1 +
 2 files changed, 31 insertions(+)
 create mode 100644 include/ginkgo/core/distributed/index_map_fwd.hpp

diff --git a/include/ginkgo/core/distributed/index_map_fwd.hpp b/include/ginkgo/core/distributed/index_map_fwd.hpp
new file mode 100644
index 00000000000..1d40a0008ec
--- /dev/null
+++ b/include/ginkgo/core/distributed/index_map_fwd.hpp
@@ -0,0 +1,30 @@
+// SPDX-FileCopyrightText: 2017 - 2024 The Ginkgo authors
+//
+// SPDX-License-Identifier: BSD-3-Clause
+
+#ifndef GINKGO_INDEX_MAP_FWD_HPP
+#define GINKGO_INDEX_MAP_FWD_HPP
+
+#include <variant>
+
+#include <ginkgo/core/base/types.hpp>
+
+
+namespace gko {
+namespace experimental {
+namespace distributed {
+
+
+template <typename LocalIndexType, typename GlobalIndexType>
+class index_map;
+
+using index_map_variant =
+    std::variant<index_map<int32, int32>, index_map<int32, int64>,
+                 index_map<int64, int64>>;
+
+
+}  // namespace distributed
+}  // namespace experimental
+}  // namespace gko
+
+#endif  // GINKGO_INDEX_MAP_FWD_HPP
diff --git a/include/ginkgo/ginkgo.hpp b/include/ginkgo/ginkgo.hpp
index 0fab93dcefe..cd40b74eed6 100644
--- a/include/ginkgo/ginkgo.hpp
+++ b/include/ginkgo/ginkgo.hpp
@@ -60,6 +60,7 @@
 
 #include <ginkgo/core/distributed/base.hpp>
 #include <ginkgo/core/distributed/index_map.hpp>
+#include <ginkgo/core/distributed/index_map_fwd.hpp>
 #include <ginkgo/core/distributed/lin_op.hpp>
 #include <ginkgo/core/distributed/matrix.hpp>
 #include <ginkgo/core/distributed/partition.hpp>