nvmath-python 0.5.0__cp313-cp313-win_amd64.whl

This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Files changed (174) hide show
  1. nvmath/__init__.pxd +0 -0
  2. nvmath/__init__.py +37 -0
  3. nvmath/_internal/__init__.py +0 -0
  4. nvmath/_internal/layout.py +58 -0
  5. nvmath/_utils.py +363 -0
  6. nvmath/bindings/__init__.pxd +0 -0
  7. nvmath/bindings/__init__.py +44 -0
  8. nvmath/bindings/_internal/__init__.pxd +0 -0
  9. nvmath/bindings/_internal/__init__.py +0 -0
  10. nvmath/bindings/_internal/cublas.cp313-win_amd64.pyd +0 -0
  11. nvmath/bindings/_internal/cublas.pxd +518 -0
  12. nvmath/bindings/_internal/cublasLt.cp313-win_amd64.pyd +0 -0
  13. nvmath/bindings/_internal/cublasLt.pxd +54 -0
  14. nvmath/bindings/_internal/cudss.cp313-win_amd64.pyd +0 -0
  15. nvmath/bindings/_internal/cudss.pxd +44 -0
  16. nvmath/bindings/_internal/cufft.cp313-win_amd64.pyd +0 -0
  17. nvmath/bindings/_internal/cufft.pxd +69 -0
  18. nvmath/bindings/_internal/cufftMp.pxd +77 -0
  19. nvmath/bindings/_internal/curand.cp313-win_amd64.pyd +0 -0
  20. nvmath/bindings/_internal/curand.pxd +42 -0
  21. nvmath/bindings/_internal/cusolver.cp313-win_amd64.pyd +0 -0
  22. nvmath/bindings/_internal/cusolver.pxd +15 -0
  23. nvmath/bindings/_internal/cusolverDn.cp313-win_amd64.pyd +0 -0
  24. nvmath/bindings/_internal/cusolverDn.pxd +386 -0
  25. nvmath/bindings/_internal/cusparse.cp313-win_amd64.pyd +0 -0
  26. nvmath/bindings/_internal/cusparse.pxd +270 -0
  27. nvmath/bindings/_internal/mathdx.cp313-win_amd64.pyd +0 -0
  28. nvmath/bindings/_internal/mathdx.pxd +79 -0
  29. nvmath/bindings/_internal/nvshmem.pxd +29 -0
  30. nvmath/bindings/_internal/utils.cp313-win_amd64.pyd +0 -0
  31. nvmath/bindings/_internal/utils.pxd +186 -0
  32. nvmath/bindings/cublas.cp313-win_amd64.pyd +0 -0
  33. nvmath/bindings/cublas.pxd +545 -0
  34. nvmath/bindings/cublas.pyi +18 -0
  35. nvmath/bindings/cublasLt.cp313-win_amd64.pyd +0 -0
  36. nvmath/bindings/cublasLt.pxd +99 -0
  37. nvmath/bindings/cudss.cp313-win_amd64.pyd +0 -0
  38. nvmath/bindings/cudss.pxd +83 -0
  39. nvmath/bindings/cudss.pyi +249 -0
  40. nvmath/bindings/cufft.cp313-win_amd64.pyd +0 -0
  41. nvmath/bindings/cufft.pxd +114 -0
  42. nvmath/bindings/cufftMp.pxd +124 -0
  43. nvmath/bindings/curand.cp313-win_amd64.pyd +0 -0
  44. nvmath/bindings/curand.pxd +71 -0
  45. nvmath/bindings/cusolver.cp313-win_amd64.pyd +0 -0
  46. nvmath/bindings/cusolver.pxd +60 -0
  47. nvmath/bindings/cusolverDn.cp313-win_amd64.pyd +0 -0
  48. nvmath/bindings/cusolverDn.pxd +410 -0
  49. nvmath/bindings/cusparse.cp313-win_amd64.pyd +0 -0
  50. nvmath/bindings/cusparse.pxd +329 -0
  51. nvmath/bindings/cycublas.cp313-win_amd64.pyd +0 -0
  52. nvmath/bindings/cycublas.pxd +657 -0
  53. nvmath/bindings/cycublasLt.cp313-win_amd64.pyd +0 -0
  54. nvmath/bindings/cycublasLt.pxd +1010 -0
  55. nvmath/bindings/cycudss.cp313-win_amd64.pyd +0 -0
  56. nvmath/bindings/cycudss.pxd +191 -0
  57. nvmath/bindings/cycufft.cp313-win_amd64.pyd +0 -0
  58. nvmath/bindings/cycufft.pxd +320 -0
  59. nvmath/bindings/cycufftMp.pxd +333 -0
  60. nvmath/bindings/cycurand.cp313-win_amd64.pyd +0 -0
  61. nvmath/bindings/cycurand.pxd +146 -0
  62. nvmath/bindings/cycusolver.cp313-win_amd64.pyd +0 -0
  63. nvmath/bindings/cycusolver.pxd +143 -0
  64. nvmath/bindings/cycusolverDn.cp313-win_amd64.pyd +0 -0
  65. nvmath/bindings/cycusolverDn.pxd +414 -0
  66. nvmath/bindings/cycusparse.cp313-win_amd64.pyd +0 -0
  67. nvmath/bindings/cycusparse.pxd +476 -0
  68. nvmath/bindings/cymathdx.cp313-win_amd64.pyd +0 -0
  69. nvmath/bindings/cymathdx.pxd +363 -0
  70. nvmath/bindings/cynvshmem.pxd +126 -0
  71. nvmath/bindings/mathdx.cp313-win_amd64.pyd +0 -0
  72. nvmath/bindings/mathdx.pxd +131 -0
  73. nvmath/bindings/mathdx.pyi +686 -0
  74. nvmath/bindings/nvpl/__init__.pxd +0 -0
  75. nvmath/bindings/nvpl/__init__.py +11 -0
  76. nvmath/bindings/nvpl/_internal/__init__.pxd +0 -0
  77. nvmath/bindings/nvpl/_internal/__init__.py +0 -0
  78. nvmath/bindings/nvpl/_internal/fft.pxd +36 -0
  79. nvmath/bindings/nvpl/cyfft.pxd +87 -0
  80. nvmath/bindings/nvpl/fft.pxd +100 -0
  81. nvmath/bindings/nvshmem.pxd +54 -0
  82. nvmath/device/__init__.py +23 -0
  83. nvmath/device/_deprecated.py +33 -0
  84. nvmath/device/caching.py +51 -0
  85. nvmath/device/common.py +131 -0
  86. nvmath/device/common_backend.py +125 -0
  87. nvmath/device/common_cuda.py +156 -0
  88. nvmath/device/common_mathdx.py +97 -0
  89. nvmath/device/common_numba.py +159 -0
  90. nvmath/device/common_opaque_tensor.py +128 -0
  91. nvmath/device/cublasdx.py +1085 -0
  92. nvmath/device/cublasdx_backend.py +552 -0
  93. nvmath/device/cublasdx_numba.py +528 -0
  94. nvmath/device/cufftdx.py +551 -0
  95. nvmath/device/cufftdx_backend.py +194 -0
  96. nvmath/device/cufftdx_numba.py +123 -0
  97. nvmath/device/curand_kernel.py +9147 -0
  98. nvmath/device/nvrtc.py +95 -0
  99. nvmath/device/patch.py +35 -0
  100. nvmath/device/random.py +417 -0
  101. nvmath/device/random_helpers.py +23 -0
  102. nvmath/device/random_states.py +189 -0
  103. nvmath/device/types.py +11 -0
  104. nvmath/device/vector_types_numba.py +203 -0
  105. nvmath/distributed/__init__.py +117 -0
  106. nvmath/distributed/_internal/__init__.py +0 -0
  107. nvmath/distributed/_internal/nvshmem.py +243 -0
  108. nvmath/distributed/_internal/tensor_ifc_cupy.py +111 -0
  109. nvmath/distributed/_internal/tensor_ifc_numpy.py +53 -0
  110. nvmath/distributed/_internal/tensor_ifc_torch.py +108 -0
  111. nvmath/distributed/_internal/tensor_wrapper.py +59 -0
  112. nvmath/distributed/_utils.py +152 -0
  113. nvmath/distributed/fft/__init__.py +6 -0
  114. nvmath/distributed/fft/_configuration.py +86 -0
  115. nvmath/distributed/fft/fft.py +1737 -0
  116. nvmath/distributed/reshape/__init__.py +6 -0
  117. nvmath/distributed/reshape/_configuration.py +39 -0
  118. nvmath/distributed/reshape/reshape.py +1249 -0
  119. nvmath/fft/__init__.py +7 -0
  120. nvmath/fft/_configuration.py +208 -0
  121. nvmath/fft/_exec_utils.py +116 -0
  122. nvmath/fft/_helpers.py +260 -0
  123. nvmath/fft/fft.py +2346 -0
  124. nvmath/internal/__init__.py +5 -0
  125. nvmath/internal/enum_utils.py +142 -0
  126. nvmath/internal/formatters.py +87 -0
  127. nvmath/internal/mem_limit.py +51 -0
  128. nvmath/internal/package_ifc.py +119 -0
  129. nvmath/internal/package_ifc_cuda.py +43 -0
  130. nvmath/internal/package_ifc_cupy.py +49 -0
  131. nvmath/internal/package_ifc_torch.py +31 -0
  132. nvmath/internal/package_wrapper.py +14 -0
  133. nvmath/internal/tensor_ifc.py +143 -0
  134. nvmath/internal/tensor_ifc_cupy.py +185 -0
  135. nvmath/internal/tensor_ifc_numpy.py +157 -0
  136. nvmath/internal/tensor_ifc_torch.py +115 -0
  137. nvmath/internal/tensor_wrapper.py +160 -0
  138. nvmath/internal/typemaps.py +154 -0
  139. nvmath/internal/utils.py +634 -0
  140. nvmath/linalg/__init__.py +11 -0
  141. nvmath/linalg/_internal/__init__.py +3 -0
  142. nvmath/linalg/_internal/algo_cap_ifc.py +86 -0
  143. nvmath/linalg/_internal/algo_config_ifc.py +87 -0
  144. nvmath/linalg/_internal/enum_to_tuples.py +64 -0
  145. nvmath/linalg/_internal/epilog_protocol.py +738 -0
  146. nvmath/linalg/_internal/matmul_desc_ifc.py +72 -0
  147. nvmath/linalg/_internal/matmul_pref_ifc.py +65 -0
  148. nvmath/linalg/_internal/matrix_layout_ifc.py +59 -0
  149. nvmath/linalg/_internal/typemaps.py +135 -0
  150. nvmath/linalg/_internal/utils.py +96 -0
  151. nvmath/linalg/advanced/__init__.py +8 -0
  152. nvmath/linalg/advanced/_algorithmmod.py +142 -0
  153. nvmath/linalg/advanced/_configuration.py +320 -0
  154. nvmath/linalg/advanced/helpers/__init__.py +5 -0
  155. nvmath/linalg/advanced/helpers/matmul.py +173 -0
  156. nvmath/linalg/advanced/matmulmod.py +2855 -0
  157. nvmath/memory.py +266 -0
  158. nvmath/sparse/__init__.py +18 -0
  159. nvmath/sparse/_internal/__init__.py +21 -0
  160. nvmath/sparse/_internal/common_utils.py +81 -0
  161. nvmath/sparse/_internal/cudss_config_ifc.py +584 -0
  162. nvmath/sparse/_internal/cudss_data_ifc.py +291 -0
  163. nvmath/sparse/_internal/cudss_utils.py +398 -0
  164. nvmath/sparse/_internal/sparse_csr_ifc.py +154 -0
  165. nvmath/sparse/_internal/sparse_format_helpers.py +88 -0
  166. nvmath/sparse/_internal/sparse_tensor_ifc.py +80 -0
  167. nvmath/sparse/advanced/__init__.py +7 -0
  168. nvmath/sparse/advanced/_configuration.py +150 -0
  169. nvmath/sparse/advanced/direct_solver.py +1705 -0
  170. nvmath_python-0.5.0.dist-info/METADATA +122 -0
  171. nvmath_python-0.5.0.dist-info/RECORD +174 -0
  172. nvmath_python-0.5.0.dist-info/WHEEL +5 -0
  173. nvmath_python-0.5.0.dist-info/licenses/LICENSE +177 -0
  174. nvmath_python-0.5.0.dist-info/top_level.txt +1 -0
