remove TODOs; fix vector CPU detection in lbstep

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
 

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