REF: Change type hints so Buffer.handle can be int, object, or None

Author: carterbox

cuda_core/cuda/core/experimental/_memory.py

@@ -6,7 +6,7 @@
 
 import abc
 import weakref
-from typing import Optional, Tuple, TypeVar
+from typing import Optional, Tuple, TypeVar, Union
 
 from cuda.core.experimental._dlpack import DLDeviceType, make_py_capsule
 from cuda.core.experimental._stream import default_stream
@@ -18,6 +18,9 @@
 # TODO: define a memory property mixin class and make Buffer and
 # MemoryResource both inherit from it
 
+DevicePointerT = Union[driver.CUdeviceptr, int, None]
+"""A type union of `Cudeviceptr`, `int` and `None` for hinting Buffer.handle."""
+
 
 class Buffer:
     """Represent a handle to allocated memory.
@@ -81,7 +84,7 @@ def close(self, stream=None):
         self._mnff.close(stream)
 
     @property
-    def handle(self) -> driver.CUdeviceptr:
+    def handle(self) -> DevicePointerT:
         """Return the buffer handle object.
 
         .. caution::