Skip to content
Snippets Groups Projects
Commit 8c13106a authored by Martin Bauer's avatar Martin Bauer
Browse files

Merge branch 'staggered_kernel' into 'master'

adapt to new create_staggered_kernel API

See merge request pycodegen/pystencils_walberla!3
parents 4311fd0e 370f658f
Branches
No related merge requests found
...@@ -34,8 +34,7 @@ def generate_sweep(generation_context, class_name, assignments, ...@@ -34,8 +34,7 @@ def generate_sweep(generation_context, class_name, assignments,
field_swaps: sequence of field pairs (field, temporary_field). The generated sweep only gets the first field field_swaps: sequence of field pairs (field, temporary_field). The generated sweep only gets the first field
as argument, creating a temporary field internally which is swapped with the first field after as argument, creating a temporary field internally which is swapped with the first field after
each iteration. each iteration.
staggered: set to True to create staggered kernels, in this case assignments has to be a tuple with arguments staggered: set to True to create staggered kernels with `pystencils.create_staggered_kernel`
to `pystencils.create_staggered_kernel`
varying_parameters: Depending on the configuration, the generated kernels may receive different arguments for varying_parameters: Depending on the configuration, the generated kernels may receive different arguments for
different setups. To not have to adapt the C++ application when then parameter change, different setups. To not have to adapt the C++ application when then parameter change,
the varying_parameters sequence can contain parameter names, which are always expected by the varying_parameters sequence can contain parameter names, which are always expected by
...@@ -52,7 +51,7 @@ def generate_sweep(generation_context, class_name, assignments, ...@@ -52,7 +51,7 @@ def generate_sweep(generation_context, class_name, assignments,
if not staggered: if not staggered:
ast = create_kernel(assignments, **create_kernel_params) ast = create_kernel(assignments, **create_kernel_params)
else: else:
ast = create_staggered_kernel(*assignments, **create_kernel_params) ast = create_staggered_kernel(assignments, **create_kernel_params)
def to_name(f): def to_name(f):
return f.name if isinstance(f, Field) else f return f.name if isinstance(f, Field) else f
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment