Add support for SSS pattern to C++ LDC example

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))