diff --git a/gpucuda/periodicity.py b/gpucuda/periodicity.py index 06b22b0a0042d6d3209b9adc4c2fb746004e9dce..d48aa900aa461050d9c934b4a55280b9800bb07d 100644 --- a/gpucuda/periodicity.py +++ b/gpucuda/periodicity.py @@ -33,7 +33,7 @@ def getPeriodicBoundaryFunctor(stencil, domainSize, indexDimensions=0, indexDimS for srcSlice, dstSlice in srcDstSliceTuples: kernels.append(createCopyKernel(domainSize, srcSlice, dstSlice, indexDimensions, indexDimShape)) - def functor(pdfs): + def functor(pdfs, **kwargs): for kernel in kernels: kernel(pdfs=pdfs) diff --git a/slicing.py b/slicing.py index b81f844982a8833f1e5445227bf2db7a71cebebe..6a12b3d579a70c9bb4ed0a384b8af68b8fee8c55 100644 --- a/slicing.py +++ b/slicing.py @@ -185,7 +185,7 @@ def getPeriodicBoundaryFunctor(stencil, ghostLayers=1, thickness=None): """ srcDstSliceTuples = getPeriodicBoundarySrcDstSlices(stencil, ghostLayers, thickness) - def functor(pdfs): + def functor(pdfs, **kwargs): for srcSlice, dstSlice in srcDstSliceTuples: pdfs[dstSlice] = pdfs[srcSlice]