From 841c50778400e95e275e741642b11f5d2af16e75 Mon Sep 17 00:00:00 2001 From: Andreas Kloeckner Date: Sun, 11 Sep 2022 14:10:11 -0500 Subject: [PATCH] Fix allocator type annotations in grudge.actx --- grudge/array_context.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/grudge/array_context.py b/grudge/array_context.py index 7e4942576..169919b6f 100644 --- a/grudge/array_context.py +++ b/grudge/array_context.py @@ -113,7 +113,7 @@ class PyOpenCLArrayContext(_PyOpenCLArrayContextBase): any, for now.) """ 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: @@ -430,7 +430,7 @@ class MPIPyOpenCLArrayContext(PyOpenCLArrayContext, MPIBasedArrayContext): def __init__(self, mpi_communicator, 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: """