From 3c8941068d0b15e03f95b0aa86226aaf102d7c85 Mon Sep 17 00:00:00 2001
From: Daniel Bauer <daniel.j.bauer@fau.de>
Date: Fri, 18 Aug 2023 10:17:12 +0200
Subject: [PATCH] fall back to symbols_defined (like the old implementation)
---
pystencils/transformations.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/pystencils/transformations.py b/pystencils/transformations.py
index 4e3d3862c..e07d871e9 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}
--
GitLab