aboutsummaryrefslogtreecommitdiff
path: root/boltzgen.py
diff options
context:
space:
mode:
Diffstat (limited to 'boltzgen.py')
-rw-r--r--boltzgen.py12
1 files changed, 4 insertions, 8 deletions
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)