opm
simulators
linalg
gpuistl
set_device.hpp
Go to the documentation of this file.
1
/*
2
Copyright 2022-2023 SINTEF AS
3
4
This file is part of the Open Porous Media project (OPM).
5
6
OPM is free software: you can redistribute it and/or modify
7
it under the terms of the GNU General Public License as published by
8
the Free Software Foundation, either version 3 of the License, or
9
(at your option) any later version.
10
11
OPM is distributed in the hope that it will be useful,
12
but WITHOUT ANY WARRANTY; without even the implied warranty of
13
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14
GNU General Public License for more details.
15
16
You should have received a copy of the GNU General Public License
17
along with OPM. If not, see <http://www.gnu.org/licenses/>.
18
*/
19
20
#ifndef OPM_CUISTL_SET_DEVICE_HEADER
21
#define OPM_CUISTL_SET_DEVICE_HEADER
22
23
namespace
Opm::gpuistl
24
{
34
void
setDevice
(
int
mpiRank,
int
numberOfMpiRanks);
35
}
// namespace Opm::gpuistl
36
#endif
Opm::gpuistl
Definition:
autotuner.hpp:29
Opm::gpuistl::setDevice
void setDevice(int mpiRank, int numberOfMpiRanks)
Sets the correct CUDA device in the setting of MPI.