Generalize stream only kernel
All threads resolved!
All threads resolved!
Files
3+ 33
− 6
@@ -54,13 +55,27 @@ def create_lbm_kernel(collision_rule, input_field, output_field, accessor):
@@ -54,13 +55,27 @@ def create_lbm_kernel(collision_rule, input_field, output_field, accessor):
def create_stream_pull_only_kernel(stencil, numpy_arr=None, src_field_name="src", dst_field_name="dst",
@@ -71,15 +86,27 @@ def create_stream_pull_only_kernel(stencil, numpy_arr=None, src_field_name="src"
@@ -71,15 +86,27 @@ def create_stream_pull_only_kernel(stencil, numpy_arr=None, src_field_name="src"
stream_assignments = [Assignment(a, b) for a, b in zip(streamed, accessor.read(src_field, stencil))]