From f1c5164b327054fc540527e973ff7843d4c01996 Mon Sep 17 00:00:00 2001
From: Adrian Kummerlaender
Date: Thu, 24 Oct 2019 14:33:59 +0200
Subject: Add test template for C++, enable switching between AOS and SOA

---
 boltzgen/kernel/generator.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

(limited to 'boltzgen/kernel/generator.py')

diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py
index 42d0430..c9fdb27 100644
--- a/boltzgen/kernel/generator.py
+++ b/boltzgen/kernel/generator.py
@@ -11,7 +11,7 @@ class Generator:
         self.collision  = collision
         self.boundary   = boundary
 
-    def kernel(self, target, precision, geometry):
+    def kernel(self, target, precision, layout, geometry):
         template_path = Path(__file__).parent/('template/basic.' + target + '.mako')
         if not template_path.exists():
             raise Exception("Target '%s' not supported" % target)
@@ -19,6 +19,7 @@ class Generator:
         return Template(filename = str(template_path)).render(
             descriptor = self.descriptor,
             geometry   = geometry,
+            layout     = layout,
 
             moments_subexpr    = self.moments[0],
             moments_assignment = self.moments[1],
-- 
cgit v1.2.3