From 00e822686bcdf5f4313af8448b9d778b89b272bb Mon Sep 17 00:00:00 2001
From: Frederik Hennig <frederik.hennig@fau.de>
Date: Fri, 25 Apr 2025 14:05:51 +0200
Subject: [PATCH] remove TODOs; fix vector CPU detection in lbstep
---
src/lbmpy/creationfunctions.py | 7 +------
src/lbmpy/lbstep.py | 4 ++--
2 files changed, 3 insertions(+), 8 deletions(-)
diff --git a/src/lbmpy/creationfunctions.py b/src/lbmpy/creationfunctions.py
index 86084d79..417ee93f 100644
--- a/src/lbmpy/creationfunctions.py
+++ b/src/lbmpy/creationfunctions.py
@@ -575,11 +575,6 @@ def create_lb_ast(update_rule=None, lbm_config=None, lbm_optimisation=None, conf
update_rule = create_lb_update_rule(lbm_config.collision_rule, lbm_config=lbm_config,
lbm_optimisation=lbm_optimisation, config=config)
- # field_types = set(fa.field.dtype for fa in update_rule.defined_symbols if isinstance(fa, Field.Access))
-
- # TODO nbackend: Implicit type collation should not happen. Either all fields have the same type,
- # or the user must explicitly include type casts where necessary.
- # config = replace(config, data_type=collate_types(field_types), ghost_layers=1)
config = replace(config, ghost_layers=1)
ast = create_kernel(update_rule, config=config)
@@ -607,7 +602,7 @@ def create_lb_update_rule(collision_rule=None, lbm_config=None, lbm_optimisation
lb_method = collision_rule.method
if IS_PYSTENCILS_2:
- fallback_field_data_type = config.get_option("default_dtype") # TODO nbackend: clarify this
+ fallback_field_data_type = config.get_option("default_dtype")
else:
fallback_field_data_type = config.data_type[lbm_config.field_name].numpy_dtype
diff --git a/src/lbmpy/lbstep.py b/src/lbmpy/lbstep.py
index 0f57dc37..52c55bad 100644
--- a/src/lbmpy/lbstep.py
+++ b/src/lbmpy/lbstep.py
@@ -101,8 +101,8 @@ class LatticeBoltzmannStep:
alignment = False
if IS_PYSTENCILS_2:
- # TODO nbackend: get alignment
- pass
+ if config.get_target().is_vector_cpu() and config.cpu.vectorize.enable:
+ alignment = alignment_if_vectorized
else:
if config.backend == Backend.C and config.cpu_vectorize_info:
alignment = alignment_if_vectorized
--
GitLab