kokkos.cc

Go to the documentation of this file.

// ------------------------------------------------------------------------
//
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (C) 2022 - 2023 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Part of the source code is dual licensed under Apache-2.0 WITH
// LLVM-exception OR LGPL-2.1-or-later. Detailed license information
// governing the source code and code contributions can be found in
// LICENSE.md and CONTRIBUTING.md at the top level directory of deal.II.
//
// ------------------------------------------------------------------------
 
#include <deal.II/base/kokkos.h>
#include <deal.II/base/multithread_info.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 internal
{
  bool dealii_initialized_kokkos = false;
 
  void
  ensure_kokkos_initialized()
  {
    if (!Kokkos::is_initialized()
#if KOKKOS_VERSION >= 30700
        && !Kokkos::is_finalized()
#endif
    )
      {
        // only execute once
        static bool dummy = []() {
          dealii_initialized_kokkos = true;
#if KOKKOS_VERSION >= 30700
          const auto settings =
            Kokkos::InitializationSettings().set_num_threads(
              MultithreadInfo::n_threads());
#else
          const Kokkos::InitArguments settings(MultithreadInfo::n_threads());
#endif
          Kokkos::initialize(settings);
          std::atexit(Kokkos::finalize);
          return true;
        }();
        (void)dummy;
      }
  }
} // namespace internal
DEAL_II_NAMESPACE_CLOSE