Coverage for cuda / core / system / __init__.py: 100.00%

14 statements  

« prev     ^ index     » next       coverage.py v7.14.0, created at 2026-05-22 01:37 +0000

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

2# 

3# SPDX-License-Identifier: Apache-2.0 

4 

5 

6# NOTE: We must maintain that it is always possible to import this module 

7# without CUDA being installed, and without CUDA being initialized or any 

8# contexts created, so that a user can use NVML to explore things about their 

9# system without loading CUDA. 

10 

11 

12__all__ = [ 

13 "CUDA_BINDINGS_NVML_IS_COMPATIBLE", 

14 "get_kernel_mode_driver_version", 

15 "get_num_devices", 

16 "get_process_name", 

17 "get_user_mode_driver_version", 

18] 

19 

20 

21from cuda.core.system import typing 

22 

23from ._system import * 

24 

25if CUDA_BINDINGS_NVML_IS_COMPATIBLE: 

26 from ._device import * 

27 from ._device import __all__ as _device_all 

28 from ._system_events import * 

29 from ._system_events import __all__ as _system_events_all 

30 from .exceptions import * 

31 from .exceptions import __all__ as _exceptions_all 

32 

33 __all__.append("get_nvml_version") 

34 __all__.extend(_device_all) 

35 __all__.extend(_system_events_all) 

36 __all__.extend(_exceptions_all)