Skip to content
Snippets Groups Projects
Commit 3962799b authored by Frederik Hennig's avatar Frederik Hennig
Browse files

Fix field duplication in create_lb_update_rule. Fix API in legacy custom_code_nodes.

parent 7f6f3a57
No related branches found
No related tags found
1 merge request!172Changes for compatibility with pystencils 2.0
Pipeline #69874 passed with warnings
......@@ -626,18 +626,9 @@ def create_lb_update_rule(collision_rule=None, lbm_config=None, lbm_optimisation
if lbm_optimisation.symbolic_temporary_field is not None:
dst_field = lbm_optimisation.symbolic_temporary_field
# else:
# FIXME: Field duplication is currently broken in pystencils.
# dst_field = src_field.new_field_with_different_name(lbm_config.temporary_field_name)
elif src_field.has_fixed_shape:
field_size = src_field.shape
dst_field = Field.create_fixed_size(lbm_config.temporary_field_name, field_size, index_dimensions=1,
layout=lbm_optimisation.field_layout, dtype=fallback_field_data_type)
else:
dst_field = Field.create_generic(lbm_config.temporary_field_name, spatial_dimensions=collision_rule.method.dim,
index_shape=(q,), layout=lbm_optimisation.field_layout,
dtype=fallback_field_data_type)
dst_field = src_field.new_field_with_different_name(lbm_config.temporary_field_name)
kernel_type = lbm_config.kernel_type
if kernel_type == 'stream_pull_only':
update_rule = create_stream_pull_with_output_kernel(lb_method, src_field, dst_field, lbm_config.output)
......
......@@ -56,7 +56,7 @@ class MirroredStencilDirections(CustomCodeNode):
def __init__(self, stencil, mirror_axis, dtype=np.int32):
offsets_dtype = create_type(dtype)
mirrored_stencil_symbol = MirroredStencilDirections._mirrored_symbol(mirror_axis)
mirrored_stencil_symbol = MirroredStencilDirections._mirrored_symbol(mirror_axis, stencil)
mirrored_directions = [stencil.index(MirroredStencilDirections.mirror_stencil(direction, mirror_axis))
for direction in stencil]
code = "\n"
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment