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
23namespace Opm::cuistl
24{
34void setDevice(int mpiRank, int numberOfMpiRanks);
35} // namespace Opm::cuistl
36#endif
Definition: CuBlockPreconditioner.hpp:29
void setDevice(int mpiRank, int numberOfMpiRanks)
Sets the correct CUDA device in the setting of MPI.