ESPResSo
Extensible Simulation Package for Research on Soft Matter Systems
Loading...
Searching...
No Matches
CollideSweepSinglePrecisionThermalizedCUDA.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 CollideSweepSinglePrecisionThermalizedCUDA.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_, float kT,
59 float omega_bulk, float omega_even,
60 float omega_odd, float omega_shear,
61 uint32_t seed, uint32_t time_step)
62 : forceID(forceID_), pdfsID(pdfsID_), kT_(kT), omega_bulk_(omega_bulk),
63 omega_even_(omega_even), omega_odd_(omega_odd),
64 omega_shear_(omega_shear), seed_(seed), time_step_(time_step),
65 configured_(false){};
66
67 void run(IBlock *block, gpuStream_t stream = nullptr);
68
69 void runOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
70 const CellInterval &globalCellInterval,
71 cell_idx_t ghostLayers, IBlock *block,
72 gpuStream_t stream = nullptr);
73
74 void operator()(IBlock *block, gpuStream_t stream = nullptr) {
76 }
77
78 static std::function<void(IBlock *)> getSweep(
79 const shared_ptr<CollideSweepSinglePrecisionThermalizedCUDA> &kernel) {
80 return [kernel](IBlock *b) { kernel->run(b); };
81 }
82
83 static std::function<void(IBlock *, gpuStream_t)> getSweepOnCellInterval(
84 const shared_ptr<CollideSweepSinglePrecisionThermalizedCUDA> &kernel,
85 const shared_ptr<StructuredBlockStorage> &blocks,
86 const CellInterval &globalCellInterval, cell_idx_t ghostLayers = 1) {
87 return [kernel, blocks, globalCellInterval,
88 ghostLayers](IBlock *b, gpuStream_t stream = nullptr) {
89 kernel->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b,
90 stream);
91 };
92 }
93
94 std::function<void(IBlock *)> getSweep(gpuStream_t stream = nullptr) {
95 return [this, stream](IBlock *b) { this->run(b, stream); };
96 }
97
98 std::function<void(IBlock *)>
99 getSweepOnCellInterval(const shared_ptr<StructuredBlockStorage> &blocks,
100 const CellInterval &globalCellInterval,
101 cell_idx_t ghostLayers = 1,
102 gpuStream_t stream = nullptr) {
103 return [this, blocks, globalCellInterval, ghostLayers, stream](IBlock *b) {
104 this->runOnCellInterval(blocks, globalCellInterval, ghostLayers, b,
105 stream);
106 };
107 }
108
109 void configure(const shared_ptr<StructuredBlockStorage> &blocks,
110 IBlock *block) {
111 Cell BlockCellBB = blocks->getBlockCellBB(*block).min();
112 block_offset_0_ = uint32_t(BlockCellBB[0]);
113 block_offset_1_ = uint32_t(BlockCellBB[1]);
114 block_offset_2_ = uint32_t(BlockCellBB[2]);
115 configured_ = true;
116 }
117
118 BlockDataID forceID;
119 BlockDataID pdfsID;
123 float kT_;
128 uint32_t seed_;
129 uint32_t time_step_;
131};
132
133} // namespace pystencils
134} // namespace walberla
135
136#if (defined WALBERLA_CXX_COMPILER_IS_GNU) || \
137 (defined WALBERLA_CXX_COMPILER_IS_CLANG)
138#pragma GCC diagnostic pop
139#endif
Definition Cell.hpp:97
CollideSweepSinglePrecisionThermalizedCUDA(BlockDataID forceID_, BlockDataID pdfsID_, float kT, float omega_bulk, float omega_even, float omega_odd, float omega_shear, uint32_t seed, uint32_t time_step)
std::function< void(IBlock *)> getSweepOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1, gpuStream_t stream=nullptr)
void runOnCellInterval(const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers, IBlock *block, gpuStream_t stream=nullptr)
void configure(const shared_ptr< StructuredBlockStorage > &blocks, IBlock *block)
static std::function< void(IBlock *, gpuStream_t)> getSweepOnCellInterval(const shared_ptr< CollideSweepSinglePrecisionThermalizedCUDA > &kernel, const shared_ptr< StructuredBlockStorage > &blocks, const CellInterval &globalCellInterval, cell_idx_t ghostLayers=1)
static std::function< void(IBlock *)> getSweep(const shared_ptr< CollideSweepSinglePrecisionThermalizedCUDA > &kernel)
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