OilfieldToolsNet/opm-simulators
4
0
Fork 0
mirror of https://github.com/OPM/opm-simulators.git synced 2024-12-28 18:21:00 -06:00
Code Issues Packages Projects Releases Wiki Activity

Disable GPU when using MPI

Browse Source
This commit is contained in:
T.D. (Tongdong) Qiu 2020-03-18 13:53:40 +01:00
parent d067c08be7
commit ce73337769
2 changed files with 13 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -236,8 +236,13 @@ protected:
converged_(false)
{
parameters_.template init<TypeTag>();
const auto& gridForConn = simulator_.vanguard().grid();
#if HAVE_CUDA
const bool use_gpu = EWOMS_GET_PARAM(TypeTag, bool, UseGpu);
bool use_gpu = EWOMS_GET_PARAM(TypeTag, bool, UseGpu);
if (gridForConn.comm().size() > 1 && use_gpu) {
OpmLog::warning("Warning cannot use GPU with MPI, GPU is disabled");
use_gpu = false;
}
const int maxit = EWOMS_GET_PARAM(TypeTag, int, LinearSolverMaxIter);
const double tolerance = EWOMS_GET_PARAM(TypeTag, double, LinearSolverReduction);
const int linear_solver_verbosity = parameters_.linear_solver_verbosity_;
@ -249,7 +254,6 @@ protected:
}
#endif
extractParallelGridInformationToISTL(simulator_.vanguard().grid(), parallelInformation_);
const auto& gridForConn = simulator_.vanguard().grid();
const auto wellsForConn = simulator_.vanguard().schedule().getWellsatEnd();
const bool useWellConn = EWOMS_GET_PARAM(TypeTag, bool, MatrixAddWellContributions);
@ -440,7 +444,7 @@ protected:
#if HAVE_CUDA
WellContributions wellContribs;
const bool addWellContribs = EWOMS_GET_PARAM(TypeTag, bool, MatrixAddWellContributions);
const bool use_gpu = EWOMS_GET_PARAM(TypeTag, bool, UseGpu);
const bool use_gpu = bdaBridge->getUseGpu();
if(!addWellContribs && use_gpu)
{
simulator_.problem().wellModel().getWellContributions(wellContribs);

6
opm/simulators/linalg/bda/BdaBridge.hpp
View File

@ -69,6 +69,12 @@ public:
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
}