Coverage for cuda / core / __init__.py: 93.10%

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

2# 

3# SPDX-License-Identifier: Apache-2.0 

4 

5from cuda.core._version import __version__ (empty)

6 

7 

8def _import_versioned_module(): (empty)

9 import importlib (empty)

10 

11 from cuda import bindings (empty)

12 

13 cuda_major = bindings.__version__.split(".")[0] (empty)

14 if cuda_major not in ("12", "13"): (empty)

15 raise ImportError("cuda.bindings 12.x or 13.x must be installed") 

16 

17 subdir = f"cu{cuda_major}" (empty)

18 try: (empty)

19 versioned_mod = importlib.import_module(f".{subdir}", __package__) (empty)

20 # Import all symbols from the module 

21 globals().update(versioned_mod.__dict__) 

22 except ImportError: (empty)

23 # This is not a wheel build, but a conda or local build, do nothing 

24 pass (empty)

25 

26 

27_import_versioned_module() (empty)

28del _import_versioned_module (empty)

29 

30 

31from cuda.core import system, utils (empty)

32from cuda.core._device import Device (empty)

33from cuda.core._event import Event, EventOptions (empty)

34from cuda.core._graphics import GraphicsResource (empty)

35from cuda.core._launch_config import LaunchConfig (empty)

36from cuda.core._launcher import launch (empty)

37from cuda.core._linker import Linker, LinkerOptions (empty)

38from cuda.core._memory import ( (empty)

39 Buffer, 

40 DeviceMemoryResource, 

41 DeviceMemoryResourceOptions, 

42 GraphMemoryResource, 

43 LegacyPinnedMemoryResource, 

44 ManagedMemoryResource, 

45 ManagedMemoryResourceOptions, 

46 MemoryResource, 

47 PinnedMemoryResource, 

48 PinnedMemoryResourceOptions, 

49 VirtualMemoryResource, 

50 VirtualMemoryResourceOptions, 

51) 

52from cuda.core._memoryview import ( (empty)

53 StridedMemoryView, 

54 args_viewable_as_strided_memory, 

55) 

56from cuda.core._module import Kernel, ObjectCode (empty)

57from cuda.core._program import Program, ProgramOptions (empty)

58from cuda.core._stream import ( (empty)

59 LEGACY_DEFAULT_STREAM, 

60 PER_THREAD_DEFAULT_STREAM, 

61 Stream, 

62 StreamOptions, 

63) 

64from cuda.core._tensor_map import TensorMapDescriptor, TensorMapDescriptorOptions (empty)

65from cuda.core.graph import ( (empty)

66 Graph, 

67 GraphAllocOptions, 

68 GraphBuilder, 

69 GraphCompleteOptions, 

70 GraphCondition, 

71 GraphDebugPrintOptions, 

72 GraphDefinition, 

73)