diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py index 9dc3928b3549e7d0a9eae4ae2e78244e4e9e4b4c..c5ff43fb96d74bcc21f3c5a1b212d8635b5fa2ce 100644 --- a/src/pystencils/backend/kernelcreation/freeze.py +++ b/src/pystencils/backend/kernelcreation/freeze.py @@ -13,7 +13,7 @@ from ...sympyextensions import ( integer_functions, ConditionalFieldAccess, ) -from ...reduction_op_mapping import reduction_op_to_expr +from ..reduction_op_mapping import reduction_op_to_expr from ...sympyextensions.typed_sympy import TypedSymbol, TypeCast, DynamicType from ...sympyextensions.pointers import AddressOf, mem_acc from ...sympyextensions.reduction import ReductionAssignment, ReductionOp diff --git a/src/pystencils/backend/platforms/cuda.py b/src/pystencils/backend/platforms/cuda.py index 8c3cd45faf2a210bf8de16e3f25d9ba7897d4e0f..05e95011dabb9e475d608e292b971dbfa9625c7b 100644 --- a/src/pystencils/backend/platforms/cuda.py +++ b/src/pystencils/backend/platforms/cuda.py @@ -23,7 +23,7 @@ from ..constants import PsConstant from ..exceptions import MaterializationError from ..functions import NumericLimitsFunctions, CFunction from ..literals import PsLiteral -from ...reduction_op_mapping import reduction_op_to_expr +from ..reduction_op_mapping import reduction_op_to_expr from ...sympyextensions import ReductionOp from ...types import PsType, PsIeeeFloatType, PsCustomType, PsPointerType, PsScalarType from ...types.quick import SInt, UInt diff --git a/src/pystencils/backend/platforms/generic_cpu.py b/src/pystencils/backend/platforms/generic_cpu.py index 3de7cf696ff47d65950e376e4aa38c63e1528dcf..4f8b562fa6fc3a69c95f5a0a7cffec3a78686bec 100644 --- a/src/pystencils/backend/platforms/generic_cpu.py +++ b/src/pystencils/backend/platforms/generic_cpu.py @@ -13,7 +13,7 @@ from ..functions import ( PsReductionFunction, ) from ..literals import PsLiteral -from ...reduction_op_mapping import reduction_op_to_expr +from ..reduction_op_mapping import reduction_op_to_expr from ...sympyextensions import ReductionOp from ...types import PsIntegerType, PsIeeeFloatType, PsScalarType, PsPointerType diff --git a/src/pystencils/reduction_op_mapping.py b/src/pystencils/backend/reduction_op_mapping.py similarity index 82% rename from src/pystencils/reduction_op_mapping.py rename to src/pystencils/backend/reduction_op_mapping.py index 06fb8aa3e981d661f8e4226a34307f03cd6d9a70..876912acdcd2e05a20de07d37c4b3e462c422e0a 100644 --- a/src/pystencils/reduction_op_mapping.py +++ b/src/pystencils/backend/reduction_op_mapping.py @@ -1,7 +1,7 @@ -from .backend.ast.expressions import PsExpression, PsCall, PsAdd, PsSub, PsMul, PsDiv -from .backend.exceptions import FreezeError -from .backend.functions import PsMathFunction, MathFunctions -from .sympyextensions.reduction import ReductionOp +from .ast.expressions import PsExpression, PsCall, PsAdd, PsSub, PsMul, PsDiv +from .exceptions import FreezeError +from .functions import PsMathFunction, MathFunctions +from ..sympyextensions.reduction import ReductionOp _available_operator_interface: set[ReductionOp] = { ReductionOp.Add,