diff --git a/src/pystencils/sympyextensions/__init__.py b/src/pystencils/sympyextensions/__init__.py index 7431416c9eb9bcd4433dab76c32fb1b755501105..6ab24e936a2355782755badbf835d7e5c3bee73e 100644 --- a/src/pystencils/sympyextensions/__init__.py +++ b/src/pystencils/sympyextensions/__init__.py @@ -1,6 +1,7 @@ from .astnodes import ConditionalFieldAccess from .typed_sympy import TypedSymbol, CastFunc from .pointers import mem_acc +from .reduction import reduced_assign from .math import ( prod, @@ -33,6 +34,7 @@ from .math import ( __all__ = [ "ConditionalFieldAccess", + "reduced_assign", "TypedSymbol", "CastFunc", "mem_acc",