diff options
author | Adrian Kummerlaender | 2019-10-24 14:33:59 +0200 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-24 14:33:59 +0200 |
commit | f1c5164b327054fc540527e973ff7843d4c01996 (patch) | |
tree | f181f050d5a6299efeab8ead91936cccc076babb /boltzgen/kernel/generator.py | |
parent | 2e07d5fac4d3fedcd2cc9ccbab27f50c69f65983 (diff) | |
download | boltzgen-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.py | 3 |
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], |