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