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

Merge branch 'SerialDataHandling.run_kernel-kwargs' into 'master'

Enable overriding of kwargs in SerialDataHandling.run_kernel

See merge request pycodegen/pystencils!143
parents 2b6ae723 64454993
No related merge requests found
...@@ -9,7 +9,8 @@ from pystencils.datahandling.datahandling_interface import DataHandling ...@@ -9,7 +9,8 @@ from pystencils.datahandling.datahandling_interface import DataHandling
from pystencils.datahandling.pycuda import PyCudaArrayHandler from pystencils.datahandling.pycuda import PyCudaArrayHandler
from pystencils.datahandling.pyopencl import PyOpenClArrayHandler from pystencils.datahandling.pyopencl import PyOpenClArrayHandler
from pystencils.field import ( from pystencils.field import (
Field, FieldType, create_numpy_array_with_layout, layout_string_to_tuple, spatial_layout_string_to_tuple) Field, FieldType, create_numpy_array_with_layout, layout_string_to_tuple,
spatial_layout_string_to_tuple)
from pystencils.slicing import normalize_slice, remove_ghost_layers from pystencils.slicing import normalize_slice, remove_ghost_layers
from pystencils.utils import DotDict from pystencils.utils import DotDict
...@@ -239,7 +240,7 @@ class SerialDataHandling(DataHandling): ...@@ -239,7 +240,7 @@ class SerialDataHandling(DataHandling):
def run_kernel(self, kernel_function, **kwargs): def run_kernel(self, kernel_function, **kwargs):
arrays = self.gpu_arrays if kernel_function.ast.backend in self._GPU_LIKE_BACKENDS else self.cpu_arrays arrays = self.gpu_arrays if kernel_function.ast.backend in self._GPU_LIKE_BACKENDS else self.cpu_arrays
kernel_function(**arrays, **kwargs) kernel_function(**{**arrays, **kwargs})
def get_kernel_kwargs(self, kernel_function, **kwargs): def get_kernel_kwargs(self, kernel_function, **kwargs):
result = {} result = {}
......
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