Minor adaptation on how symbols are given reduction property

src/pystencils/backend/kernelcreation/context.py

@@ -179,13 +179,14 @@ class KernelCreationContext:
         """
         if self.find_symbol(symbol.name) is None:
             raise PsInternalCompilerError(
-                "add_reduction_to_symbol: Symbol does not exist in the symbol table"
+                f"add_reduction_to_symbol: {symbol.name} does not exist in the symbol table"
             )
 
         if symbol not in self._symbols_with_reduction and not symbol.get_properties(ReductionSymbolProperty):
+            symbol.add_property(reduction)
             self._symbols_with_reduction[symbol] = reduction
         else:
-            raise PsInternalCompilerError(f"add_reduction_to_symbol: Symbol {symbol.name} already has a reduction property")
+            raise PsInternalCompilerError(f"add_reduction_to_symbol: {symbol.name} already has a reduction property")
 
     def duplicate_symbol(
         self, symb: PsSymbol, new_dtype: PsType | None = None

src/pystencils/backend/kernelcreation/freeze.py

@@ -213,7 +213,7 @@ class FreezeExpressions:
             case _:
                 raise FreezeError(f"Unsupported reduced assignment: {expr.op}.")
 
-        lhs.symbol.add_property(ReductionSymbolProperty(expr.op, init_val))
+        self._ctx.add_reduction_to_symbol(lhs.symbol, ReductionSymbolProperty(expr.op, init_val))
 
         return PsAssignment(lhs, op(lhs.clone(), rhs))