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