Panzer_NodeType.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Panzer: A partial differential equation assembly
//       engine for strongly coupled complex multiphysics systems
//
// Copyright 2011 NTESS and the Panzer contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef PANZER_KOKKOS_NODE_TYPE_HPP
#define PANZER_KOKKOS_NODE_TYPE_HPP
 
#include "Phalanx_KokkosDeviceTypes.hpp"
#include <Tpetra_KokkosCompat_ClassicNodeAPI_Wrapper.hpp>
#include <type_traits>
 
namespace panzer {

  using TpetraNodeType =
      Tpetra::KokkosCompat::KokkosDeviceWrapperNode<PHX::ExecutionSpace,
                                                    PHX::MemorySpace>;

  template <typename Node>
  struct TpetraNodeIsInstantiated : std::false_type {};
 
#if defined(HAVE_TPETRA_INST_SERIAL) && defined(KOKKOS_ENABLE_SERIAL)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosSerialWrapperNode>
      : std::true_type {};
#endif
#if defined(HAVE_TPETRA_INST_OPENMP) && defined(KOKKOS_ENABLE_OPENMP)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosOpenMPWrapperNode>
      : std::true_type {};
#endif
#if defined(HAVE_TPETRA_INST_PTHREAD) && defined(KOKKOS_ENABLE_THREADS)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosThreadsWrapperNode>
      : std::true_type {};
#endif
#if defined(HAVE_TPETRA_INST_CUDA) && defined(KOKKOS_ENABLE_CUDA)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosCudaWrapperNode>
      : std::true_type {};
#endif
#if defined(HAVE_TPETRA_INST_HIP) && defined(KOKKOS_ENABLE_HIP)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosHIPWrapperNode>
      : std::true_type {};
#endif
#if defined(HAVE_TPETRA_INST_SYCL) && defined(KOKKOS_ENABLE_SYCL)
  template <>
  struct TpetraNodeIsInstantiated<Tpetra::KokkosCompat::KokkosSYCLWrapperNode>
      : std::true_type {};
#endif
 
  static_assert(TpetraNodeIsInstantiated<TpetraNodeType>::value,
                "panzer: the Tpetra node type panzer derives from Phalanx' "
                "execution and memory spaces is not one that Tpetra "
                "instantiated, so every Tpetra object panzer creates would "
                "fail to link.  Either set the Tpetra_INST_<BACKEND> matching "
                "Phalanx_DEFAULT_EXECUTION_SPACE -- note Tpetra_INST_HIP and "
                "Tpetra_INST_SYCL are OFF by default, so a HIP or SYCL build "
                "needs one set explicitly -- or make "
                "Tpetra_ALLOCATE_IN_SHARED_SPACE agree with "
                "Phalanx_ENABLE_SHARED_SPACE, since the two select the same "
                "Kokkos::SharedSpace and enabling one alone changes the node's "
                "memory space to one Tpetra did not instantiate.");
 
}
 
#endif