aboutsummaryrefslogtreecommitdiff
path: root/boltzgen/kernel/generator.py
diff options
context:
space:
mode:
authorAdrian Kummerlaender2019-10-24 14:33:59 +0200
committerAdrian Kummerlaender2019-10-24 14:33:59 +0200
commitf1c5164b327054fc540527e973ff7843d4c01996 (patch)
treef181f050d5a6299efeab8ead91936cccc076babb /boltzgen/kernel/generator.py
parent2e07d5fac4d3fedcd2cc9ccbab27f50c69f65983 (diff)
downloadboltzgen-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/kernel/generator.py')
-rw-r--r--boltzgen/kernel/generator.py3
1 files changed, 2 insertions, 1 deletions
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],