From 7ce8b103213a35295b498151c85787d0cc6ebf04 Mon Sep 17 00:00:00 2001
From: Daniel Arndt <arndtd@ornl.gov>
Date: Wed, 30 Nov 2022 16:32:44 +0000
Subject: [PATCH] Add files for making sure Kokkos is initialized

---
 include/deal.II/base/kokkos.h | 31 ++++++++++++++++++
 source/base/kokkos.cc         | 59 +++++++++++++++++++++++++++++++++++
 2 files changed, 90 insertions(+)
 create mode 100644 include/deal.II/base/kokkos.h
 create mode 100644 source/base/kokkos.cc

diff --git a/include/deal.II/base/kokkos.h b/include/deal.II/base/kokkos.h
new file mode 100644
index 0000000000..7117f02fa8
--- /dev/null
+++ b/include/deal.II/base/kokkos.h
@@ -0,0 +1,31 @@
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2011 - 2021 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE.md at
+// the top level directory of deal.II.
+//
+// ---------------------------------------------------------------------
+
+#ifndef dealii_kokkos_h
+#define dealii_kokkos_h
+
+#include <deal.II/base/config.h>
+
+DEAL_II_NAMESPACE_OPEN
+
+namespace Impl
+{
+  void
+  ensure_kokkos_initialized();
+}
+
+DEAL_II_NAMESPACE_CLOSE
+
+#endif
diff --git a/source/base/kokkos.cc b/source/base/kokkos.cc
new file mode 100644
index 0000000000..629c0063d1
--- /dev/null
+++ b/source/base/kokkos.cc
@@ -0,0 +1,59 @@
+// ---------------------------------------------------------------------
+//
+// Copyright (C) 2011 - 2021 by the deal.II authors
+//
+// This file is part of the deal.II library.
+//
+// The deal.II library is free software; you can use it, redistribute
+// it, and/or modify it under the terms of the GNU Lesser General
+// Public License as published by the Free Software Foundation; either
+// version 2.1 of the License, or (at your option) any later version.
+// The full text of the license can be found in the file LICENSE.md at
+// the top level directory of deal.II.
+//
+// ---------------------------------------------------------------------
+
+#include <deal.II/base/kokkos.h>
+
+#include <deal.II/lac/la_parallel_vector.h>
+#include <deal.II/lac/vector_memory.h>
+
+#include <Kokkos_Core.hpp>
+
+DEAL_II_NAMESPACE_OPEN
+
+namespace Impl
+{
+  void
+  ensure_kokkos_initialized()
+  {
+    if (!Kokkos::is_initialized())
+      GrowingVectorMemory<
+        LinearAlgebra::distributed::Vector<double, MemorySpace::Host>>{};
+    GrowingVectorMemory<
+      LinearAlgebra::distributed::Vector<float, MemorySpace::Host>>{};
+    GrowingVectorMemory<
+      LinearAlgebra::distributed::Vector<double, MemorySpace::Device>>{};
+    GrowingVectorMemory<
+      LinearAlgebra::distributed::Vector<float, MemorySpace::Device>>{};
+    Kokkos::push_finalize_hook(
+      GrowingVectorMemory<
+        LinearAlgebra::distributed::Vector<double, MemorySpace::Host>>::
+        release_unused_memory);
+    Kokkos::push_finalize_hook(
+      GrowingVectorMemory<
+        LinearAlgebra::distributed::Vector<float, MemorySpace::Host>>::
+        release_unused_memory);
+    Kokkos::push_finalize_hook(
+      GrowingVectorMemory<
+        LinearAlgebra::distributed::Vector<double, MemorySpace::Device>>::
+        release_unused_memory);
+    Kokkos::push_finalize_hook(
+      GrowingVectorMemory<
+        LinearAlgebra::distributed::Vector<float, MemorySpace::Device>>::
+        release_unused_memory);
+    Kokkos::initialize();
+    std::atexit(Kokkos::finalize);
+  }
+} // namespace Impl
+DEAL_II_NAMESPACE_CLOSE
-- 
2.39.5