Skip to content
Snippets Groups Projects
Commit 9069cc00 authored by Martin Bauer's avatar Martin Bauer
Browse files

Adapt to changes in pystencils

parent 45c04582
No related merge requests found
...@@ -2,7 +2,8 @@ import jinja2 ...@@ -2,7 +2,8 @@ import jinja2
import sympy as sp import sympy as sp
from pystencils import TypedSymbol 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.data_types import get_base_type
from pystencils.field import FieldType from pystencils.field import FieldType
from pystencils.kernelparameters import SHAPE_DTYPE from pystencils.kernelparameters import SHAPE_DTYPE
...@@ -263,7 +264,7 @@ def generate_call(ctx, kernel_info, ghost_layers_to_include=0, cell_interval=Non ...@@ -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] spatial_shape_symbols = [TypedSymbol(s, SHAPE_DTYPE) for s in spatial_shape_symbols]
indexing_dict = ast.indexing.call_parameters(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 += [ kernel_call_lines += [
"dim3 _block(int(%s), int(%s), int(%s));" % tuple(sp_printer_c.doprint(e) for e in indexing_dict['block']), "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']), "dim3 _grid(int(%s), int(%s), int(%s));" % tuple(sp_printer_c.doprint(e) for e in indexing_dict['grid']),
......
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