Skip to content
Snippets Groups Projects
Commit f86bbcb0 authored by Markus Holzer's avatar Markus Holzer Committed by Michael Kuron
Browse files

[BugFix] PackInfo generation

parent 59b1775e
Branches
Tags
No related merge requests found
from collections import defaultdict
from lbmpy.stencils import get_stencil
from lbmpy.advanced_streaming.utility import Timestep, get_accessor, get_timesteps
from lbmpy.advanced_streaming.communication import _extend_dir
from pystencils.stencil import inverse_direction
......@@ -59,10 +60,12 @@ def generate_lb_pack_info(generation_context,
for comm_dir in comm_directions(comm_direction):
common_spec[(comm_dir,)].add(fa.field.center(*fa.index))
full_stencil = get_stencil('D3Q27') if len(stencil[0]) == 3 else get_stencil('D2Q9')
for t in timesteps:
spec = common_spec.copy()
write_accesses = get_accessor(streaming_pattern, t).write(pdf_field, stencil)
for comm_dir in stencil:
for comm_dir in full_stencil:
if all(d == 0 for d in comm_dir):
continue
......
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