diff --git a/src/lbmpy/creationfunctions.py b/src/lbmpy/creationfunctions.py index 86084d79b6867e37deabfe26de7c23d538bd2fc1..417ee93fd45cbf183ac0fae8dd932975ec051b9c 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 0f57dc376dfc4b502537fa8c92b9353005043b80..52c55bad5c0c71f318bbd5fe2d6cec70f11e6736 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