From 913d20406df870946dfb2fa62a424a714a3985cc Mon Sep 17 00:00:00 2001
From: Martin Bauer <martin.bauer@fau.de>
Date: Thu, 9 May 2019 16:15:12 +0200
Subject: [PATCH] Optional parameters to be passed to update rule (entropic,
smagorinsky...)
---
lbmpy_walberla/walberla_lbm_generation.py | 5 +++--
1 file changed, 3 insertions(+), 2 deletions(-)
diff --git a/lbmpy_walberla/walberla_lbm_generation.py b/lbmpy_walberla/walberla_lbm_generation.py
index 5e094a1..90e0550 100644
--- a/lbmpy_walberla/walberla_lbm_generation.py
+++ b/lbmpy_walberla/walberla_lbm_generation.py
@@ -24,7 +24,7 @@ cpp_printer = CustomSympyPrinter(dialect='c')
REFINEMENT_SCALE_FACTOR = sp.Symbol("level_scale_factor")
-def generate_lattice_model(generation_context, class_name, lb_method, refinement_scaling=None,
+def generate_lattice_model(generation_context, class_name, lb_method, refinement_scaling=None, update_rule_params={},
**create_kernel_params):
# usually a numpy layout is chosen by default i.e. xyzf - which is bad for waLBerla where at least the spatial
@@ -36,7 +36,8 @@ def generate_lattice_model(generation_context, class_name, lb_method, refinement
raise ValueError("Lattice Models can only be generated for CPUs. To generate LBM on GPUs use sweeps directly")
is_float = not generation_context.double_accuracy
- params, opt_params = update_with_default_parameters({'lb_method': lb_method}, optimization)
+ update_rule_params['lb_method'] = lb_method
+ params, opt_params = update_with_default_parameters(update_rule_params, optimization)
stencil_name = get_stencil_name(lb_method.stencil)
if not stencil_name:
--
GitLab