BelosStochasticCGIteration.hpp

Go to the documentation of this file.

// @HEADER
// *****************************************************************************
//                 Belos: Block Linear Solvers Package
//
// Copyright 2004-2016 NTESS and the Belos contributors.
// SPDX-License-Identifier: BSD-3-Clause
// *****************************************************************************
// @HEADER

#ifndef BELOS_CG_STOCHASTIC_ITERATION_HPP
#define BELOS_CG_STOCHASTIC_ITERATION_HPP

#include "BelosConfigDefs.hpp"
#include "BelosTypes.hpp"
#include "BelosIteration.hpp"
#include "BelosCGIteration.hpp"

namespace Belos {


  
  template <class ScalarType, class MV>
  struct StochasticCGIterationState {

    Teuchos::RCP<const MV> R;

    Teuchos::RCP<const MV> Z;

    Teuchos::RCP<const MV> P;

    Teuchos::RCP<const MV> AP;
    
    Teuchos::RCP<const MV> Y;

    StochasticCGIterationState() : R(Teuchos::null), Z(Teuchos::null), 
       P(Teuchos::null), AP(Teuchos::null), Y(Teuchos::null)
    {}
  };

template<class ScalarType, class MV, class OP>
class StochasticCGIteration : virtual public Iteration<ScalarType,MV,OP> {

  public:



  virtual void initializeCG(StochasticCGIterationState<ScalarType,MV>& newstate) = 0;

  virtual StochasticCGIterationState<ScalarType,MV> getState() const = 0;

};

} // end Belos namespace

#endif /* BELOS_STOCHASTIC_CG_ITERATION_HPP */