nvmath/bindings/mathdx.pyi ADDED
@@ -0,0 +1,686 @@
1
+ # Copyright (c) 2024-2025, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
2
+ #
3
+ # SPDX-License-Identifier: Apache-2.0
4
+
5
+ import _cython_3_0_12
6
+ import enum
7
+ from typing import Any, Callable, ClassVar
8
+
9
+ __pyx_capi__: dict
10
+ __test__: dict
11
+ check_status: _cython_3_0_12.cython_function_or_method
12
+ commondx_create_code: _cython_3_0_12.cython_function_or_method
13
+ commondx_destroy_code: _cython_3_0_12.cython_function_or_method
14
+ commondx_get_code_ltoir: _cython_3_0_12.cython_function_or_method
15
+ commondx_get_code_ltoir_size: _cython_3_0_12.cython_function_or_method
16
+ commondx_get_code_ltoir_sizes: _cython_3_0_12.cython_function_or_method
17
+ commondx_get_code_ltoirs: _cython_3_0_12.cython_function_or_method
18
+ commondx_get_code_num_ltoirs: _cython_3_0_12.cython_function_or_method
19
+ commondx_get_code_option_int64: _cython_3_0_12.cython_function_or_method
20
+ commondx_get_code_options_int64s: _cython_3_0_12.cython_function_or_method
21
+ commondx_set_code_option_int64: _cython_3_0_12.cython_function_or_method
22
+ commondx_set_code_option_str: _cython_3_0_12.cython_function_or_method
23
+ commondx_status_to_str: _cython_3_0_12.cython_function_or_method
24
+ cublasdx_bind_device_function: _cython_3_0_12.cython_function_or_method
25
+ cublasdx_bind_tensor: _cython_3_0_12.cython_function_or_method
26
+ cublasdx_create_descriptor: _cython_3_0_12.cython_function_or_method
27
+ cublasdx_destroy_descriptor: _cython_3_0_12.cython_function_or_method
28
+ cublasdx_finalize_code: _cython_3_0_12.cython_function_or_method
29
+ cublasdx_finalize_device_functions: _cython_3_0_12.cython_function_or_method
30
+ cublasdx_finalize_tensors: _cython_3_0_12.cython_function_or_method
31
+ cublasdx_get_device_function_trait_str: _cython_3_0_12.cython_function_or_method
32
+ cublasdx_get_device_function_trait_str_size: _cython_3_0_12.cython_function_or_method
33
+ cublasdx_get_ltoir: _cython_3_0_12.cython_function_or_method
34
+ cublasdx_get_ltoir_size: _cython_3_0_12.cython_function_or_method
35
+ cublasdx_get_tensor_trait_int64: _cython_3_0_12.cython_function_or_method
36
+ cublasdx_get_tensor_trait_str: _cython_3_0_12.cython_function_or_method
37
+ cublasdx_get_tensor_trait_str_size: _cython_3_0_12.cython_function_or_method
38
+ cublasdx_get_trait_int64: _cython_3_0_12.cython_function_or_method
39
+ cublasdx_get_trait_int64s: _cython_3_0_12.cython_function_or_method
40
+ cublasdx_get_trait_str: _cython_3_0_12.cython_function_or_method
41
+ cublasdx_get_trait_str_size: _cython_3_0_12.cython_function_or_method
42
+ cublasdx_operator_type_to_str: _cython_3_0_12.cython_function_or_method
43
+ cublasdx_set_operator_int64: _cython_3_0_12.cython_function_or_method
44
+ cublasdx_set_operator_int64s: _cython_3_0_12.cython_function_or_method
45
+ cublasdx_set_option_str: _cython_3_0_12.cython_function_or_method
46
+ cublasdx_set_tensor_option_int64: _cython_3_0_12.cython_function_or_method
47
+ cublasdx_trait_type_to_str: _cython_3_0_12.cython_function_or_method
48
+ cufftdx_create_descriptor: _cython_3_0_12.cython_function_or_method
49
+ cufftdx_destroy_descriptor: _cython_3_0_12.cython_function_or_method
50
+ cufftdx_finalize_code: _cython_3_0_12.cython_function_or_method
51
+ cufftdx_get_knob_int64s: _cython_3_0_12.cython_function_or_method
52
+ cufftdx_get_knob_int64size: _cython_3_0_12.cython_function_or_method
53
+ cufftdx_get_ltoir: _cython_3_0_12.cython_function_or_method
54
+ cufftdx_get_ltoir_size: _cython_3_0_12.cython_function_or_method
55
+ cufftdx_get_trait_commondx_data_type: _cython_3_0_12.cython_function_or_method
56
+ cufftdx_get_trait_int64: _cython_3_0_12.cython_function_or_method
57
+ cufftdx_get_trait_int64s: _cython_3_0_12.cython_function_or_method
58
+ cufftdx_get_trait_str: _cython_3_0_12.cython_function_or_method
59
+ cufftdx_get_trait_str_size: _cython_3_0_12.cython_function_or_method
60
+ cufftdx_operator_type_to_str: _cython_3_0_12.cython_function_or_method
61
+ cufftdx_set_operator_int64: _cython_3_0_12.cython_function_or_method
62
+ cufftdx_set_operator_int64s: _cython_3_0_12.cython_function_or_method
63
+ cufftdx_set_option_str: _cython_3_0_12.cython_function_or_method
64
+ cufftdx_trait_type_to_str: _cython_3_0_12.cython_function_or_method
65
+ cusolverdx_create_descriptor: _cython_3_0_12.cython_function_or_method
66
+ cusolverdx_destroy_descriptor: _cython_3_0_12.cython_function_or_method
67
+ cusolverdx_finalize_code: _cython_3_0_12.cython_function_or_method
68
+ cusolverdx_get_ltoir: _cython_3_0_12.cython_function_or_method
69
+ cusolverdx_get_ltoir_size: _cython_3_0_12.cython_function_or_method
70
+ cusolverdx_get_trait_int64: _cython_3_0_12.cython_function_or_method
71
+ cusolverdx_get_trait_str: _cython_3_0_12.cython_function_or_method
72
+ cusolverdx_get_trait_str_size: _cython_3_0_12.cython_function_or_method
73
+ cusolverdx_get_universal_fatbin: _cython_3_0_12.cython_function_or_method
74
+ cusolverdx_get_universal_fatbin_size: _cython_3_0_12.cython_function_or_method
75
+ cusolverdx_operator_type_to_str: _cython_3_0_12.cython_function_or_method
76
+ cusolverdx_set_operator_int64: _cython_3_0_12.cython_function_or_method
77
+ cusolverdx_set_operator_int64s: _cython_3_0_12.cython_function_or_method
78
+ cusolverdx_set_option_str: _cython_3_0_12.cython_function_or_method
79
+ cusolverdx_trait_type_to_str: _cython_3_0_12.cython_function_or_method
80
+ get_version: _cython_3_0_12.cython_function_or_method
81
+ get_version_ex: _cython_3_0_12.cython_function_or_method
82
+
83
+ class CommondxCodeContainer(enum.IntEnum):
84
+ __new__: ClassVar[Callable] = ...
85
+ FATBIN: ClassVar[CommondxCodeContainer] = ...
86
+ LTOIR: ClassVar[CommondxCodeContainer] = ...
87
+ _generate_next_value_: ClassVar[Callable] = ...
88
+ _member_map_: ClassVar[dict] = ...
89
+ _member_names_: ClassVar[list] = ...
90
+ _member_type_: ClassVar[type[int]] = ...
91
+ _unhashable_values_: ClassVar[list] = ...
92
+ _use_args_: ClassVar[bool] = ...
93
+ _value2member_map_: ClassVar[dict] = ...
94
+ def __format__(self, *args, **kwargs) -> str: ...
95
+
96
+ class CommondxExecution(enum.IntEnum):
97
+ __new__: ClassVar[Callable] = ...
98
+ BLOCK: ClassVar[CommondxExecution] = ...
99
+ THREAD: ClassVar[CommondxExecution] = ...
100
+ _generate_next_value_: ClassVar[Callable] = ...
101
+ _member_map_: ClassVar[dict] = ...
102
+ _member_names_: ClassVar[list] = ...
103
+ _member_type_: ClassVar[type[int]] = ...
104
+ _unhashable_values_: ClassVar[list] = ...
105
+ _use_args_: ClassVar[bool] = ...
106
+ _value2member_map_: ClassVar[dict] = ...
107
+ def __format__(self, *args, **kwargs) -> str: ...
108
+
109
+ class CommondxOption(enum.IntEnum):
110
+ __new__: ClassVar[Callable] = ...
111
+ CODE_CONTAINER: ClassVar[CommondxOption] = ...
112
+ CODE_ISA: ClassVar[CommondxOption] = ...
113
+ EXTRA_NVTRC_ARGS: ClassVar[CommondxOption] = ...
114
+ SYMBOL_NAME: ClassVar[CommondxOption] = ...
115
+ TARGET_SM: ClassVar[CommondxOption] = ...
116
+ _generate_next_value_: ClassVar[Callable] = ...
117
+ _member_map_: ClassVar[dict] = ...
118
+ _member_names_: ClassVar[list] = ...
119
+ _member_type_: ClassVar[type[int]] = ...
120
+ _unhashable_values_: ClassVar[list] = ...
121
+ _use_args_: ClassVar[bool] = ...
122
+ _value2member_map_: ClassVar[dict] = ...
123
+ def __format__(self, *args, **kwargs) -> str: ...
124
+
125
+ class CommondxPrecision(enum.IntEnum):
126
+ __new__: ClassVar[Callable] = ...
127
+ BF16: ClassVar[CommondxPrecision] = ...
128
+ F16: ClassVar[CommondxPrecision] = ...
129
+ F32: ClassVar[CommondxPrecision] = ...
130
+ F64: ClassVar[CommondxPrecision] = ...
131
+ F8_E4M3: ClassVar[CommondxPrecision] = ...
132
+ F8_E5M2: ClassVar[CommondxPrecision] = ...
133
+ I16: ClassVar[CommondxPrecision] = ...
134
+ I32: ClassVar[CommondxPrecision] = ...
135
+ I64: ClassVar[CommondxPrecision] = ...
136
+ I8: ClassVar[CommondxPrecision] = ...
137
+ TF32: ClassVar[CommondxPrecision] = ...
138
+ UI16: ClassVar[CommondxPrecision] = ...
139
+ UI32: ClassVar[CommondxPrecision] = ...
140
+ UI64: ClassVar[CommondxPrecision] = ...
141
+ UI8: ClassVar[CommondxPrecision] = ...
142
+ _generate_next_value_: ClassVar[Callable] = ...
143
+ _member_map_: ClassVar[dict] = ...
144
+ _member_names_: ClassVar[list] = ...
145
+ _member_type_: ClassVar[type[int]] = ...
146
+ _unhashable_values_: ClassVar[list] = ...
147
+ _use_args_: ClassVar[bool] = ...
148
+ _value2member_map_: ClassVar[dict] = ...
149
+ def __format__(self, *args, **kwargs) -> str: ...
150
+
151
+ class CommondxStatusType(enum.IntEnum):
152
+ __new__: ClassVar[Callable] = ...
153
+ COMPILATION_ERROR: ClassVar[CommondxStatusType] = ...
154
+ CUFFT_ERROR: ClassVar[CommondxStatusType] = ...
155
+ INTERNAL_ERROR: ClassVar[CommondxStatusType] = ...
156
+ INVALID_VALUE: ClassVar[CommondxStatusType] = ...
157
+ SUCCESS: ClassVar[CommondxStatusType] = ...
158
+ _generate_next_value_: ClassVar[Callable] = ...
159
+ _member_map_: ClassVar[dict] = ...
160
+ _member_names_: ClassVar[list] = ...
161
+ _member_type_: ClassVar[type[int]] = ...
162
+ _unhashable_values_: ClassVar[list] = ...
163
+ _use_args_: ClassVar[bool] = ...
164
+ _value2member_map_: ClassVar[dict] = ...
165
+ def __format__(self, *args, **kwargs) -> str: ...
166
+
167
+ class CommondxValueType(enum.IntEnum):
168
+ __new__: ClassVar[Callable] = ...
169
+ C_16BF: ClassVar[CommondxValueType] = ...
170
+ C_16F: ClassVar[CommondxValueType] = ...
171
+ C_16F2: ClassVar[CommondxValueType] = ...
172
+ C_16I: ClassVar[CommondxValueType] = ...
173
+ C_16UI: ClassVar[CommondxValueType] = ...
174
+ C_32F: ClassVar[CommondxValueType] = ...
175
+ C_32I: ClassVar[CommondxValueType] = ...
176
+ C_32TF: ClassVar[CommondxValueType] = ...
177
+ C_32UI: ClassVar[CommondxValueType] = ...
178
+ C_64F: ClassVar[CommondxValueType] = ...
179
+ C_64I: ClassVar[CommondxValueType] = ...
180
+ C_64UI: ClassVar[CommondxValueType] = ...
181
+ C_8F_E4M3: ClassVar[CommondxValueType] = ...
182
+ C_8F_E5M2: ClassVar[CommondxValueType] = ...
183
+ C_8I: ClassVar[CommondxValueType] = ...
184
+ C_8UI: ClassVar[CommondxValueType] = ...
185
+ R_16BF: ClassVar[CommondxValueType] = ...
186
+ R_16F: ClassVar[CommondxValueType] = ...
187
+ R_16F2: ClassVar[CommondxValueType] = ...
188
+ R_16I: ClassVar[CommondxValueType] = ...
189
+ R_16UI: ClassVar[CommondxValueType] = ...
190
+ R_32F: ClassVar[CommondxValueType] = ...
191
+ R_32I: ClassVar[CommondxValueType] = ...
192
+ R_32TF: ClassVar[CommondxValueType] = ...
193
+ R_32UI: ClassVar[CommondxValueType] = ...
194
+ R_64F: ClassVar[CommondxValueType] = ...
195
+ R_64I: ClassVar[CommondxValueType] = ...
196
+ R_64UI: ClassVar[CommondxValueType] = ...
197
+ R_8F_E4M3: ClassVar[CommondxValueType] = ...
198
+ R_8F_E5M2: ClassVar[CommondxValueType] = ...
199
+ R_8I: ClassVar[CommondxValueType] = ...
200
+ R_8UI: ClassVar[CommondxValueType] = ...
201
+ _generate_next_value_: ClassVar[Callable] = ...
202
+ _member_map_: ClassVar[dict] = ...
203
+ _member_names_: ClassVar[list] = ...
204
+ _member_type_: ClassVar[type[int]] = ...
205
+ _unhashable_values_: ClassVar[list] = ...
206
+ _use_args_: ClassVar[bool] = ...
207
+ _value2member_map_: ClassVar[dict] = ...
208
+ def __format__(self, *args, **kwargs) -> str: ...
209
+
210
+ class CublasdxApi(enum.IntEnum):
211
+ __new__: ClassVar[Callable] = ...
212
+ SMEM: ClassVar[CublasdxApi] = ...
213
+ SMEM_DYNAMIC_LD: ClassVar[CublasdxApi] = ...
214
+ TENSORS: ClassVar[CublasdxApi] = ...
215
+ _generate_next_value_: ClassVar[Callable] = ...
216
+ _member_map_: ClassVar[dict] = ...
217
+ _member_names_: ClassVar[list] = ...
218
+ _member_type_: ClassVar[type[int]] = ...
219
+ _unhashable_values_: ClassVar[list] = ...
220
+ _use_args_: ClassVar[bool] = ...
221
+ _value2member_map_: ClassVar[dict] = ...
222
+ def __format__(self, *args, **kwargs) -> str: ...
223
+
224
+ class CublasdxArrangement(enum.IntEnum):
225
+ __new__: ClassVar[Callable] = ...
226
+ COL_MAJOR: ClassVar[CublasdxArrangement] = ...
227
+ ROW_MAJOR: ClassVar[CublasdxArrangement] = ...
228
+ _generate_next_value_: ClassVar[Callable] = ...
229
+ _member_map_: ClassVar[dict] = ...
230
+ _member_names_: ClassVar[list] = ...
231
+ _member_type_: ClassVar[type[int]] = ...
232
+ _unhashable_values_: ClassVar[list] = ...
233
+ _use_args_: ClassVar[bool] = ...
234
+ _value2member_map_: ClassVar[dict] = ...
235
+ def __format__(self, *args, **kwargs) -> str: ...
236
+
237
+ class CublasdxDeviceFunctionOption(enum.IntEnum):
238
+ __new__: ClassVar[Callable] = ...
239
+ COPY_ALIGNMENT: ClassVar[CublasdxDeviceFunctionOption] = ...
240
+ _generate_next_value_: ClassVar[Callable] = ...
241
+ _member_map_: ClassVar[dict] = ...
242
+ _member_names_: ClassVar[list] = ...
243
+ _member_type_: ClassVar[type[int]] = ...
244
+ _unhashable_values_: ClassVar[list] = ...
245
+ _use_args_: ClassVar[bool] = ...
246
+ _value2member_map_: ClassVar[dict] = ...
247
+ def __format__(self, *args, **kwargs) -> str: ...
248
+
249
+ class CublasdxDeviceFunctionTrait(enum.IntEnum):
250
+ __new__: ClassVar[Callable] = ...
251
+ NAME: ClassVar[CublasdxDeviceFunctionTrait] = ...
252
+ SYMBOL: ClassVar[CublasdxDeviceFunctionTrait] = ...
253
+ _generate_next_value_: ClassVar[Callable] = ...
254
+ _member_map_: ClassVar[dict] = ...
255
+ _member_names_: ClassVar[list] = ...
256
+ _member_type_: ClassVar[type[int]] = ...
257
+ _unhashable_values_: ClassVar[list] = ...
258
+ _use_args_: ClassVar[bool] = ...
259
+ _value2member_map_: ClassVar[dict] = ...
260
+ def __format__(self, *args, **kwargs) -> str: ...
261
+
262
+ class CublasdxDeviceFunctionType(enum.IntEnum):
263
+ __new__: ClassVar[Callable] = ...
264
+ AXPBY: ClassVar[CublasdxDeviceFunctionType] = ...
265
+ CLEAR: ClassVar[CublasdxDeviceFunctionType] = ...
266
+ COPY: ClassVar[CublasdxDeviceFunctionType] = ...
267
+ COPY_WAIT: ClassVar[CublasdxDeviceFunctionType] = ...
268
+ EXECUTE: ClassVar[CublasdxDeviceFunctionType] = ...
269
+ _generate_next_value_: ClassVar[Callable] = ...
270
+ _member_map_: ClassVar[dict] = ...
271
+ _member_names_: ClassVar[list] = ...
272
+ _member_type_: ClassVar[type[int]] = ...
273
+ _unhashable_values_: ClassVar[list] = ...
274
+ _use_args_: ClassVar[bool] = ...
275
+ _value2member_map_: ClassVar[dict] = ...
276
+ def __format__(self, *args, **kwargs) -> str: ...
277
+
278
+ class CublasdxFunction(enum.IntEnum):
279
+ __new__: ClassVar[Callable] = ...
280
+ MM: ClassVar[CublasdxFunction] = ...
281
+ _generate_next_value_: ClassVar[Callable] = ...
282
+ _member_map_: ClassVar[dict] = ...
283
+ _member_names_: ClassVar[list] = ...
284
+ _member_type_: ClassVar[type[int]] = ...
285
+ _unhashable_values_: ClassVar[list] = ...
286
+ _use_args_: ClassVar[bool] = ...
287
+ _value2member_map_: ClassVar[dict] = ...
288
+ def __format__(self, *args, **kwargs) -> str: ...
289
+
290
+ class CublasdxOperatorType(enum.IntEnum):
291
+ __new__: ClassVar[Callable] = ...
292
+ ALIGNMENT: ClassVar[CublasdxOperatorType] = ...
293
+ API: ClassVar[CublasdxOperatorType] = ...
294
+ ARRANGEMENT: ClassVar[CublasdxOperatorType] = ...
295
+ BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
296
+ EXECUTION: ClassVar[CublasdxOperatorType] = ...
297
+ FUNCTION: ClassVar[CublasdxOperatorType] = ...
298
+ LEADING_DIMENSION: ClassVar[CublasdxOperatorType] = ...
299
+ PRECISION: ClassVar[CublasdxOperatorType] = ...
300
+ SIZE: ClassVar[CublasdxOperatorType] = ...
301
+ SM: ClassVar[CublasdxOperatorType] = ...
302
+ STATIC_BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
303
+ TRANSPOSE_MODE: ClassVar[CublasdxOperatorType] = ...
304
+ TYPE: ClassVar[CublasdxOperatorType] = ...
305
+ _generate_next_value_: ClassVar[Callable] = ...
306
+ _member_map_: ClassVar[dict] = ...
307
+ _member_names_: ClassVar[list] = ...
308
+ _member_type_: ClassVar[type[int]] = ...
309
+ _unhashable_values_: ClassVar[list] = ...
310
+ _use_args_: ClassVar[bool] = ...
311
+ _value2member_map_: ClassVar[dict] = ...
312
+ def __format__(self, *args, **kwargs) -> str: ...
313
+
314
+ class CublasdxTensorOption(enum.IntEnum):
315
+ __new__: ClassVar[Callable] = ...
316
+ ALIGNMENT_BYTES: ClassVar[CublasdxTensorOption] = ...
317
+ _generate_next_value_: ClassVar[Callable] = ...
318
+ _member_map_: ClassVar[dict] = ...
319
+ _member_names_: ClassVar[list] = ...
320
+ _member_type_: ClassVar[type[int]] = ...
321
+ _unhashable_values_: ClassVar[list] = ...
322
+ _use_args_: ClassVar[bool] = ...
323
+ _value2member_map_: ClassVar[dict] = ...
324
+ def __format__(self, *args, **kwargs) -> str: ...
325
+
326
+ class CublasdxTensorTrait(enum.IntEnum):
327
+ __new__: ClassVar[Callable] = ...
328
+ ALIGNMENT_BYTES: ClassVar[CublasdxTensorTrait] = ...
329
+ OPAQUE_NAME: ClassVar[CublasdxTensorTrait] = ...
330
+ STORAGE_BYTES: ClassVar[CublasdxTensorTrait] = ...
331
+ UID: ClassVar[CublasdxTensorTrait] = ...
332
+ _generate_next_value_: ClassVar[Callable] = ...
333
+ _member_map_: ClassVar[dict] = ...
334
+ _member_names_: ClassVar[list] = ...
335
+ _member_type_: ClassVar[type[int]] = ...
336
+ _unhashable_values_: ClassVar[list] = ...
337
+ _use_args_: ClassVar[bool] = ...
338
+ _value2member_map_: ClassVar[dict] = ...
339
+ def __format__(self, *args, **kwargs) -> str: ...
340
+
341
+ class CublasdxTensorType(enum.IntEnum):
342
+ __new__: ClassVar[Callable] = ...
343
+ GMEM_A: ClassVar[CublasdxTensorType] = ...
344
+ GMEM_B: ClassVar[CublasdxTensorType] = ...
345
+ GMEM_C: ClassVar[CublasdxTensorType] = ...
346
+ SMEM_A: ClassVar[CublasdxTensorType] = ...
347
+ SMEM_B: ClassVar[CublasdxTensorType] = ...
348
+ SMEM_C: ClassVar[CublasdxTensorType] = ...
349
+ SUGGESTED_RMEM_C: ClassVar[CublasdxTensorType] = ...
350
+ SUGGESTED_SMEM_A: ClassVar[CublasdxTensorType] = ...
351
+ SUGGESTED_SMEM_B: ClassVar[CublasdxTensorType] = ...
352
+ SUGGESTED_SMEM_C: ClassVar[CublasdxTensorType] = ...
353
+ _generate_next_value_: ClassVar[Callable] = ...
354
+ _member_map_: ClassVar[dict] = ...
355
+ _member_names_: ClassVar[list] = ...
356
+ _member_type_: ClassVar[type[int]] = ...
357
+ _unhashable_values_: ClassVar[list] = ...
358
+ _use_args_: ClassVar[bool] = ...
359
+ _value2member_map_: ClassVar[dict] = ...
360
+ def __format__(self, *args, **kwargs) -> str: ...
361
+
362
+ class CublasdxTraitType(enum.IntEnum):
363
+ __new__: ClassVar[Callable] = ...
364
+ ALIGNMENT: ClassVar[CublasdxTraitType] = ...
365
+ ARRANGEMENT: ClassVar[CublasdxTraitType] = ...
366
+ BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
367
+ BLOCK_SIZE: ClassVar[CublasdxTraitType] = ...
368
+ LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
369
+ MAX_THREADS_PER_BLOCK: ClassVar[CublasdxTraitType] = ...
370
+ SIZE: ClassVar[CublasdxTraitType] = ...
371
+ SUGGESTED_BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
372
+ SUGGESTED_LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
373
+ SYMBOL_NAME: ClassVar[CublasdxTraitType] = ...
374
+ VALUE_TYPE: ClassVar[CublasdxTraitType] = ...
375
+ _generate_next_value_: ClassVar[Callable] = ...
376
+ _member_map_: ClassVar[dict] = ...
377
+ _member_names_: ClassVar[list] = ...
378
+ _member_type_: ClassVar[type[int]] = ...
379
+ _unhashable_values_: ClassVar[list] = ...
380
+ _use_args_: ClassVar[bool] = ...
381
+ _value2member_map_: ClassVar[dict] = ...
382
+ def __format__(self, *args, **kwargs) -> str: ...
383
+
384
+ class CublasdxTransposeMode(enum.IntEnum):
385
+ __new__: ClassVar[Callable] = ...
386
+ CONJ_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
387
+ NON_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
388
+ TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
389
+ _generate_next_value_: ClassVar[Callable] = ...
390
+ _member_map_: ClassVar[dict] = ...
391
+ _member_names_: ClassVar[list] = ...
392
+ _member_type_: ClassVar[type[int]] = ...
393
+ _unhashable_values_: ClassVar[list] = ...
394
+ _use_args_: ClassVar[bool] = ...
395
+ _value2member_map_: ClassVar[dict] = ...
396
+ def __format__(self, *args, **kwargs) -> str: ...
397
+
398
+ class CublasdxType(enum.IntEnum):
399
+ __new__: ClassVar[Callable] = ...
400
+ COMPLEX: ClassVar[CublasdxType] = ...
401
+ REAL: ClassVar[CublasdxType] = ...
402
+ _generate_next_value_: ClassVar[Callable] = ...
403
+ _member_map_: ClassVar[dict] = ...
404
+ _member_names_: ClassVar[list] = ...
405
+ _member_type_: ClassVar[type[int]] = ...
406
+ _unhashable_values_: ClassVar[list] = ...
407
+ _use_args_: ClassVar[bool] = ...
408
+ _value2member_map_: ClassVar[dict] = ...
409
+ def __format__(self, *args, **kwargs) -> str: ...
410
+
411
+ class CufftdxApi(enum.IntEnum):
412
+ __new__: ClassVar[Callable] = ...
413
+ LMEM: ClassVar[CufftdxApi] = ...
414
+ SMEM: ClassVar[CufftdxApi] = ...
415
+ _generate_next_value_: ClassVar[Callable] = ...
416
+ _member_map_: ClassVar[dict] = ...
417
+ _member_names_: ClassVar[list] = ...
418
+ _member_type_: ClassVar[type[int]] = ...
419
+ _unhashable_values_: ClassVar[list] = ...
420
+ _use_args_: ClassVar[bool] = ...
421
+ _value2member_map_: ClassVar[dict] = ...
422
+ def __format__(self, *args, **kwargs) -> str: ...
423
+
424
+ class CufftdxCodeType(enum.IntEnum):
425
+ __new__: ClassVar[Callable] = ...
426
+ LTOIR: ClassVar[CufftdxCodeType] = ...
427
+ PTX: ClassVar[CufftdxCodeType] = ...
428
+ _generate_next_value_: ClassVar[Callable] = ...
429
+ _member_map_: ClassVar[dict] = ...
430
+ _member_names_: ClassVar[list] = ...
431
+ _member_type_: ClassVar[type[int]] = ...
432
+ _unhashable_values_: ClassVar[list] = ...
433
+ _use_args_: ClassVar[bool] = ...
434
+ _value2member_map_: ClassVar[dict] = ...
435
+ def __format__(self, *args, **kwargs) -> str: ...
436
+
437
+ class CufftdxComplexLayout(enum.IntEnum):
438
+ __new__: ClassVar[Callable] = ...
439
+ FULL: ClassVar[CufftdxComplexLayout] = ...
440
+ NATURAL: ClassVar[CufftdxComplexLayout] = ...
441
+ PACKED: ClassVar[CufftdxComplexLayout] = ...
442
+ _generate_next_value_: ClassVar[Callable] = ...
443
+ _member_map_: ClassVar[dict] = ...
444
+ _member_names_: ClassVar[list] = ...
445
+ _member_type_: ClassVar[type[int]] = ...
446
+ _unhashable_values_: ClassVar[list] = ...
447
+ _use_args_: ClassVar[bool] = ...
448
+ _value2member_map_: ClassVar[dict] = ...
449
+ def __format__(self, *args, **kwargs) -> str: ...
450
+
451
+ class CufftdxDirection(enum.IntEnum):
452
+ __new__: ClassVar[Callable] = ...
453
+ FORWARD: ClassVar[CufftdxDirection] = ...
454
+ INVERSE: ClassVar[CufftdxDirection] = ...
455
+ _generate_next_value_: ClassVar[Callable] = ...
456
+ _member_map_: ClassVar[dict] = ...
457
+ _member_names_: ClassVar[list] = ...
458
+ _member_type_: ClassVar[type[int]] = ...
459
+ _unhashable_values_: ClassVar[list] = ...
460
+ _use_args_: ClassVar[bool] = ...
461
+ _value2member_map_: ClassVar[dict] = ...
462
+ def __format__(self, *args, **kwargs) -> str: ...
463
+
464
+ class CufftdxKnobType(enum.IntEnum):
465
+ __new__: ClassVar[Callable] = ...
466
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxKnobType] = ...
467
+ FFTS_PER_BLOCK: ClassVar[CufftdxKnobType] = ...
468
+ _generate_next_value_: ClassVar[Callable] = ...
469
+ _member_map_: ClassVar[dict] = ...
470
+ _member_names_: ClassVar[list] = ...
471
+ _member_type_: ClassVar[type[int]] = ...
472
+ _unhashable_values_: ClassVar[list] = ...
473
+ _use_args_: ClassVar[bool] = ...
474
+ _value2member_map_: ClassVar[dict] = ...
475
+ def __format__(self, *args, **kwargs) -> str: ...
476
+
477
+ class CufftdxOperatorType(enum.IntEnum):
478
+ __new__: ClassVar[Callable] = ...
479
+ API: ClassVar[CufftdxOperatorType] = ...
480
+ BLOCK_DIM: ClassVar[CufftdxOperatorType] = ...
481
+ CODE_TYPE: ClassVar[CufftdxOperatorType] = ...
482
+ DIRECTION: ClassVar[CufftdxOperatorType] = ...
483
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxOperatorType] = ...
484
+ EXECUTION: ClassVar[CufftdxOperatorType] = ...
485
+ FFTS_PER_BLOCK: ClassVar[CufftdxOperatorType] = ...
486
+ PRECISION: ClassVar[CufftdxOperatorType] = ...
487
+ REAL_FFT_OPTIONS: ClassVar[CufftdxOperatorType] = ...
488
+ SIZE: ClassVar[CufftdxOperatorType] = ...
489
+ SM: ClassVar[CufftdxOperatorType] = ...
490
+ TYPE: ClassVar[CufftdxOperatorType] = ...
491
+ _generate_next_value_: ClassVar[Callable] = ...
492
+ _member_map_: ClassVar[dict] = ...
493
+ _member_names_: ClassVar[list] = ...
494
+ _member_type_: ClassVar[type[int]] = ...
495
+ _unhashable_values_: ClassVar[list] = ...
496
+ _use_args_: ClassVar[bool] = ...
497
+ _value2member_map_: ClassVar[dict] = ...
498
+ def __format__(self, *args, **kwargs) -> str: ...
499
+
500
+ class CufftdxRealMode(enum.IntEnum):
501
+ __new__: ClassVar[Callable] = ...
502
+ FOLDED: ClassVar[CufftdxRealMode] = ...
503
+ NORMAL: ClassVar[CufftdxRealMode] = ...
504
+ _generate_next_value_: ClassVar[Callable] = ...
505
+ _member_map_: ClassVar[dict] = ...
506
+ _member_names_: ClassVar[list] = ...
507
+ _member_type_: ClassVar[type[int]] = ...
508
+ _unhashable_values_: ClassVar[list] = ...
509
+ _use_args_: ClassVar[bool] = ...
510
+ _value2member_map_: ClassVar[dict] = ...
511
+ def __format__(self, *args, **kwargs) -> str: ...
512
+
513
+ class CufftdxTraitType(enum.IntEnum):
514
+ __new__: ClassVar[Callable] = ...
515
+ BLOCK_DIM: ClassVar[CufftdxTraitType] = ...
516
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
517
+ FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
518
+ IMPLICIT_TYPE_BATCHING: ClassVar[CufftdxTraitType] = ...
519
+ INPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
520
+ INPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
521
+ INPUT_TYPE: ClassVar[CufftdxTraitType] = ...
522
+ OUTPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
523
+ OUTPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
524
+ OUTPUT_TYPE: ClassVar[CufftdxTraitType] = ...
525
+ SHARED_MEMORY_SIZE: ClassVar[CufftdxTraitType] = ...
526
+ STORAGE_SIZE: ClassVar[CufftdxTraitType] = ...
527
+ STRIDE: ClassVar[CufftdxTraitType] = ...
528
+ SUGGESTED_FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
529
+ SYMBOL_NAME: ClassVar[CufftdxTraitType] = ...
530
+ VALUE_TYPE: ClassVar[CufftdxTraitType] = ...
531
+ _generate_next_value_: ClassVar[Callable] = ...
532
+ _member_map_: ClassVar[dict] = ...
533
+ _member_names_: ClassVar[list] = ...
534
+ _member_type_: ClassVar[type[int]] = ...
535
+ _unhashable_values_: ClassVar[list] = ...
536
+ _use_args_: ClassVar[bool] = ...
537
+ _value2member_map_: ClassVar[dict] = ...
538
+ def __format__(self, *args, **kwargs) -> str: ...
539
+
540
+ class CufftdxType(enum.IntEnum):
541
+ __new__: ClassVar[Callable] = ...
542
+ C2C: ClassVar[CufftdxType] = ...
543
+ C2R: ClassVar[CufftdxType] = ...
544
+ R2C: ClassVar[CufftdxType] = ...
545
+ _generate_next_value_: ClassVar[Callable] = ...
546
+ _member_map_: ClassVar[dict] = ...
547
+ _member_names_: ClassVar[list] = ...
548
+ _member_type_: ClassVar[type[int]] = ...
549
+ _unhashable_values_: ClassVar[list] = ...
550
+ _use_args_: ClassVar[bool] = ...
551
+ _value2member_map_: ClassVar[dict] = ...
552
+ def __format__(self, *args, **kwargs) -> str: ...
553
+
554
+ class CusolverdxApi(enum.IntEnum):
555
+ __new__: ClassVar[Callable] = ...
556
+ SMEM: ClassVar[CusolverdxApi] = ...
557
+ SMEM_DYNAMIC_LD: ClassVar[CusolverdxApi] = ...
558
+ _generate_next_value_: ClassVar[Callable] = ...
559
+ _member_map_: ClassVar[dict] = ...
560
+ _member_names_: ClassVar[list] = ...
561
+ _member_type_: ClassVar[type[int]] = ...
562
+ _unhashable_values_: ClassVar[list] = ...
563
+ _use_args_: ClassVar[bool] = ...
564
+ _value2member_map_: ClassVar[dict] = ...
565
+ def __format__(self, *args, **kwargs) -> str: ...
566
+
567
+ class CusolverdxArrangement(enum.IntEnum):
568
+ __new__: ClassVar[Callable] = ...
569
+ COL_MAJOR: ClassVar[CusolverdxArrangement] = ...
570
+ ROW_MAJOR: ClassVar[CusolverdxArrangement] = ...
571
+ _generate_next_value_: ClassVar[Callable] = ...
572
+ _member_map_: ClassVar[dict] = ...
573
+ _member_names_: ClassVar[list] = ...
574
+ _member_type_: ClassVar[type[int]] = ...
575
+ _unhashable_values_: ClassVar[list] = ...
576
+ _use_args_: ClassVar[bool] = ...
577
+ _value2member_map_: ClassVar[dict] = ...
578
+ def __format__(self, *args, **kwargs) -> str: ...
579
+
580
+ class CusolverdxDiag(enum.IntEnum):
581
+ __new__: ClassVar[Callable] = ...
582
+ NON_UNIT: ClassVar[CusolverdxDiag] = ...
583
+ UNIT: ClassVar[CusolverdxDiag] = ...
584
+ _generate_next_value_: ClassVar[Callable] = ...
585
+ _member_map_: ClassVar[dict] = ...
586
+ _member_names_: ClassVar[list] = ...
587
+ _member_type_: ClassVar[type[int]] = ...
588
+ _unhashable_values_: ClassVar[list] = ...
589
+ _use_args_: ClassVar[bool] = ...
590
+ _value2member_map_: ClassVar[dict] = ...
591
+ def __format__(self, *args, **kwargs) -> str: ...
592
+
593
+ class CusolverdxFillMode(enum.IntEnum):
594
+ __new__: ClassVar[Callable] = ...
595
+ LOWER: ClassVar[CusolverdxFillMode] = ...
596
+ UPPER: ClassVar[CusolverdxFillMode] = ...
597
+ _generate_next_value_: ClassVar[Callable] = ...
598
+ _member_map_: ClassVar[dict] = ...
599
+ _member_names_: ClassVar[list] = ...
600
+ _member_type_: ClassVar[type[int]] = ...
601
+ _unhashable_values_: ClassVar[list] = ...
602
+ _use_args_: ClassVar[bool] = ...
603
+ _value2member_map_: ClassVar[dict] = ...
604
+ def __format__(self, *args, **kwargs) -> str: ...
605
+
606
+ class CusolverdxFunction(enum.IntEnum):
607
+ __new__: ClassVar[Callable] = ...
608
+ GETRF_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
609
+ GETRS_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
610
+ POTRF: ClassVar[CusolverdxFunction] = ...
611
+ POTRS: ClassVar[CusolverdxFunction] = ...
612
+ TRSM: ClassVar[CusolverdxFunction] = ...
613
+ _generate_next_value_: ClassVar[Callable] = ...
614
+ _member_map_: ClassVar[dict] = ...
615
+ _member_names_: ClassVar[list] = ...
616
+ _member_type_: ClassVar[type[int]] = ...
617
+ _unhashable_values_: ClassVar[list] = ...
618
+ _use_args_: ClassVar[bool] = ...
619
+ _value2member_map_: ClassVar[dict] = ...
620
+ def __format__(self, *args, **kwargs) -> str: ...
621
+
622
+ class CusolverdxOperatorType(enum.IntEnum):
623
+ __new__: ClassVar[Callable] = ...
624
+ API: ClassVar[CusolverdxOperatorType] = ...
625
+ ARRANGEMENT: ClassVar[CusolverdxOperatorType] = ...
626
+ BLOCK_DIM: ClassVar[CusolverdxOperatorType] = ...
627
+ DIAG: ClassVar[CusolverdxOperatorType] = ...
628
+ EXECUTION: ClassVar[CusolverdxOperatorType] = ...
629
+ FILL_MODE: ClassVar[CusolverdxOperatorType] = ...
630
+ FUNCTION: ClassVar[CusolverdxOperatorType] = ...
631
+ PRECISION: ClassVar[CusolverdxOperatorType] = ...
632
+ SIDE: ClassVar[CusolverdxOperatorType] = ...
633
+ SIZE: ClassVar[CusolverdxOperatorType] = ...
634
+ SM: ClassVar[CusolverdxOperatorType] = ...
635
+ TYPE: ClassVar[CusolverdxOperatorType] = ...
636
+ _generate_next_value_: ClassVar[Callable] = ...
637
+ _member_map_: ClassVar[dict] = ...
638
+ _member_names_: ClassVar[list] = ...
639
+ _member_type_: ClassVar[type[int]] = ...
640
+ _unhashable_values_: ClassVar[list] = ...
641
+ _use_args_: ClassVar[bool] = ...
642
+ _value2member_map_: ClassVar[dict] = ...
643
+ def __format__(self, *args, **kwargs) -> str: ...
644
+
645
+ class CusolverdxSide(enum.IntEnum):
646
+ __new__: ClassVar[Callable] = ...
647
+ LEFT: ClassVar[CusolverdxSide] = ...
648
+ RIGHT: ClassVar[CusolverdxSide] = ...
649
+ _generate_next_value_: ClassVar[Callable] = ...
650
+ _member_map_: ClassVar[dict] = ...
651
+ _member_names_: ClassVar[list] = ...
652
+ _member_type_: ClassVar[type[int]] = ...
653
+ _unhashable_values_: ClassVar[list] = ...
654
+ _use_args_: ClassVar[bool] = ...
655
+ _value2member_map_: ClassVar[dict] = ...
656
+ def __format__(self, *args, **kwargs) -> str: ...
657
+
658
+ class CusolverdxTraitType(enum.IntEnum):
659
+ __new__: ClassVar[Callable] = ...
660
+ SHARED_MEMORY_SIZE: ClassVar[CusolverdxTraitType] = ...
661
+ SYMBOL_NAME: ClassVar[CusolverdxTraitType] = ...
662
+ _generate_next_value_: ClassVar[Callable] = ...
663
+ _member_map_: ClassVar[dict] = ...
664
+ _member_names_: ClassVar[list] = ...
665
+ _member_type_: ClassVar[type[int]] = ...
666
+ _unhashable_values_: ClassVar[list] = ...
667
+ _use_args_: ClassVar[bool] = ...
668
+ _value2member_map_: ClassVar[dict] = ...
669
+ def __format__(self, *args, **kwargs) -> str: ...
670
+
671
+ class CusolverdxType(enum.IntEnum):
672
+ __new__: ClassVar[Callable] = ...
673
+ COMPLEX: ClassVar[CusolverdxType] = ...
674
+ REAL: ClassVar[CusolverdxType] = ...
675
+ _generate_next_value_: ClassVar[Callable] = ...
676
+ _member_map_: ClassVar[dict] = ...
677
+ _member_names_: ClassVar[list] = ...
678
+ _member_type_: ClassVar[type[int]] = ...
679
+ _unhashable_values_: ClassVar[list] = ...
680
+ _use_args_: ClassVar[bool] = ...
681
+ _value2member_map_: ClassVar[dict] = ...
682
+ def __format__(self, *args, **kwargs) -> str: ...
683
+
684
+ class LibMathDxError(Exception):
685
+ def __init__(self, status) -> Any: ...
686
+ def __reduce__(self) -> Any: ...
nvmath/bindings/nvpl/__init__.pxd ADDED
File without changes