From 4c4c98eb2b2ab0a636ecf082f3ff56e8e1a2099c Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 30 Oct 2020 10:27:30 +0100
Subject: [PATCH] Allow kernel_creation_args in
create_lattice_boltzmann_boundary_kernel
---
lbmpy/boundaries/boundaryhandling.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/lbmpy/boundaries/boundaryhandling.py b/lbmpy/boundaries/boundaryhandling.py
index 575a046d..b9761abc 100644
--- a/lbmpy/boundaries/boundaryhandling.py
+++ b/lbmpy/boundaries/boundaryhandling.py
@@ -97,11 +97,11 @@ class LbmWeightInfo(CustomCodeNode):
def create_lattice_boltzmann_boundary_kernel(pdf_field, index_field, lb_method, boundary_functor,
- target='cpu', openmp=True):
+ target='cpu', openmp=True, **kernel_creation_args):
elements = [BoundaryOffsetInfo(lb_method.stencil), LbmWeightInfo(lb_method)]
index_arr_dtype = index_field.dtype.numpy_dtype
dir_symbol = TypedSymbol("dir", index_arr_dtype.fields['dir'][0])
elements += [Assignment(dir_symbol, index_field[0]('dir'))]
elements += boundary_functor(pdf_field=pdf_field, direction_symbol=dir_symbol,
lb_method=lb_method, index_field=index_field)
- return create_indexed_kernel(elements, [index_field], target=target, cpu_openmp=openmp)
+ return create_indexed_kernel(elements, [index_field], target=target, cpu_openmp=openmp, **kernel_creation_args)
--
GitLab