mirror of
https://github.com/OPM/opm-simulators.git
synced 2025-01-27 11:16:28 -06:00
83 lines
2.8 KiB
C++
83 lines
2.8 KiB
C++
/*
|
|
Copyright 2019 Equinor ASA
|
|
|
|
This file is part of the Open Porous Media project (OPM).
|
|
|
|
OPM 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.
|
|
|
|
OPM is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
You should have received a copy of the GNU General Public License
|
|
along with OPM. If not, see <http://www.gnu.org/licenses/>.
|
|
*/
|
|
|
|
#ifndef BDABRIDGE_HEADER_INCLUDED
|
|
#define BDABRIDGE_HEADER_INCLUDED
|
|
|
|
#include <config.h>
|
|
#include "dune/istl/solver.hh" // for struct InverseOperatorResult
|
|
|
|
#include "dune/istl/bcrsmatrix.hh"
|
|
#include <opm/simulators/linalg/matrixblock.hh>
|
|
|
|
#include <opm/simulators/linalg/bda/WellContributions.hpp>
|
|
|
|
#if HAVE_CUDA
|
|
#include <opm/simulators/linalg/bda/cusparseSolverBackend.hpp>
|
|
#endif
|
|
|
|
namespace Opm
|
|
{
|
|
|
|
typedef Dune::InverseOperatorResult InverseOperatorResult;
|
|
|
|
/// BdaBridge acts as interface between opm-simulators with the cusparseSolver
|
|
/// if CUDA was not found during CMake, function bodies of this class are empty
|
|
class BdaBridge
|
|
{
|
|
private:
|
|
#if HAVE_CUDA
|
|
std::unique_ptr<cusparseSolverBackend> backend;
|
|
bool use_gpu;
|
|
#endif
|
|
|
|
public:
|
|
/// Construct a BdaBridge
|
|
/// \param[in] use_gpu true iff the cusparseSolver is used, is passed via command-line: '--use-gpu=[true|false]'
|
|
/// \param[in] linear_solver_verbosity verbosity of cusparseSolver
|
|
/// \param[in] maxit maximum number of iterations for cusparseSolver
|
|
/// \param[in] tolerance required relative tolerance for cusparseSolver
|
|
BdaBridge(bool use_gpu, int linear_solver_verbosity, int maxit, double tolerance);
|
|
|
|
|
|
/// Solve linear system, A*x = b
|
|
/// \param[in] mat matrix A, should be of type Dune::BCRSMatrix
|
|
/// \param[in] b vector b, should be of type Dune::BlockVector
|
|
/// \param[in] wellContribs contains all WellContributions, to apply them separately, instead of adding them to matrix A
|
|
/// \param[inout] result summary of solver result
|
|
template <class BridgeMatrix, class BridgeVector>
|
|
void solve_system(BridgeMatrix *mat, BridgeVector &b, WellContributions& wellContribs, InverseOperatorResult &result);
|
|
|
|
/// Get the resulting x vector
|
|
/// \param[inout] x vector x, should be of type Dune::BlockVector
|
|
template <class BridgeVector>
|
|
void get_result(BridgeVector &x);
|
|
|
|
/// Return whether the BdaBridge will use the GPU or not
|
|
/// return whether the BdaBridge will use the GPU or not
|
|
bool getUseGpu(){
|
|
return use_gpu;
|
|
}
|
|
|
|
}; // end class BdaBridge
|
|
|
|
}
|
|
|
|
#endif
|