diff --git a/src/pystencils/jit/gpu_cupy.py b/src/pystencils/jit/gpu_cupy.py index c208ac2196151d079ca5081f1377c55d18a9393c..467e86be708962f363ac71c9fc7ef38342cf49cf 100644 --- a/src/pystencils/jit/gpu_cupy.py +++ b/src/pystencils/jit/gpu_cupy.py @@ -11,7 +11,7 @@ except ImportError: from ..codegen import Target from ..field import FieldType -from ..types import PsType +from ..types import PsType, PsPointerType from .jit import JitBase, JitError, KernelWrapper from ..codegen import ( Kernel, @@ -183,6 +183,9 @@ class CupyKernelWrapper(KernelWrapper): kparam.dtype, ) break + elif isinstance(kparam.dtype, PsPointerType): + val = kwargs[kparam.name] + args.append(val) else: # scalar parameter val: Any = kwargs[kparam.name]