ESPResSo
Extensible Simulation Package for Research on Soft Matter Systems
Loading...
Searching...
No Matches
lb_walberla_init.cu
Go to the documentation of this file.
1/*
2 * Copyright (C) 2022-2023 The ESPResSo project
3 *
4 * This file is part of ESPResSo.
5 *
6 * ESPResSo 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 * ESPResSo 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 this program. If not, see <http://www.gnu.org/licenses/>.
18 */
19
21
22#if defined(__NVCC__)
23#define RESTRICT __restrict__
24#if defined(__NVCC_DIAG_PRAGMA_SUPPORT__)
25#pragma nv_diagnostic push
26#pragma nv_diag_suppress 554 // no implicit or explicit cast
27#else
28#pragma push
29#pragma diag_suppress 554 // no implicit or explicit cast
30#endif
31#endif
32
33#include "LBWalberlaImpl.hpp"
34
35#if defined(__NVCC__)
36#if defined(__NVCC_DIAG_PRAGMA_SUPPORT__)
37#pragma nv_diagnostic pop
38#else
39#pragma pop
40#endif
41#endif
42
46
47#include <gpu/DeviceSelectMPI.h>
48
49#include <memory>
50
51std::shared_ptr<LBWalberlaBase>
52new_lb_walberla_gpu(std::shared_ptr<LatticeWalberla> const &lattice,
53 double viscosity, double density, bool single_precision) {
54 if (single_precision) {
55 return std::make_shared<walberla::LBWalberlaImpl<float, lbmpy::Arch::GPU>>(
56 lattice, viscosity, density);
57 }
58 return std::make_shared<walberla::LBWalberlaImpl<double, lbmpy::Arch::GPU>>(
59 lattice, viscosity, density);
60}
61
62void set_device_id_per_rank() { walberla::gpu::selectDeviceBasedOnMpiRank(); }
LBWalberlaBase provides the public interface of the LB waLBerla bridge.
walberla::LBWalberlaImpl implements the interface of the LB waLBerla bridge using sweeps generated by...
std::shared_ptr< LBWalberlaBase > new_lb_walberla_gpu(std::shared_ptr< LatticeWalberla > const &lattice, double viscosity, double density, bool single_precision)
void set_device_id_per_rank()