aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--boltzgen.py2
-rw-r--r--boltzgen/kernel/generator.py5
-rw-r--r--boltzgen/kernel/template/basic.cpp.mako2
3 files changed, 6 insertions, 3 deletions
diff --git a/boltzgen.py b/boltzgen.py
index e5a8588..b487b07 100644
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -6,7 +6,7 @@ from boltzgen import *
argparser = argparse.ArgumentParser(description='Generate LBM kernels in various languages using a symbolic description.')
argparser.add_argument('language', help = 'Target language (currently either "cl" or "cpp")')
argparser.add_argument('--layout', dest = 'layout', help = 'Memory layout ("AOS" or "SOA")', required = True)
-argparser.add_argument('--precision', dest = 'precision', help = 'Floating precision ("float" or "double")', required = True)
+argparser.add_argument('--precision', dest = 'precision', help = 'Floating precision ("single" or "double")', required = True)
args = argparser.parse_args()
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];