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 typing import Any, Callable (empty)

4 

5from ._envvar import envvar_bool (empty)

6from ._nvvm_utils import check_nvvm_compiler_options (empty)

7from ._ptx_utils import get_minimal_required_cuda_ver_from_ptx_ver, get_ptx_ver (empty)

8from ._version_check import warn_if_cuda_major_version_mismatch (empty)

9 

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

11 

12 

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

14 _handle_getters[t] = getter (empty)

15 

16 

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

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

19 

20 Parameters 

21 ---------- 

22 obj : Any 

23 CUDA Python object 

24 

25 Returns 

26 ------- 

27 int : The object address. 

28 """ 

29 obj_type = type(obj) 22 ctx1fghijklmnopqrstuvwbcde

30 try: 22 ctx1fghijklmnopqrstuvwbcde

31 getter = _handle_getters[obj_type] 22 ctx1fghijklmnopqrstuvwbcde

32 except KeyError: 4 ctx1bcde

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

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

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

36 return getter(obj) 18 ctx1fghijklmnopqrstuvw