Coverage for cuda/bindings/utils/__init__.py: 100.00%

1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved. 

2# SPDX-License-Identifier: Apache-2.0 

3from collections.abc import Callable (empty)

4from typing import Any (empty)

5 

6from ._envvar import envvar_bool (empty)

7from ._nvvm_utils import check_nvvm_compiler_options (empty)

8from ._ptx_utils import get_minimal_required_cuda_ver_from_ptx_ver, get_ptx_ver (empty)

9from ._version_check import warn_if_cuda_major_version_mismatch (empty)

10 

11_handle_getters: dict[type, Callable[[Any], int]] = {} (empty)

12 

13 

14def _add_cuda_native_handle_getter(t: type, getter: Callable[[Any], int]) -> None: (empty)

15 _handle_getters[t] = getter (empty)

16 

17 

18def get_cuda_native_handle(obj: Any) -> int: (empty)

19 """Returns the address of the provided CUDA Python object as a Python int. 

20 

21 Parameters 

22 ---------- 

23 obj : Any 

24 CUDA Python object 

25 

26 Returns 

27 ------- 

28 int : The object address. 

29 """ 

30 obj_type = type(obj) 22 ctx1fghijklmnopqrstuvwbcde

31 try: 22 ctx1fghijklmnopqrstuvwbcde

32 getter = _handle_getters[obj_type] 22 ctx1fghijklmnopqrstuvwbcde

33 except KeyError: 4 ctx1bcde

34 raise TypeError("Unknown type: " + str(obj_type)) from None 4 ctx1bcde

35 # Deliberately outside the try: a KeyError raised by the getter itself is a 

36 # bug in that getter, not an unregistered type. 

37 return getter(obj) 18 ctx1fghijklmnopqrstuvw