diff --git a/pystencils/transformations.py b/pystencils/transformations.py index 4e3d3862c874df879ec5c021e429cc14709f75e9..e07d871e97d413df0f8a9087a5bfbdc590b8067b 100644 --- a/pystencils/transformations.py +++ b/pystencils/transformations.py @@ -607,8 +607,8 @@ def move_constants_before_loop(ast_node): elif isinstance(node, ast.KernelFunction): return False else: - raise NotImplementedError(f'Due to defensive programming we handle only specific expressions.\n' - f'The expression {node} of type {type(node)} is not known yet.') + defs = {s.name for s in node.symbols_defined} + return bool(symbol_names.intersection(defs)) dependencies = {s.name for s in node.undefined_symbols}