diff options
| author | Adrian Kummerlaender | 2019-10-24 14:33:59 +0200 |
|---|---|---|
| committer | Adrian Kummerlaender | 2019-10-24 14:33:59 +0200 |
| commit | f1c5164b327054fc540527e973ff7843d4c01996 (patch) | |
| tree | f181f050d5a6299efeab8ead91936cccc076babb /boltzgen.py | |
| parent | 2e07d5fac4d3fedcd2cc9ccbab27f50c69f65983 (diff) | |
| download | boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar.gz boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar.bz2 boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar.lz boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar.xz boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.tar.zst boltzgen-f1c5164b327054fc540527e973ff7843d4c01996.zip | |
Add test template for C++, enable switching between AOS and SOA
Diffstat (limited to 'boltzgen.py')
| -rw-r--r-- | boltzgen.py | 5 |
1 files changed, 3 insertions, 2 deletions
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) |