OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2025-02-25 18:55:30 -06:00
Code Issues Packages Projects Releases Wiki Activity

Only do CPU decomposition if needed, when using GPU

Browse Source
This commit is contained in:
Tong Dong Qiu 2023-03-22 11:06:36 +01:00
parent 5c1d964478
commit 0bf91691fe
2 changed files with 21 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
opm/simulators/linalg/ISTLSolverEbos.cpp
View File

@ -265,6 +265,13 @@ apply(Vector& rhs,
return false; return false;
} }
template<class Matrix, class Vector>
bool BdaSolverInfo<Matrix,Vector>::
gpuActive()
{
return bridge_->getUseGpu();
}
template<class Matrix, class Vector> template<class Matrix, class Vector>
template<class Grid> template<class Grid>
void BdaSolverInfo<Matrix,Vector>:: void BdaSolverInfo<Matrix,Vector>::

14
opm/simulators/linalg/ISTLSolverEbos.hpp
View File

@ -149,6 +149,8 @@ struct BdaSolverInfo
Vector& x, Vector& x,
Dune::InverseOperatorResult& result); Dune::InverseOperatorResult& result);
bool gpuActive();
int numJacobiBlocks_ = 0; int numJacobiBlocks_ = 0;
private: private:
@ -351,7 +353,14 @@ std::unique_ptr<Matrix> blockJacobiAdjacency(const Grid& grid,
if (isParallel() && prm_.get<std::string>("preconditioner.type") != "ParOverILU0") { if (isParallel() && prm_.get<std::string>("preconditioner.type") != "ParOverILU0") {
detail::makeOverlapRowsInvalid(getMatrix(), overlapRows_); detail::makeOverlapRowsInvalid(getMatrix(), overlapRows_);
} }
#if COMPILE_BDA_BRIDGE
if(!bdaBridge->gpuActive()){
prepareFlexibleSolver();
}
#else
prepareFlexibleSolver(); prepareFlexibleSolver();
#endif
firstcall = false; firstcall = false;
} }
@ -401,6 +410,11 @@ std::unique_ptr<Matrix> blockJacobiAdjacency(const Grid& grid,
#endif #endif
{ {
OPM_TIMEBLOCK(flexibleSolverApply); OPM_TIMEBLOCK(flexibleSolverApply);
#if COMPILE_BDA_BRIDGE
if(bdaBridge->gpuActive()){
prepareFlexibleSolver();
}
#endif
assert(flexibleSolver_.solver_); assert(flexibleSolver_.solver_);
flexibleSolver_.solver_->apply(x, *rhs_, result); flexibleSolver_.solver_->apply(x, *rhs_, result);
} }