Amesos2_SolverCore_decl.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//           Amesos2: Templated Direct Sparse Solver Package
//
// Copyright 2011 NTESS and the Amesos2 contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER
 
#ifndef AMESOS2_SOLVERCORE_DECL_HPP
#define AMESOS2_SOLVERCORE_DECL_HPP
 
#include <string>
 
#include <Teuchos_RCP.hpp>
#include <Teuchos_ParameterList.hpp>
 
#include "Amesos2_Solver.hpp"
#include "Amesos2_MatrixTraits.hpp"
#include "Amesos2_MatrixAdapter_decl.hpp"
#include "Amesos2_MultiVecAdapter_decl.hpp"
#include "Amesos2_TypeDecl.hpp"
 
#include "Amesos2_Control.hpp"
#include "Amesos2_Status.hpp"
#include "Amesos2_Timers.hpp"
 
namespace Amesos2 {
 
 
  /* This is the base class to be used in a *statically* polymorphic
   * way. E.g. for the Superlu solver:
   *
   * In Amesos2_Superlu.hpp:
   * class Superlu : SolverCore<Superlu> { ... }
   *
   * Each concrete solver will implement several private sub-functions
   * that will be called within the common code for each function.
   */
 
  template <template <class,class> class ConcreteSolver,
      class Matrix,
      class Vector >
  class SolverCore : public Amesos2::Solver<Matrix,Vector>
  {
  public:
 
    // Grant public access to contained types
    typedef SolverCore<ConcreteSolver,Matrix,Vector>                       type;
    typedef Solver<Matrix,Vector>                                    super_type;
    typedef ConcreteSolver<Matrix,Vector>                           solver_type;
    typedef Matrix                                                  matrix_type;
    typedef Vector                                                  vector_type;
    typedef typename MatrixAdapter<matrix_type>::scalar_t           scalar_type;
    typedef typename MatrixAdapter<matrix_type>::local_ordinal_t    local_ordinal_type;
    typedef typename MatrixAdapter<matrix_type>::global_ordinal_t   global_ordinal_type;
    typedef typename MatrixAdapter<matrix_type>::global_size_t      global_size_type;
    typedef typename MatrixAdapter<matrix_type>::node_t             node_type;
 
 
 
    SolverCore( Teuchos::RCP<const Matrix> A,
    Teuchos::RCP<Vector>       X,
    Teuchos::RCP<const Vector> B );
 
 
    ~SolverCore( );
 
 
    SolverCore(const solver_type& rhs);
 
 
    super_type& operator=(const solver_type* rhs);
 
 
 
 
    super_type& preOrdering() override;
 
 
    super_type& symbolicFactorization() override;
 
 
    super_type& numericFactorization() override;
 
 
    void solve() override;
    void solve(const Teuchos::Ptr<Vector> X, const Teuchos::Ptr<const Vector> B) const override;
    void solve(Vector* X, const Vector* B) const override;
 
    int solve_ir(const Teuchos::Ptr<Vector> X, const Teuchos::Ptr<const Vector> B,
                 const int maxNumIters, const bool verbose) const;
    int solve_ir(Vector* X, const Vector* B,
                 const int maxNumIters, const bool verbose) const;
    int solve_ir(const int maxNumIters, const bool verbose);
 
 
 
    bool matrixShapeOK() override;
 
    void setA( const Teuchos::RCP<const Matrix> a, EPhase keep_phase = CLEAN ) override;
 
    void setA( const Matrix* a, EPhase keep_phase = CLEAN ) override { setA(Teuchos::rcp(a), keep_phase); }
 
    void setX(const Teuchos::RCP<Vector> x) override { multiVecX_ = x; }
 
    void setX(Vector* x) override { multiVecX_ = Teuchos::rcp(x); }
  
    const Teuchos::RCP<Vector> getX() override { return( multiVecX_ ); }
 
    Vector* getXRaw() override { return multiVecX_.getRawPtr(); }
 
    void setB(const Teuchos::RCP<const Vector> b) override { multiVecB_ = b; }
 
    void setB(const Vector* b) override { multiVecB_ = Teuchos::rcp(b); }
 
    const Teuchos::RCP<const Vector> getB() override { return( multiVecB_ ); }
 
    const Vector* getBRaw() override { return multiVecB_.getRawPtr(); }
 
 
 
 
    super_type& setParameters(
            const Teuchos::RCP<Teuchos::ParameterList> & parameterList ) override;
 
 
    Teuchos::RCP<const Teuchos::ParameterList> getValidParameters() const override;
 
 
    void setParameterList(const Teuchos::RCP<Teuchos::ParameterList> & parameterList)
    {
      setParameters(parameterList);
    }
 
 
    Teuchos::RCP<Teuchos::ParameterList> getNonconstParameterList()
    {
      return Teuchos::null;
    }
 
 
    Teuchos::RCP<Teuchos::ParameterList> unsetParameterList()
    {
      return Teuchos::null;
    }
 
 
 
 
 
 
    Teuchos::RCP<const Teuchos::Comm<int> > getComm() const override
    {
      return matrixA_->getComm();
    }
 
  
    inline Status& getStatus() const override { return( status_ ); }
 
 
 
 
    std::string description() const override;
 
 
    void describe(Teuchos::FancyOStream &out,
      const Teuchos::EVerbosityLevel verbLevel=Teuchos::Describable::verbLevel_default) const override;
 
 
    void printTiming(
         Teuchos::FancyOStream &out,
         const Teuchos::EVerbosityLevel verbLevel) const override;
 
 
    void getTiming(Teuchos::ParameterList& timingParameterList) const override;
 
 
    std::string name() const override;
 
  private:
 
    void loadA(EPhase current_phase);
 
  protected:
 
    void setNnzLU(size_t nnz){ status_.lu_nnz_ = nnz; }
 
    Teuchos::RCP<const MatrixAdapter<Matrix> > matrixA_;
 
    bool matrix_loaded_;
 
 
    Teuchos::RCP<Vector> multiVecX_;
 
    Teuchos::RCP<const Vector> multiVecB_;
 
    global_size_type globalNumRows_;
 
    global_size_type globalNumCols_;
 
    global_size_type globalNumNonZeros_;
 
    global_size_type rowIndexBase_;
 
    global_size_type columnIndexBase_;
 
    mutable Status status_;
 
    Control control_;
 
    mutable Timers timers_;
 
 
    /* Useful MPI vars */
 
    int rank_;
 
    bool root_;
 
    int nprocs_;
 
    typedef Tpetra::Map<local_ordinal_type,
                        global_ordinal_type,
                        node_type> map_type;
    Teuchos::RCP<const map_type> contig_rowmap_;
    Teuchos::RCP<const map_type> contig_colmap_;
 
    typedef Kokkos::DefaultHostExecutionSpace                 HostExecSpaceType;
    typedef Kokkos::View<local_ordinal_type*, HostExecSpaceType> host_ordinal_type_array;
    typedef Kokkos::View<       scalar_type*, HostExecSpaceType> host_scalar_type_array;
    host_ordinal_type_array perm_g2l;
    host_ordinal_type_array recvCountRows, recvDisplRows;
    host_ordinal_type_array recvCounts, recvDispls;
    host_ordinal_type_array transpose_map;
    host_scalar_type_array  nzvals_t;
  };        // End class Amesos2::SolverCore
 
 
} // end namespace Amesos2
 
#endif  // AMESOS2_SOLVERCORE_DECL_HPP