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: LicenseRef-NVIDIA-SOFTWARE-LICENSE 

3from typing import Any, Callable (empty)

4 

5from ._nvvm_utils import check_nvvm_compiler_options (empty)

6from ._ptx_utils import get_minimal_required_cuda_ver_from_ptx_ver, get_ptx_ver (empty)

7from ._version_check import warn_if_cuda_major_version_mismatch (empty)

8 

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

10 

11 

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

13 _handle_getters[t] = getter (empty)

14 

15 

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

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

18 

19 Parameters 

20 ---------- 

21 obj : Any 

22 CUDA Python object 

23 

24 Returns 

25 ------- 

26 int : The object address. 

27 """ 

28 obj_type = type(obj) 21 ctx1fghijklmnopqrstuvbcde

29 try: 21 ctx1fghijklmnopqrstuvbcde

30 return _handle_getters[obj_type](obj) 21 ctx1fghijklmnopqrstuvbcde

31 except KeyError: 4 ctx1bcde

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