diff options
Diffstat (limited to 'lid_driven_cavity/cpp/generate.py')
-rwxr-xr-x | lid_driven_cavity/cpp/generate.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/lid_driven_cavity/cpp/generate.py b/lid_driven_cavity/cpp/generate.py index dae2514..973a659 100755 --- a/lid_driven_cavity/cpp/generate.py +++ b/lid_driven_cavity/cpp/generate.py @@ -20,13 +20,16 @@ generator = Generator( precision = config.precision, streaming = config.streaming, index = 'XYZ', - layout = 'AOS') + layout = 'SOA') if args.output is None: args.output = '.' functions = ['collide_and_stream', 'equilibrilize', 'collect_moments', 'momenta_boundary'] +if config.streaming == 'SSS': + functions += ['update_sss_control_structure'] + with open('%s/kernel.h' % args.output, 'w') as kernel: kernel.write(generator.kernel(config.geometry, functions)) |