diff options
author | Adrian Kummerlaender | 2019-10-26 15:01:39 +0200 |
---|---|---|
committer | Adrian Kummerlaender | 2019-10-26 15:01:39 +0200 |
commit | 1bf577b1c5e606ac2c0553857297ce8c0c04ccb7 (patch) | |
tree | 8ad97c3cd8549c1cee3b52a460f1b8c1e4e07cfe /boltzgen/kernel/generator.py | |
parent | 71e0d07ec0cdf528eb5d962ce97bd6c55da52eae (diff) | |
download | boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar.gz boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar.bz2 boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar.lz boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar.xz boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.tar.zst boltzgen-1bf577b1c5e606ac2c0553857297ce8c0c04ccb7.zip |
Support passing additional string arguments to the generator
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 ) |