ESPResSo
Extensible Simulation Package for Research on Soft Matter Systems
Loading...
Searching...
No Matches
InitialPDFsSetterDoublePrecisionCUDA.h
Go to the documentation of this file.
1//======================================================================================================================
2//
3// This file is part of waLBerla. waLBerla is free software: you can
4// redistribute it and/or modify it under the terms of the GNU General Public
5// License as published by the Free Software Foundation, either version 3 of
6// the License, or (at your option) any later version.
7//
8// waLBerla is distributed in the hope that it will be useful, but WITHOUT
9// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
10// FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
11// for more details.
12//
13// You should have received a copy of the GNU General Public License along
14// with waLBerla (see COPYING.txt). If not, see <http://www.gnu.org/licenses/>.
15//
16//! \\file InitialPDFsSetterDoublePrecisionCUDA.h
17//! \\author pystencils
18//======================================================================================================================
19
20// kernel generated with pystencils v1.3.3, lbmpy v1.3.3,
21// lbmpy_walberla/pystencils_walberla from waLBerla commit
22// b0842e1a493ce19ef1bbb8d2cf382fc343970a7f
23
24#pragma once
25#include "core/DataTypes.h"
26#include "core/logging/Logging.h"
27
28#include "gpu/GPUField.h"
29#include "gpu/GPUWrapper.h"
30
31#include "domain_decomposition/BlockDataID.h"
32#include "domain_decomposition/IBlock.h"
33#include "domain_decomposition/StructuredBlockStorage.h"
34#include "field/SwapableCompare.h"
35#include <set>
36
37#ifdef __GNUC__
38#define RESTRICT __restrict__
39#elif _MSC_VER
40#define RESTRICT __restrict
41#else
42#define RESTRICT
43#endif
44
45#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
46 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
47#pragma GCC diagnostic push
48#pragma GCC diagnostic ignored "-Wunused-parameter"
49#pragma GCC diagnostic ignored "-Wreorder"
50#endif
51
52namespace walberla {
53namespace pystencils {
54
56public:
58 BlockDataID pdfsID_,
59 BlockDataID velocityID_, double rho_0)
60 : forceID(forceID_), pdfsID(pdfsID_), velocityID(velocityID_),
61 rho_0_(rho_0){};
62
63 void run(IBlock *block, gpuStream_t stream = nullptr);
64
65 void runOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
66 const CellInterval &globalCellInterval,
67 cell_idx_t ghostLayers, IBlock *block,
68 gpuStream_t stream = nullptr);
69
70 void operator()(IBlock *block, gpuStream_t stream = nullptr) {
72 }
73
74 static std::function<void(IBlock *)>
75 getSweep(const shared_ptr<InitialPDFsSetterDoublePrecisionCUDA> &kernel) {
76 return [kernel](IBlock *b) { kernel->run(b); };
77 }
78
79 static std::function<void(IBlock *, gpuStream_t)> getSweepOnCellInterval(
80 const shared_ptr<InitialPDFsSetterDoublePrecisionCUDA> &kernel,
81 const shared_ptr<StructuredBlockStorage> &blocks,
82 const CellInterval &globalCellInterval, cell_idx_t ghostLayers = 1) {
83 return [kernel, blocks, globalCellInterval,
84 ghostLayers](IBlock *b, gpuStream_t stream = nullptr) {
85 kernel->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b,
86 stream);
87 };
88 }
89
90 std::function<void(IBlock *)> getSweep(gpuStream_t stream = nullptr) {
91 return [this, stream](IBlock *b) { this->run(b, stream); };
92 }
93
94 std::function<void(IBlock *)>
95 getSweepOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
96 const CellInterval &globalCellInterval,
97 cell_idx_t ghostLayers = 1,
98 gpuStream_t stream = nullptr) {
99 return [this, blocks, globalCellInterval, ghostLayers, stream](IBlock *b) {
100 this->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b,
101 stream);
102 };
103 }
104
105 void configure(const shared_ptr<StructuredBlockStorage> &blocks,
106 IBlock *block) {}
107
108 BlockDataID forceID;
109 BlockDataID pdfsID;
110 BlockDataID velocityID;
111 double rho_0_;
112};
113
114} // namespace pystencils
115} // namespace walberla
116
117#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
118 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
119#pragma GCC diagnostic pop
120#endif
void runOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers, IBlock *block, gpuStream_t stream=nullptr)
std::function< void(IBlock *)> getSweep(gpuStream_t stream=nullptr)
std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1, gpuStream_t stream=nullptr)
static std::function< void(IBlock *)> getSweep(const shared_ptr< InitialPDFsSetterDoublePrecisionCUDA > &kernel)
InitialPDFsSetterDoublePrecisionCUDA(BlockDataID forceID_, BlockDataID pdfsID_, BlockDataID velocityID_, double rho_0)
static std::function< void(IBlock *, gpuStream_t)> getSweepOnCellInterval(const shared_ptr< InitialPDFsSetterDoublePrecisionCUDA > &kernel, const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1)
void configure(const shared_ptr< StructuredBlockStorage > &blocks, IBlock *block)
cudaStream_t stream[1]
CUDA streams for parallel computing on CPU and GPU.
static double * block(double *p, std::size_t index, std::size_t size)
Definition elc.cpp:172
\file PackInfoPdfDoublePrecision.cpp \author pystencils