trilinos_tpetra_types.h

Go to the documentation of this file.

// -----------------------------------------------------------------------------
//
// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception OR LGPL-2.1-or-later
// Copyright (C) 2018 - 2026 by the deal.II authors
//
// This file is part of the deal.II library.
//
// Detailed license information governing the source code and contributions
// can be found in LICENSE.md and CONTRIBUTING.md at the top level directory.
//
// -----------------------------------------------------------------------------
 
#ifndef dealii_trilinos_tpetra_types_h
#define dealii_trilinos_tpetra_types_h
 
#include <deal.II/base/config.h>
 
#include <deal.II/base/types.h>
 
#ifdef DEAL_II_TRILINOS_WITH_TPETRA
 
#  include <deal.II/base/memory_space.h>
 
#  include <Kokkos_DualView.hpp>
#  include <Tpetra_Details_DefaultTypes.hpp>
 
 
// Forward declarations
#  ifndef DOXYGEN
#    ifdef DEAL_II_TRILINOS_WITH_IFPACK2
#      include <Ifpack2_Preconditioner.hpp>
#    endif
#    include <Tpetra_CrsGraph_fwd.hpp>
#    include <Tpetra_CrsMatrix_fwd.hpp>
#    include <Tpetra_Export_fwd.hpp>
#    include <Tpetra_Import_fwd.hpp>
#    include <Tpetra_Map_fwd.hpp>
#    include <Tpetra_MultiVector_fwd.hpp>
#    include <Tpetra_Operator_fwd.hpp>
#    include <Tpetra_RowMatrix_fwd.hpp>
#    include <Tpetra_Vector_fwd.hpp>
#  endif
 
#endif
 
DEAL_II_NAMESPACE_OPEN
 
#ifdef DEAL_II_TRILINOS_WITH_TPETRA
namespace LinearAlgebra
{
  namespace TpetraWrappers
  {
 
    namespace TpetraTypes
    {
      using LO = int;
 
 
      using GO = types::signed_global_dof_index;
 
 
#  if DEAL_II_TRILINOS_VERSION_GTE(14, 2, 0)
      //
      template <typename MemorySpace>
      using NodeType = Tpetra::KokkosCompat::KokkosDeviceWrapperNode<
        typename MemorySpace::kokkos_space::execution_space,
        typename MemorySpace::kokkos_space>;
#  else
      template <typename MemorySpace>
      using NodeType = Kokkos::Compat::KokkosDeviceWrapperNode<
        typename MemorySpace::kokkos_space::execution_space,
        typename MemorySpace::kokkos_space>;
#  endif
 
      template <typename MemorySpace>
      using ExportType = Tpetra::Export<LO, GO, NodeType<MemorySpace>>;
 
      template <typename MemorySpace>
      using ImportType = Tpetra::Import<LO, GO, NodeType<MemorySpace>>;
 
      template <typename MemorySpace>
      using MapType = Tpetra::Map<LO, GO, NodeType<MemorySpace>>;
 
      template <typename MemorySpace>
      using GraphType = Tpetra::CrsGraph<LO, GO, NodeType<MemorySpace>>;
 
      template <typename Number, typename MemorySpace>
      using VectorType = Tpetra::Vector<Number, LO, GO, NodeType<MemorySpace>>;
 
      template <typename Number, typename MemorySpace>
      using MultiVectorType =
        Tpetra::MultiVector<Number, LO, GO, NodeType<MemorySpace>>;
 
 
      template <typename Number, typename MemorySpace>
      using LinearOperator =
        Tpetra::Operator<Number, LO, GO, NodeType<MemorySpace>>;
 
 
      template <typename Number, typename MemorySpace>
      using MatrixType =
        Tpetra::CrsMatrix<Number, LO, GO, NodeType<MemorySpace>>;
 
 
      template <typename Number, typename MemorySpace>
      using RowMatrixType =
        Tpetra::RowMatrix<Number, LO, GO, NodeType<MemorySpace>>;
 
      template <typename Number, typename MemorySpace>
      using DualViewType =
        typename VectorType<Number, MemorySpace>::dual_view_type;
 
      template <typename Number>
      using HostViewType =
        typename DualViewType<Number, ::MemorySpace::Host>::t_host;
 
#  ifdef DEAL_II_TRILINOS_WITH_IFPACK2
      template <typename Number, typename MemorySpace>
      using Ifpack2PreconType =
        Ifpack2::Preconditioner<Number, LO, GO, NodeType<MemorySpace>>;
#  endif
 
    } // namespace TpetraTypes
  }   // namespace TpetraWrappers
} // namespace LinearAlgebra
 
#endif
 
DEAL_II_NAMESPACE_CLOSE
 
#endif