diff options
Diffstat (limited to 'boltzgen/kernel/generator.py')
| -rw-r--r-- | boltzgen/kernel/generator.py | 12 |
1 files changed, 6 insertions, 6 deletions
diff --git a/boltzgen/kernel/generator.py b/boltzgen/kernel/generator.py index b2a632c..30e2c04 100644 --- a/boltzgen/kernel/generator.py +++ b/boltzgen/kernel/generator.py @@ -3,23 +3,23 @@ import sympy from mako.template import Template from pathlib import Path -from . import target +from . import memory class Generator: - def __init__(self, descriptor, moments, collision, language, precision, index, layout): + def __init__(self, descriptor, moments, collision, target, precision, index, layout): self.descriptor = descriptor self.moments = moments self.collision = collision - self.target = language - self.float_type = eval("target.precision.%s" % language).get_float_type(precision) + self.target = target + self.float_type = eval("memory.precision.%s" % target).get_float_type(precision) try: - self.index_impl = eval("target.index.%s" % index) + self.index_impl = eval("memory.index.%s" % index) except AttributeError: raise Exception("There is no cell indexing scheme '%s'" % index) from None try: - self.layout_impl = eval("target.layout.%s" % layout) + self.layout_impl = eval("memory.layout.%s" % layout) except AttributeError: raise Exception("There is no layout '%s'" % layout) from None |