From 96b5cbf286a29882b496faee3b9fe3be481d8bb3 Mon Sep 17 00:00:00 2001 From: zy69guqi <richard.angersbach@fau.de> Date: Tue, 28 Jan 2025 12:44:21 +0100 Subject: [PATCH] Fix lint --- src/pystencils/backend/kernelcreation/freeze.py | 5 ++--- src/pystencils/codegen/driver.py | 2 +- 2 files changed, 3 insertions(+), 4 deletions(-) diff --git a/src/pystencils/backend/kernelcreation/freeze.py b/src/pystencils/backend/kernelcreation/freeze.py index f5f207acf..1238f16af 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 8b8ecd15b..b47ad8a9e 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 -- GitLab