Skip to content
Snippets Groups Projects
Commit 8a3ac66b authored by Helen Schottenhamml's avatar Helen Schottenhamml
Browse files

Merge branch 'Fix_output_field_plus_split' into 'master'

Fix split error

See merge request !33
parents f19f0b1b eeb6c436
Branches
Tags
1 merge request!33Fix split error
Pipeline #24579 passed
...@@ -2,4 +2,4 @@ ...@@ -2,4 +2,4 @@
max-line-length=120 max-line-length=120
exclude=lbmpy/plot.py exclude=lbmpy/plot.py
lbmpy/session.py lbmpy/session.py
ignore = W293 W503 W291 ignore = W293 W503 W291 E741
...@@ -322,6 +322,11 @@ def create_lb_collision_rule(lb_method=None, optimization={}, **kwargs): ...@@ -322,6 +322,11 @@ def create_lb_collision_rule(lb_method=None, optimization={}, **kwargs):
else: else:
collision_rule = lb_method.get_collision_rule(keep_rrs_symbolic=keep_rrs_symbolic) collision_rule = lb_method.get_collision_rule(keep_rrs_symbolic=keep_rrs_symbolic)
if params['output'] and params['kernel_type'] == 'stream_pull_collide':
cqc = lb_method.conserved_quantity_computation
output_eqs = cqc.output_equations_from_pdfs(lb_method.pre_collision_pdf_symbols, params['output'])
collision_rule = collision_rule.new_merged(output_eqs)
if opt_params['simplification'] == 'auto': if opt_params['simplification'] == 'auto':
simplification = create_simplification_strategy(lb_method, split_inner_loop=split_inner_loop) simplification = create_simplification_strategy(lb_method, split_inner_loop=split_inner_loop)
else: else:
...@@ -358,11 +363,6 @@ def create_lb_collision_rule(lb_method=None, optimization={}, **kwargs): ...@@ -358,11 +363,6 @@ def create_lb_collision_rule(lb_method=None, optimization={}, **kwargs):
if cse_global: if cse_global:
collision_rule = sympy_cse(collision_rule) collision_rule = sympy_cse(collision_rule)
if params['output'] and params['kernel_type'] == 'stream_pull_collide':
cqc = lb_method.conserved_quantity_computation
output_eqs = cqc.output_equations_from_pdfs(lb_method.pre_collision_pdf_symbols, params['output'])
collision_rule = collision_rule.new_merged(output_eqs)
return collision_rule return collision_rule
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment