diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py index f5f207acf303542c2479f7956f519cbce21edf2e..1238f16af8d7ae9619e664526c86df897e81b2a9 100644 --- a/src/pystencils/backend/kernelcreation/freeze.py +++ b/src/pystencils/backend/kernelcreation/freeze.py @@ -1,9 +1,8 @@ from typing import overload, cast, Any from functools import reduce -from operator import add, mul, sub, truediv +from operator import add, mul, sub import sympy as sp -import sympy.core.relational import sympy.logic.boolalg from sympy.codegen.ast import AssignmentBase, AugmentedAssignment @@ -184,7 +183,7 @@ class FreezeExpressions: assert isinstance(lhs, PsSymbolExpr) orig_lhs_symb = lhs.symbol - dtype = rhs.dtype # TODO: kernel with (implicit) up/downcasts? + dtype = rhs.dtype # TODO: kernel with (implicit) up/downcasts? # replace original symbol with pointer-based type used for export orig_lhs_symb_as_ptr = PsSymbol(orig_lhs_symb.name, PsPointerType(dtype)) diff --git a/src/pystencils/codegen/driver.py b/src/pystencils/codegen/driver.py index 8b8ecd15b4dc28e8f2ca4dc8fb7866e62852288a..b47ad8a9e4dd7c3b624ec7fd003bafce8193752c 100644 --- a/src/pystencils/codegen/driver.py +++ b/src/pystencils/codegen/driver.py @@ -7,7 +7,7 @@ from .config import CreateKernelConfig, OpenMpConfig, VectorizationConfig, AUTO from .kernel import Kernel, GpuKernel, GpuThreadsRange from .properties import PsSymbolProperty, FieldShape, FieldStride, FieldBasePtr, ReductionPointerVariable from .parameters import Parameter -from ..backend.ast.expressions import PsSymbolExpr, PsMemAcc, PsConstantExpr, PsExpression +from ..backend.ast.expressions import PsSymbolExpr, PsMemAcc, PsConstantExpr from ..sympyextensions.binop_mapping import binop_str_to_expr from ..types import create_numeric_type, PsIntegerType, PsScalarType