diff --git a/src/pystencils/backend/kernelcreation/context.py b/src/pystencils/backend/kernelcreation/context.py index b9df6f6826c6575974f061e89b3b7ef0ae03cbe3..686646815f41330c1430d4cae12692bae67ee011 100644 --- a/src/pystencils/backend/kernelcreation/context.py +++ b/src/pystencils/backend/kernelcreation/context.py @@ -9,6 +9,8 @@ from ...defaults import DEFAULTS from ...field import Field, FieldType from ...sympyextensions.typed_sympy import TypedSymbol, DynamicType +from ...codegen.properties import ReductionSymbolProperty + from ..memory import PsSymbol, PsBuffer from ..constants import PsConstant from ...types import ( diff --git a/src/pystencils/codegen/properties.py b/src/pystencils/codegen/properties.py index cc4ff4101e9762b7bce6750eed3fef1bdc8151a9..2b0af986aa2ab74fc03ebf534a448adaca25c3ea 100644 --- a/src/pystencils/codegen/properties.py +++ b/src/pystencils/codegen/properties.py @@ -2,9 +2,7 @@ from __future__ import annotations from dataclasses import dataclass from ..field import Field - -from ..backend.ast.expressions import PsExpression - +from typing import Any @dataclass(frozen=True) class PsSymbolProperty: @@ -21,7 +19,7 @@ class ReductionSymbolProperty(UniqueSymbolProperty): """Symbol acts as a base pointer to a field.""" op: str - init_val: PsExpression + init_val: Any # TODO: type? @dataclass(frozen=True)