AmrMultiGridDefs.h

Go to the documentation of this file.

//
// Header file AmrMultiGridDefs
//   Extends the AMR namespace with Trilinos types
//
// Copyright (c) 2017 - 2020, Matthias Frey, Paul Scherrer Institut, Villigen PSI, Switzerland
// All rights reserved
//
// Implemented as part of the PhD thesis
// "Precise Simulations of Multibunches in High Intensity Cyclotrons"
//
// This file is part of OPAL.
//
// OPAL is free software: you can redistribute it and/or modify
// it under the terms of the GNU General Public License as published by
// the Free Software Foundation, either version 3 of the License, or
// (at your option) any later version.
//
// You should have received a copy of the GNU General Public License
// along with OPAL. If not, see <https://www.gnu.org/licenses/>.
//
#ifndef AMR_MULTI_GRID_DEFS_H
#define AMR_MULTI_GRID_DEFS_H
 
// Trilinos headers
#include <Tpetra_Map.hpp>
#include <Tpetra_Vector.hpp>
#include <Tpetra_CrsMatrix.hpp>
 
#include <Teuchos_RCP.hpp>
#include <Teuchos_ArrayRCP.hpp>
#include <Teuchos_DefaultMpiComm.hpp> // wrapper for our communicator
 
#include <Kokkos_DefaultNode.hpp>
 
namespace amr {
    // All Tpetra
    typedef double scalar_t;
    typedef int local_ordinal_t;
    typedef long global_ordinal_t;
    
#ifdef AMR_MG_SERIAL_NODE
    typedef ::Kokkos::Compat::KokkosSerialWrapperNode node_t;
#elif AMR_MG_PTHREAD_NODE
    typedef ::Kokkos::Compat::KokkosThreadsWrapperNode node_t;
#elif AMR_MG_OPENMP_NODE
    typedef ::Kokkos::Compat::KokkosOpenMPWrapperNode node_t;
#elif AMR_MG_CUDA_NODE
    typedef ::Kokkos::Compat::KokkosCudaWrapperNode node_t;
#else
    typedef KokkosClassic::DefaultNode::DefaultNodeType node_t;
#endif
    
    typedef Tpetra::CrsMatrix<scalar_t,
                              local_ordinal_t,
                              global_ordinal_t,
                              node_t
            > matrix_t;
            
    typedef Tpetra::Vector<scalar_t,
                           local_ordinal_t,
                           global_ordinal_t,
                           node_t
            > vector_t;
    
    typedef Tpetra::Operator<scalar_t,
                             local_ordinal_t,
                             global_ordinal_t,
                             node_t
            > operator_t;
    
    typedef Tpetra::MultiVector<scalar_t,
                                local_ordinal_t,
                                global_ordinal_t,
                                node_t
            > multivector_t;
    
    
    typedef Tpetra::Map<local_ordinal_t,
                        global_ordinal_t,
                        node_t
            > dmap_t;
    
    typedef Teuchos::MpiComm<int>    comm_t;
}
 
#endif