From bd6b63abbd916ef538714954dbc788bd118b263b Mon Sep 17 00:00:00 2001 From: Frederik Hennig <frederik.hennig@fau.de> Date: Fri, 26 Jul 2024 14:02:37 +0200 Subject: [PATCH] fix types and linting --- src/pystencils/backend/kernelcreation/cpu_optimization.py | 4 ++-- src/pystencils/backend/kernelcreation/typification.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/src/pystencils/backend/kernelcreation/cpu_optimization.py b/src/pystencils/backend/kernelcreation/cpu_optimization.py index be9e50af3..46fef6603 100644 --- a/src/pystencils/backend/kernelcreation/cpu_optimization.py +++ b/src/pystencils/backend/kernelcreation/cpu_optimization.py @@ -22,8 +22,8 @@ def optimize_cpu( canonicalize = CanonicalizeSymbols(ctx, True) kernel_ast = cast(PsBlock, canonicalize(kernel_ast)) - # hoist_invariants = HoistLoopInvariantDeclarations(ctx) - # kernel_ast = cast(PsBlock, hoist_invariants(kernel_ast)) + hoist_invariants = HoistLoopInvariantDeclarations(ctx) + kernel_ast = cast(PsBlock, hoist_invariants(kernel_ast)) if cfg is None: return kernel_ast diff --git a/src/pystencils/backend/kernelcreation/typification.py b/src/pystencils/backend/kernelcreation/typification.py index 2e81a0779..3f69e1217 100644 --- a/src/pystencils/backend/kernelcreation/typification.py +++ b/src/pystencils/backend/kernelcreation/typification.py @@ -24,7 +24,6 @@ from ..ast.structural import ( PsExpression, PsAssignment, PsDeclaration, - PsStatement, PsEmptyLeafMixIn, ) from ..ast.expressions import ( @@ -704,6 +703,7 @@ class Typifier: ) items_tc.apply_dtype(deconstify(elem_type)) tc.infer_dtype(expr) + return None case DereferencableTo(elem_type_or_hint): if isinstance(elem_type_or_hint, PsType): -- GitLab