From 73de5f16efc696cf0c88beec086eb9a4df9098dd Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Thu, 24 Oct 2019 21:52:45 +0200
Subject: Extract offset helper into target and layout specific classes

---
 boltzgen.py | 12 ++++--------
 1 file changed, 4 insertions(+), 8 deletions(-)

(limited to 'boltzgen.py')

diff --git a/boltzgen.py b/boltzgen.py
index 04651e4..e5a8588 100644
--- a/boltzgen.py
+++ b/boltzgen.py
@@ -4,16 +4,12 @@ import string
 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', dest = 'layout',
-    help = 'Memory layout ("aos" or "soa" for C++, ignored for OpenCL')
+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)
 
 args = argparser.parse_args()
 
-if args.language == 'cpp' and args.layout is None:
-    raise Exception('Please specify the memory layout')
-
 lbm = LBM(D2Q9)
 generator = Generator(
     descriptor = D2Q9,
@@ -22,5 +18,5 @@ generator = Generator(
 
 geometry = Geometry(1024,1024)
 
-src = generator.kernel(args.language, 'double', args.layout, geometry)
+src = generator.kernel(args.language, args.precision, args.layout, geometry)
 print(src)
-- 
cgit v1.2.3