diff --git a/arraycontext/impl/pyopencl/__init__.py b/arraycontext/impl/pyopencl/__init__.py index 9ad36251adeee03940afa21d2bab82bcb28db8d5..71b04c721552547a2389bdd95d22bea6bee9c8d2 100644 --- a/arraycontext/impl/pyopencl/__init__.py +++ b/arraycontext/impl/pyopencl/__init__.py @@ -74,7 +74,7 @@ class PyOpenCLArrayContext(ArrayContext): def __init__(self, queue: "pyopencl.CommandQueue", - allocator: Optional["pyopencl.tools.AllocatorInterface"] = None, + allocator: Optional["pyopencl.tools.AllocatorBase"] = None, wait_event_queue_length: Optional[int] = None, force_device_scalars: bool = False) -> None: r"""