diff options
Diffstat (limited to 'boltzgen/kernel')
-rw-r--r-- | boltzgen/kernel/generator.py | 5 | ||||
-rw-r--r-- | boltzgen/kernel/template/basic.cpp.mako | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py index e897325..d91ddd5 100644 --- a/boltzgen/kernel/generator.py +++ b/boltzgen/kernel/generator.py @@ -35,7 +35,10 @@ class Generator: collision_subexpr = self.collision[0], collision_assignment = self.collision[1], - float_type = precision, + float_type = { + 'single': 'float', + 'double': 'double' + }.get(precision), boundary_src = Template(self.boundary).render( descriptor = self.descriptor, diff --git a/boltzgen/kernel/template/basic.cpp.mako b/boltzgen/kernel/template/basic.cpp.mako index 529c1de..1036c8f 100644 --- a/boltzgen/kernel/template/basic.cpp.mako +++ b/boltzgen/kernel/template/basic.cpp.mako @@ -131,7 +131,7 @@ void test(std::size_t nStep) std::cout << "MLUPS: " << nStep*${geometry.volume}/(1e6*duration.count()) << std::endl; // calculate average rho as a basic quality check - double rho_sum = 0.0; + ${float_type} rho_sum = 0.0; for (std::size_t i = 0; i < ${geometry.volume*descriptor.q}; ++i) { rho_sum += f_next[i]; |