Expression: Better way to "clone" non-typed symbols in mapper

ecmwf-ifs · Oct 31, 2024 · 7f0c0ed · 7f0c0ed
1 parent 6ebcd89
commit 7f0c0ed
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/loki/expression/mappers.py b/loki/expression/mappers.py
@@ -522,7 +522,10 @@ def _rebuild(expr):
         """ Utility to safely rebuild any symbol """
         if hasattr(expr, 'clone'):
             return expr.clone()
-        return deepcopy(expr)
+
+        # Re-create symbol Pymbolic-style
+        cargs = dict(zip(expr.init_arg_names, expr.__getinitargs__()))
+        return type(expr)(**cargs)
 
     def __call__(self, expr, *args, **kwargs):
         if expr is None: