trilinos_tpetra_communication_pattern.h

Go to the documentation of this file.

// ------------------------------------------------------------------------
//
// SPDX-License-Identifier: LGPL-2.1-or-later
// Copyright (C) 2018 - 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.
//
// ------------------------------------------------------------------------
 
#ifndef dealii_trilinos_tpetra_communication_pattern_h
#define dealii_trilinos_tpetra_communication_pattern_h
 
 
#include <deal.II/base/config.h>
 
#include "deal.II/base/memory_space.h"
 
#include <deal.II/lac/trilinos_tpetra_types.h>
 
#ifdef DEAL_II_TRILINOS_WITH_TPETRA
 
#  include <deal.II/base/communication_pattern_base.h>
#  include <deal.II/base/memory_space.h>
 
#  include <Tpetra_Export.hpp>
#  include <Tpetra_Import.hpp>
 
#  include <memory>
 
DEAL_II_NAMESPACE_OPEN
 
namespace LinearAlgebra
{
  namespace TpetraWrappers
  {
    template <typename MemorySpace = ::MemorySpace::Host>
    class CommunicationPattern : public Utilities::MPI::CommunicationPatternBase
    {
      static_assert(std::is_same_v<MemorySpace, ::MemorySpace::Default> ||
                    std::is_same_v<MemorySpace, ::MemorySpace::Host>);
 
    public:
      CommunicationPattern(const IndexSet &locally_owned_indices,
                           const IndexSet &ghost_indices,
                           const MPI_Comm  communicator);
 
      virtual void
      reinit(const IndexSet &locally_owned_indices,
             const IndexSet &ghost_indices,
             const MPI_Comm  communicator) override;
 
      virtual MPI_Comm
      get_mpi_communicator() const override;
 
      const TpetraTypes::ImportType<MemorySpace> &
      get_tpetra_import() const;
 
      Teuchos::RCP<TpetraTypes::ImportType<MemorySpace>>
      get_tpetra_import_rcp() const;
 
      const TpetraTypes::ExportType<MemorySpace> &
      get_tpetra_export() const;
 
      Teuchos::RCP<TpetraTypes::ExportType<MemorySpace>>
      get_tpetra_export_rcp() const;
 
    private:
      Teuchos::RCP<const MPI_Comm> comm;
 
      Teuchos::RCP<TpetraTypes::ImportType<MemorySpace>> tpetra_import;
 
      Teuchos::RCP<TpetraTypes::ExportType<MemorySpace>> tpetra_export;
    };
  } // end of namespace TpetraWrappers
} // end of namespace LinearAlgebra
 
DEAL_II_NAMESPACE_CLOSE
 
#endif
 
#endif