From f1c5164b327054fc540527e973ff7843d4c01996 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Thu, 24 Oct 2019 14:33:59 +0200
Subject: Add test template for C++, enable switching between AOS and SOA

---
 boltzgen.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'boltzgen.py')

diff --git a/boltzgen.py b/boltzgen.py
index 27a34fa..14aa292 100644
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -5,6 +5,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 "opencl" or "cpp")')
+argparser.add_argument('layout',   help = 'Memory layout ("aos" or "soa" for C++, ignored for OpenCL')
 
 args = argparser.parse_args()
 
@@ -14,7 +15,7 @@ generator = Generator(
     moments    = lbm.moments(),
     collision  = lbm.bgk(f_eq = lbm.equilibrium(), tau = 0.6))
 
-geometry = Geometry(32,32)
+geometry = Geometry(1024,1024)
 
-src = generator.kernel(args.language, 'float', geometry)
+src = generator.kernel(args.language, 'double', args.layout, geometry)
 print(src)
-- 
cgit v1.2.3