From 9069cc00e86ab97acc681d9c3ef80fe8b4bf37a4 Mon Sep 17 00:00:00 2001 From: Martin Bauer <martin.bauer@fau.de> Date: Fri, 26 Jul 2019 12:16:29 +0200 Subject: [PATCH] Adapt to changes in pystencils --- pystencils_walberla/jinja_filters.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/pystencils_walberla/jinja_filters.py b/pystencils_walberla/jinja_filters.py index 9ce43a8..7cc2d25 100644 --- a/pystencils_walberla/jinja_filters.py +++ b/pystencils_walberla/jinja_filters.py @@ -2,7 +2,8 @@ import jinja2 import sympy as sp from pystencils import TypedSymbol -from pystencils.backends.cbackend import CustomSympyPrinter, generate_c +from pystencils.backends.cbackend import generate_c +from pystencils.backends.cuda_backend import CudaSympyPrinter from pystencils.data_types import get_base_type from pystencils.field import FieldType from pystencils.kernelparameters import SHAPE_DTYPE @@ -263,7 +264,7 @@ def generate_call(ctx, kernel_info, ghost_layers_to_include=0, cell_interval=Non spatial_shape_symbols = [TypedSymbol(s, SHAPE_DTYPE) for s in spatial_shape_symbols] indexing_dict = ast.indexing.call_parameters(spatial_shape_symbols) - sp_printer_c = CustomSympyPrinter(dialect='cuda') + sp_printer_c = CudaSympyPrinter() kernel_call_lines += [ "dim3 _block(int(%s), int(%s), int(%s));" % tuple(sp_printer_c.doprint(e) for e in indexing_dict['block']), "dim3 _grid(int(%s), int(%s), int(%s));" % tuple(sp_printer_c.doprint(e) for e in indexing_dict['grid']), -- GitLab