diff options
Diffstat (limited to 'boltzgen/kernel/generator.py')
| -rw-r--r-- | boltzgen/kernel/generator.py | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py index 434d163..8d0de37 100644 --- a/boltzgen/kernel/generator.py +++ b/boltzgen/kernel/generator.py @@ -13,7 +13,7 @@ class Generator: self.collision = collision self.boundary = boundary - def kernel(self, target, precision, layout, geometry): + def kernel(self, target, precision, layout, geometry, extras = []): template_path = Path(__file__).parent/('template/basic.' + target + '.mako') if not template_path.exists(): raise Exception("Target '%s' not supported" % target) @@ -34,9 +34,9 @@ class Generator: moments_subexpr = self.moments[0], moments_assignment = self.moments[1], - collision_subexpr = self.collision[0], collision_assignment = self.collision[1], + ccode = sympy.ccode, float_type = { 'single': 'float', @@ -49,5 +49,5 @@ class Generator: float_type = precision ), - ccode = sympy.ccode + extras = extras ) |