aboutsummaryrefslogtreecommitdiff
path: root/lid_driven_cavity/cpp/generate.py
diff options
context:
space:
mode:
Diffstat (limited to 'lid_driven_cavity/cpp/generate.py')
-rwxr-xr-xlid_driven_cavity/cpp/generate.py5
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))