TpetraExt_TripleMatrixMultiply_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//          Tpetra: Templated Linear Algebra Services Package
//
// Copyright 2008 NTESS and the Tpetra contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef TPETRA_TRIPLEMATRIXMULTIPLY_DECL_HPP
#define TPETRA_TRIPLEMATRIXMULTIPLY_DECL_HPP
 
#include <string>
#include <Teuchos_RCP.hpp>
#include <Teuchos_Array.hpp>
#include "Tpetra_ConfigDefs.hpp"
#include "Tpetra_CrsMatrix.hpp"
#include "Tpetra_Vector.hpp"
#include "TpetraExt_MMHelpers.hpp"
 
namespace Tpetra {
 
namespace TripleMatrixMultiply {
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
void MultiplyRAP(
    const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& R,
    bool transposeR,
    const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& A,
    bool transposeA,
    const CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& P,
    bool transposeP,
    CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
    bool call_FillComplete_on_result                   = true,
    const std::string& label                           = std::string(),
    const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
 
}  // namespace TripleMatrixMultiply
 
namespace MMdetails {
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
void mult_R_A_P_newmatrix(
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Rview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Aview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Pview,
    CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
    const std::string& label                           = std::string(),
    const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
void mult_R_A_P_reuse(
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Rview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Aview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Pview,
    CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
    const std::string& label                           = std::string(),
    const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
void mult_PT_A_P_newmatrix(
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Aview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Pview,
    CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
    const std::string& label                           = std::string(),
    const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
 
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
void mult_PT_A_P_reuse(
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Aview,
    CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Pview,
    CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
    const std::string& label                           = std::string(),
    const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
 
// Kernel wrappers struct (for non-specialized kernels)
// Because C++ doesn't support partial template specialization of functions.
template <class Scalar,
          class LocalOrdinal,
          class GlobalOrdinal,
          class Node>
struct KernelWrappers3MMM {
  static inline void mult_PT_A_P_newmatrix_kernel_wrapper_2pass(CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Aview,
                                                                CrsMatrixStruct<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Pview,
                                                                const Teuchos::Array<LocalOrdinal>& Acol2PRow,
                                                                const Teuchos::Array<LocalOrdinal>& Acol2PRowImport,
                                                                const Teuchos::Array<LocalOrdinal>& Pcol2Accol,
                                                                const Teuchos::Array<LocalOrdinal>& PIcol2Accol,
                                                                CrsMatrix<Scalar, LocalOrdinal, GlobalOrdinal, Node>& Ac,
                                                                Teuchos::RCP<const Import<LocalOrdinal, GlobalOrdinal, Node> > Acimport,
                                                                const std::string& label                           = std::string(),
                                                                const Teuchos::RCP<Teuchos::ParameterList>& params = Teuchos::null);
};
 
}  // end namespace MMdetails
 
}  // namespace Tpetra
 
#endif  // TPETRA_TRIPLEMATRIXMULTIPLY_DECL_HPP