improve type hints to make Ruff happy

Author: leofang

cuda_core/cuda/core/experimental/_event.py

@@ -2,12 +2,17 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+from __future__ import annotations
+
 import weakref
 from dataclasses import dataclass
-from typing import Optional
+from typing import TYPE_CHECKING, Optional
 
 from cuda.core.experimental._utils import CUDAError, check_or_create_options, driver, handle_return
 
+if TYPE_CHECKING:
+    import cuda.bindings
+
 
 @dataclass
 class EventOptions:
@@ -130,6 +135,6 @@ def is_done(self) -> bool:
             raise CUDAError(f"unexpected error: {result}")
 
     @property
-    def handle(self) -> "CUevent":
+    def handle(self) -> cuda.bindings.driver.CUevent:
         """Return the underlying cudaEvent_t pointer address as Python int."""
         return self._mnff.handle

cuda_core/cuda/core/experimental/_linker.py

@@ -2,13 +2,18 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+from __future__ import annotations
+
 import ctypes
 import weakref
 from contextlib import contextmanager
 from dataclasses import dataclass
-from typing import List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 from warnings import warn
 
+if TYPE_CHECKING:
+    import cuda.bindings
+
 from cuda.core.experimental._device import Device
 from cuda.core.experimental._module import ObjectCode
 from cuda.core.experimental._utils import check_or_create_options, driver, handle_return, is_sequence
@@ -324,7 +329,7 @@ def _exception_manager(self):
 
 
 nvJitLinkHandleT = int
-LinkerHandleT = Union[nvJitLinkHandleT, "CUlinkState"]
+LinkerHandleT = Union[nvJitLinkHandleT, "cuda.bindings.driver.CUlinkState"]
 
 
 class Linker:

cuda_core/cuda/core/experimental/_program.py

@@ -2,11 +2,16 @@
 #
 # SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE
 
+from __future__ import annotations
+
 import weakref
 from dataclasses import dataclass
-from typing import List, Optional, Tuple, Union
+from typing import TYPE_CHECKING, List, Optional, Tuple, Union
 from warnings import warn
 
+if TYPE_CHECKING:
+    import cuda.bindings
+
 from cuda.core.experimental._device import Device
 from cuda.core.experimental._linker import Linker, LinkerHandleT, LinkerOptions
 from cuda.core.experimental._module import ObjectCode
@@ -331,6 +336,9 @@ def __repr__(self):
         return self._formatted_options
 
 
+ProgramHandleT = Union["cuda.bindings.nvrtc.nvrtcProgram", LinkerHandleT]
+
+
 class Program:
     """Represent a compilation machinery to process programs into
     :obj:`~_module.ObjectCode`.
@@ -498,6 +506,6 @@ def backend(self) -> str:
         return self._backend
 
     @property
-    def handle(self) -> Union["nvrtcProgram", LinkerHandleT]:
+    def handle(self) -> ProgramHandleT:
         """Return the underlying handle object."""
         return self._mnff.handle

cuda_core/cuda/core/experimental/_stream.py

@@ -11,6 +11,7 @@
 from typing import TYPE_CHECKING, Optional, Tuple, Union
 
 if TYPE_CHECKING:
+    import cuda.bindings
     from cuda.core.experimental._device import Device
 from cuda.core.experimental._context import Context
 from cuda.core.experimental._event import Event, EventOptions
@@ -147,7 +148,7 @@ def __cuda_stream__(self) -> Tuple[int, int]:
         return (0, self.handle)
 
     @property
-    def handle(self) -> "CUstream":
+    def handle(self) -> cuda.bindings.driver.CUstream:
         """Return the underlying cudaStream_t pointer address as Python int."""
         return self._mnff.handle