Coverage for cuda / core / typing.py: 96.77%

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

2# 

3# SPDX-License-Identifier: Apache-2.0 

4 

5"""Public type aliases, protocols, and enumerations used in cuda.core API signatures.""" 

6 

7try: (empty)

8 from enum import StrEnum (empty)

9except ImportError: 

10 from backports.strenum import StrEnum 

11from typing import Literal as _Literal (empty)

12 

13from cuda.core._context import DeviceResourcesType (empty)

14from cuda.core._stream import IsStreamType (empty)

15from cuda.core._utils.cuda_utils import driver (empty)

16 

17__all__ = [ (empty)

18 "CompilerBackendType", 

19 "DevicePointerType", 

20 "DeviceResourcesType", 

21 "GraphConditionalType", 

22 "GraphMemoryType", 

23 "IsStreamType", 

24 "ManagedMemoryLocationType", 

25 "ObjectCodeFormatType", 

26 "PCHStatusType", 

27 "ProcessStateType", 

28 "SourceCodeType", 

29 "VirtualMemoryAccessType", 

30 "VirtualMemoryAllocationType", 

31 "VirtualMemoryGranularityType", 

32 "VirtualMemoryHandleType", 

33 "VirtualMemoryLocationType", 

34] 

35 

36 

37# A type union of :obj:`~driver.CUdeviceptr`, `int` and `None` for hinting 

38# :attr:`Buffer.handle`. 

39DevicePointerType = driver.CUdeviceptr | int | None (empty)

40 

41 

42ProcessStateType = _Literal["running", "locked", "checkpointed", "failed"] (empty)

43 

44 

45class SourceCodeType(StrEnum): (empty)

46 CXX = "c++" (empty)

47 PTX = "ptx" (empty)

48 NVVM = "nvvm" (empty)

49 

50 

51class ObjectCodeFormatType(StrEnum): (empty)

52 PTX = "ptx" (empty)

53 CUBIN = "cubin" (empty)

54 LTOIR = "ltoir" (empty)

55 FATBIN = "fatbin" (empty)

56 OBJECT = "object" (empty)

57 LIBRARY = "library" (empty)

58 

59 

60class CompilerBackendType(StrEnum): (empty)

61 NVRTC = "NVRTC" (empty)

62 NVVM = "NVVM" (empty)

63 NVJITLINK = "nvJitLink" (empty)

64 DRIVER = "driver" (empty)

65 

66 

67class PCHStatusType(StrEnum): (empty)

68 CREATED = "created" (empty)

69 NOT_ATTEMPTED = "not_attempted" (empty)

70 FAILED = "failed" (empty)

71 

72 

73class GraphConditionalType(StrEnum): (empty)

74 IF = "if" (empty)

75 WHILE = "while" (empty)

76 SWITCH = "switch" (empty)

77 

78 

79class GraphMemoryType(StrEnum): (empty)

80 DEVICE = "device" (empty)

81 HOST = "host" (empty)

82 MANAGED = "managed" (empty)

83 

84 

85class ManagedMemoryLocationType(StrEnum): (empty)

86 DEVICE = "device" (empty)

87 HOST = "host" (empty)

88 HOST_NUMA = "host_numa" (empty)

89 

90 

91class VirtualMemoryHandleType(StrEnum): (empty)

92 POSIX_FD = "posix_fd" (empty)

93 WIN32_KMT = "win32_kmt" (empty)

94 FABRIC = "fabric" (empty)

95 

96 

97class VirtualMemoryLocationType(StrEnum): (empty)

98 DEVICE = "device" (empty)

99 HOST = "host" (empty)

100 HOST_NUMA = "host_numa" (empty)

101 HOST_NUMA_CURRENT = "host_numa_current" (empty)

102 

103 

104class VirtualMemoryGranularityType(StrEnum): (empty)

105 MINIMUM = "minimum" (empty)

106 RECOMMENDED = "recommended" (empty)

107 

108 

109class VirtualMemoryAccessType(StrEnum): (empty)

110 READ_WRITE = "rw" (empty)

111 READ = "r" (empty)

112 

113 

114class VirtualMemoryAllocationType(StrEnum): (empty)

115 PINNED = "pinned" (empty)

116 MANAGED = "managed" (empty)

117 

118 

119del StrEnum (empty)