nvmath-python 1.0.0__cp314-cp314t-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 (332) hide show
  1. nvmath/__init__.pxd +0 -0
  2. nvmath/__init__.py +45 -0
  3. nvmath/_internal/__init__.py +0 -0
  4. nvmath/_internal/attribute_ifc_factory.py +330 -0
  5. nvmath/_internal/layout.py +70 -0
  6. nvmath/_internal/templates.py +130 -0
  7. nvmath/_internal/threadsafe.py +106 -0
  8. nvmath/_internal/utils.py +43 -0
  9. nvmath/_internal/workspace.py +490 -0
  10. nvmath/_utils.py +147 -0
  11. nvmath/bindings/__init__.py +60 -0
  12. nvmath/bindings/_internal/__init__.pxd +0 -0
  13. nvmath/bindings/_internal/__init__.py +0 -0
  14. nvmath/bindings/_internal/common_types.pxd +31 -0
  15. nvmath/bindings/_internal/cublas.cp314t-win_amd64.pyd +0 -0
  16. nvmath/bindings/_internal/cublas.pxd +530 -0
  17. nvmath/bindings/_internal/cublasLt.cp314t-win_amd64.pyd +0 -0
  18. nvmath/bindings/_internal/cublasLt.pxd +59 -0
  19. nvmath/bindings/_internal/cublasMp.pxd +52 -0
  20. nvmath/bindings/_internal/cudss.cp314t-win_amd64.pyd +0 -0
  21. nvmath/bindings/_internal/cudss.pxd +54 -0
  22. nvmath/bindings/_internal/cufft.cp314t-win_amd64.pyd +0 -0
  23. nvmath/bindings/_internal/cufft.pxd +70 -0
  24. nvmath/bindings/_internal/cufftMp.pxd +77 -0
  25. nvmath/bindings/_internal/curand.cp314t-win_amd64.pyd +0 -0
  26. nvmath/bindings/_internal/curand.pxd +42 -0
  27. nvmath/bindings/_internal/cusolver.cp314t-win_amd64.pyd +0 -0
  28. nvmath/bindings/_internal/cusolver.pxd +15 -0
  29. nvmath/bindings/_internal/cusolverDn.cp314t-win_amd64.pyd +0 -0
  30. nvmath/bindings/_internal/cusolverDn.pxd +406 -0
  31. nvmath/bindings/_internal/cusolverMp.pxd +71 -0
  32. nvmath/bindings/_internal/cusolverSp.cp314t-win_amd64.pyd +0 -0
  33. nvmath/bindings/_internal/cusolverSp.pxd +75 -0
  34. nvmath/bindings/_internal/cusparse.cp314t-win_amd64.pyd +0 -0
  35. nvmath/bindings/_internal/cusparse.pxd +471 -0
  36. nvmath/bindings/_internal/cusparseLt.cp314t-win_amd64.pyd +0 -0
  37. nvmath/bindings/_internal/cusparseLt.pxd +48 -0
  38. nvmath/bindings/_internal/cutensor.cp314t-win_amd64.pyd +0 -0
  39. nvmath/bindings/_internal/cutensor.pxd +58 -0
  40. nvmath/bindings/_internal/mathdx.cp314t-win_amd64.pyd +0 -0
  41. nvmath/bindings/_internal/mathdx.pxd +116 -0
  42. nvmath/bindings/_internal/nvshmem.pxd +29 -0
  43. nvmath/bindings/_internal/utils.cp314t-win_amd64.pyd +0 -0
  44. nvmath/bindings/_internal/utils.pxd +174 -0
  45. nvmath/bindings/_internal/utils.pyi +10 -0
  46. nvmath/bindings/cublas.cp314t-win_amd64.pyd +0 -0
  47. nvmath/bindings/cublas.pxd +558 -0
  48. nvmath/bindings/cublas.pyi +812 -0
  49. nvmath/bindings/cublasLt.cp314t-win_amd64.pyd +0 -0
  50. nvmath/bindings/cublasLt.pxd +109 -0
  51. nvmath/bindings/cublasLt.pyi +1461 -0
  52. nvmath/bindings/cublasMp.pxd +85 -0
  53. nvmath/bindings/cublasMp.pyi +267 -0
  54. nvmath/bindings/cudss.cp314t-win_amd64.pyd +0 -0
  55. nvmath/bindings/cudss.pxd +98 -0
  56. nvmath/bindings/cudss.pyi +443 -0
  57. nvmath/bindings/cufft.cp314t-win_amd64.pyd +0 -0
  58. nvmath/bindings/cufft.pxd +118 -0
  59. nvmath/bindings/cufft.pyi +301 -0
  60. nvmath/bindings/cufftMp.pxd +124 -0
  61. nvmath/bindings/cufftMp.pyi +326 -0
  62. nvmath/bindings/curand.cp314t-win_amd64.pyd +0 -0
  63. nvmath/bindings/curand.pxd +71 -0
  64. nvmath/bindings/curand.pyi +189 -0
  65. nvmath/bindings/cusolver.cp314t-win_amd64.pyd +0 -0
  66. nvmath/bindings/cusolver.pxd +62 -0
  67. nvmath/bindings/cusolver.pyi +320 -0
  68. nvmath/bindings/cusolverDn.cp314t-win_amd64.pyd +0 -0
  69. nvmath/bindings/cusolverDn.pxd +430 -0
  70. nvmath/bindings/cusolverDn.pyi +422 -0
  71. nvmath/bindings/cusolverMp.pxd +98 -0
  72. nvmath/bindings/cusolverMp.pyi +114 -0
  73. nvmath/bindings/cusolverSp.cp314t-win_amd64.pyd +0 -0
  74. nvmath/bindings/cusolverSp.pxd +95 -0
  75. nvmath/bindings/cusolverSp.pyi +70 -0
  76. nvmath/bindings/cusparse.cp314t-win_amd64.pyd +0 -0
  77. nvmath/bindings/cusparse.pxd +546 -0
  78. nvmath/bindings/cusparse.pyi +1017 -0
  79. nvmath/bindings/cusparseLt.cp314t-win_amd64.pyd +0 -0
  80. nvmath/bindings/cusparseLt.pxd +99 -0
  81. nvmath/bindings/cusparseLt.pyi +252 -0
  82. nvmath/bindings/cutensor.cp314t-win_amd64.pyd +0 -0
  83. nvmath/bindings/cutensor.pxd +98 -0
  84. nvmath/bindings/cutensor.pyi +324 -0
  85. nvmath/bindings/cycublas.cp314t-win_amd64.pyd +0 -0
  86. nvmath/bindings/cycublas.pxd +664 -0
  87. nvmath/bindings/cycublasLt.cp314t-win_amd64.pyd +0 -0
  88. nvmath/bindings/cycublasLt.pxd +1045 -0
  89. nvmath/bindings/cycublasMp.pxd +171 -0
  90. nvmath/bindings/cycudss.cp314t-win_amd64.pyd +0 -0
  91. nvmath/bindings/cycudss.pxd +277 -0
  92. nvmath/bindings/cycufft.cp314t-win_amd64.pyd +0 -0
  93. nvmath/bindings/cycufft.pxd +333 -0
  94. nvmath/bindings/cycufftMp.pxd +342 -0
  95. nvmath/bindings/cycurand.cp314t-win_amd64.pyd +0 -0
  96. nvmath/bindings/cycurand.pxd +141 -0
  97. nvmath/bindings/cycusolver.cp314t-win_amd64.pyd +0 -0
  98. nvmath/bindings/cycusolver.pxd +137 -0
  99. nvmath/bindings/cycusolverDn.cp314t-win_amd64.pyd +0 -0
  100. nvmath/bindings/cycusolverDn.pxd +443 -0
  101. nvmath/bindings/cycusolverMp.pxd +107 -0
  102. nvmath/bindings/cycusolverSp.cp314t-win_amd64.pyd +0 -0
  103. nvmath/bindings/cycusolverSp.pxd +93 -0
  104. nvmath/bindings/cycusparse.cp314t-win_amd64.pyd +0 -0
  105. nvmath/bindings/cycusparse.pxd +679 -0
  106. nvmath/bindings/cycusparseLt.cp314t-win_amd64.pyd +0 -0
  107. nvmath/bindings/cycusparseLt.pxd +135 -0
  108. nvmath/bindings/cycutensor.cp314t-win_amd64.pyd +0 -0
  109. nvmath/bindings/cycutensor.pxd +189 -0
  110. nvmath/bindings/cymathdx.cp314t-win_amd64.pyd +0 -0
  111. nvmath/bindings/cymathdx.pxd +552 -0
  112. nvmath/bindings/cynvshmem.pxd +118 -0
  113. nvmath/bindings/mathdx.cp314t-win_amd64.pyd +0 -0
  114. nvmath/bindings/mathdx.pxd +182 -0
  115. nvmath/bindings/mathdx.pyi +1562 -0
  116. nvmath/bindings/nvpl/__init__.pxd +0 -0
  117. nvmath/bindings/nvpl/__init__.py +13 -0
  118. nvmath/bindings/nvpl/_internal/__init__.pxd +0 -0
  119. nvmath/bindings/nvpl/_internal/__init__.py +0 -0
  120. nvmath/bindings/nvpl/_internal/blas.cp314t-win_amd64.pyd +0 -0
  121. nvmath/bindings/nvpl/_internal/blas.pxd +237 -0
  122. nvmath/bindings/nvpl/_internal/fft.cp314t-win_amd64.pyd +0 -0
  123. nvmath/bindings/nvpl/_internal/fft.pxd +36 -0
  124. nvmath/bindings/nvpl/blas.cp314t-win_amd64.pyd +0 -0
  125. nvmath/bindings/nvpl/blas.pxd +131 -0
  126. nvmath/bindings/nvpl/blas.pyi +168 -0
  127. nvmath/bindings/nvpl/cyblas.cp314t-win_amd64.pyd +0 -0
  128. nvmath/bindings/nvpl/cyblas.pxd +280 -0
  129. nvmath/bindings/nvpl/cyfft.cp314t-win_amd64.pyd +0 -0
  130. nvmath/bindings/nvpl/cyfft.pxd +93 -0
  131. nvmath/bindings/nvpl/fft.cp314t-win_amd64.pyd +0 -0
  132. nvmath/bindings/nvpl/fft.pxd +100 -0
  133. nvmath/bindings/nvpl/fft.pyi +168 -0
  134. nvmath/bindings/nvshmem.pxd +54 -0
  135. nvmath/bindings/nvshmem.pyi +191 -0
  136. nvmath/device/__init__.py +38 -0
  137. nvmath/device/_deprecated.py +33 -0
  138. nvmath/device/common.py +315 -0
  139. nvmath/device/common_backend.py +131 -0
  140. nvmath/device/common_cuda.py +201 -0
  141. nvmath/device/common_numba.py +300 -0
  142. nvmath/device/common_numba_cuda_mlir.py +202 -0
  143. nvmath/device/common_opaque_tensor.py +201 -0
  144. nvmath/device/cublasdx.py +1606 -0
  145. nvmath/device/cublasdx_backend.py +860 -0
  146. nvmath/device/cublasdx_numba.py +1534 -0
  147. nvmath/device/cublasdx_numba_cuda_mlir.py +208 -0
  148. nvmath/device/cufftdx.py +373 -0
  149. nvmath/device/cufftdx_backend.py +220 -0
  150. nvmath/device/cufftdx_numba.py +140 -0
  151. nvmath/device/cufftdx_numba_cuda_mlir.py +79 -0
  152. nvmath/device/curand_kernel.py +9147 -0
  153. nvmath/device/cusolverdx.py +2708 -0
  154. nvmath/device/cusolverdx_backend.py +440 -0
  155. nvmath/device/cusolverdx_numba.py +567 -0
  156. nvmath/device/cusolverdx_numba_cuda_mlir.py +604 -0
  157. nvmath/device/cusolverdx_overload_backend.py +1029 -0
  158. nvmath/device/llvm_array.py +29 -0
  159. nvmath/device/random.py +441 -0
  160. nvmath/device/random_helpers.py +23 -0
  161. nvmath/device/random_states.py +187 -0
  162. nvmath/device/types.py +138 -0
  163. nvmath/device/vector_types_numba.py +259 -0
  164. nvmath/distributed/__init__.py +200 -0
  165. nvmath/distributed/_internal/__init__.py +0 -0
  166. nvmath/distributed/_internal/nccl.py +86 -0
  167. nvmath/distributed/_internal/nvshmem.py +307 -0
  168. nvmath/distributed/_internal/symmetric_memory.py +35 -0
  169. nvmath/distributed/_internal/tensor_ifc.py +70 -0
  170. nvmath/distributed/_internal/tensor_ifc_cupy.py +68 -0
  171. nvmath/distributed/_internal/tensor_ifc_host_device.py +172 -0
  172. nvmath/distributed/_internal/tensor_ifc_numpy.py +46 -0
  173. nvmath/distributed/_internal/tensor_ifc_torch.py +162 -0
  174. nvmath/distributed/_internal/tensor_wrapper.py +81 -0
  175. nvmath/distributed/_utils.py +167 -0
  176. nvmath/distributed/distribution/__init__.py +30 -0
  177. nvmath/distributed/distribution/_configuration.py +39 -0
  178. nvmath/distributed/distribution/distributions.py +1024 -0
  179. nvmath/distributed/distribution/redistribute.py +1284 -0
  180. nvmath/distributed/fft/__init__.py +7 -0
  181. nvmath/distributed/fft/_configuration.py +82 -0
  182. nvmath/distributed/fft/fft.py +2742 -0
  183. nvmath/distributed/linalg/__init__.py +22 -0
  184. nvmath/distributed/linalg/_internal/__init__.py +3 -0
  185. nvmath/distributed/linalg/_internal/epilog_protocol.py +586 -0
  186. nvmath/distributed/linalg/_internal/matmul_desc_ifc.py +28 -0
  187. nvmath/distributed/linalg/advanced/__init__.py +8 -0
  188. nvmath/distributed/linalg/advanced/_configuration.py +171 -0
  189. nvmath/distributed/linalg/advanced/matmulmod.py +3573 -0
  190. nvmath/distributed/linalg/generic/__init__.py +8 -0
  191. nvmath/distributed/linalg/generic/_caching.py +66 -0
  192. nvmath/distributed/linalg/generic/_configuration.py +61 -0
  193. nvmath/distributed/linalg/generic/_factorization.py +172 -0
  194. nvmath/distributed/linalg/generic/_initialization.py +966 -0
  195. nvmath/distributed/linalg/generic/_problem_spec.py +511 -0
  196. nvmath/distributed/linalg/generic/solvermod.py +1368 -0
  197. nvmath/distributed/process_group.py +408 -0
  198. nvmath/fft/__init__.py +7 -0
  199. nvmath/fft/_configuration.py +189 -0
  200. nvmath/fft/_exec_utils.py +82 -0
  201. nvmath/fft/_helpers.py +237 -0
  202. nvmath/fft/fft.py +3122 -0
  203. nvmath/internal/__init__.pxd +3 -0
  204. nvmath/internal/__init__.py +10 -0
  205. nvmath/internal/_bindings.cp314t-win_amd64.pyd +0 -0
  206. nvmath/internal/_bindings.pxd +18 -0
  207. nvmath/internal/_device_utils.py +45 -0
  208. nvmath/internal/_layout/__init__.pxd +3 -0
  209. nvmath/internal/_layout/__init__.py +7 -0
  210. nvmath/internal/_layout/_layout.cp314t-win_amd64.pyd +0 -0
  211. nvmath/internal/_layout/_layout.pxd +1303 -0
  212. nvmath/internal/_layout/_layout.pyi +1145 -0
  213. nvmath/internal/enum_utils.py +142 -0
  214. nvmath/internal/formatters.py +87 -0
  215. nvmath/internal/mem_limit.py +51 -0
  216. nvmath/internal/memory.cp314t-win_amd64.pyd +0 -0
  217. nvmath/internal/memory.pxd +13 -0
  218. nvmath/internal/memory.pyi +50 -0
  219. nvmath/internal/ndbuffer/__init__.pxd +3 -0
  220. nvmath/internal/ndbuffer/__init__.py +9 -0
  221. nvmath/internal/ndbuffer/_copy_kernel.cp314t-win_amd64.pyd +0 -0
  222. nvmath/internal/ndbuffer/_copy_kernel.pxd +10 -0
  223. nvmath/internal/ndbuffer/_jit.cp314t-win_amd64.pyd +0 -0
  224. nvmath/internal/ndbuffer/_jit.pxd +7 -0
  225. nvmath/internal/ndbuffer/_ndbuffer.cp314t-win_amd64.pyd +0 -0
  226. nvmath/internal/ndbuffer/_ndbuffer.pxd +40 -0
  227. nvmath/internal/ndbuffer/_ndbuffer.pyi +463 -0
  228. nvmath/internal/ndbuffer/copy_kernel/args.h +34 -0
  229. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/array_view.h +52 -0
  230. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/elementwise.h +68 -0
  231. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/grid_indexer.h +69 -0
  232. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/transposed.h +242 -0
  233. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/type_utils.h +39 -0
  234. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/utils.h +132 -0
  235. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/vec.h +159 -0
  236. nvmath/internal/ndbuffer/copy_kernel/elementwise.h +53 -0
  237. nvmath/internal/ndbuffer/copy_kernel/transposed.h +58 -0
  238. nvmath/internal/package_ifc.py +168 -0
  239. nvmath/internal/package_ifc_cuda.py +57 -0
  240. nvmath/internal/package_ifc_cupy.py +67 -0
  241. nvmath/internal/package_ifc_torch.py +69 -0
  242. nvmath/internal/package_wrapper.py +14 -0
  243. nvmath/internal/tensor_ifc.py +179 -0
  244. nvmath/internal/tensor_ifc_cupy.py +234 -0
  245. nvmath/internal/tensor_ifc_ndbuffer.py +147 -0
  246. nvmath/internal/tensor_ifc_numpy.py +184 -0
  247. nvmath/internal/tensor_ifc_torch.py +178 -0
  248. nvmath/internal/tensor_wrapper.py +160 -0
  249. nvmath/internal/typemaps.py +113 -0
  250. nvmath/internal/utils.py +805 -0
  251. nvmath/linalg/__init__.py +56 -0
  252. nvmath/linalg/_internal/__init__.py +3 -0
  253. nvmath/linalg/_internal/algo_cap_ifc.py +82 -0
  254. nvmath/linalg/_internal/algo_config_ifc.py +43 -0
  255. nvmath/linalg/_internal/batch.py +234 -0
  256. nvmath/linalg/_internal/enum_to_tuples.py +64 -0
  257. nvmath/linalg/_internal/epilog_protocol.py +766 -0
  258. nvmath/linalg/_internal/layout.py +624 -0
  259. nvmath/linalg/_internal/matmul_desc_ifc.py +28 -0
  260. nvmath/linalg/_internal/matmul_pref_ifc.py +27 -0
  261. nvmath/linalg/_internal/matrix_layout_ifc.py +26 -0
  262. nvmath/linalg/_internal/solver_utils.py +432 -0
  263. nvmath/linalg/_internal/typemaps.py +144 -0
  264. nvmath/linalg/_internal/utils.py +157 -0
  265. nvmath/linalg/advanced/__init__.py +8 -0
  266. nvmath/linalg/advanced/_algorithmmod.py +170 -0
  267. nvmath/linalg/advanced/_configuration.py +351 -0
  268. nvmath/linalg/advanced/helpers/__init__.py +5 -0
  269. nvmath/linalg/advanced/helpers/matmul.py +1316 -0
  270. nvmath/linalg/advanced/matmulmod.py +3734 -0
  271. nvmath/linalg/generic/__init__.py +53 -0
  272. nvmath/linalg/generic/_configuration/__init__.py +39 -0
  273. nvmath/linalg/generic/_configuration/layout.py +263 -0
  274. nvmath/linalg/generic/_configuration/match.py +734 -0
  275. nvmath/linalg/generic/_configuration/qualifiers.py +493 -0
  276. nvmath/linalg/generic/_configuration/solver_configuration.py +59 -0
  277. nvmath/linalg/generic/_configuration/wrap.py +217 -0
  278. nvmath/linalg/generic/_dtype.py +15 -0
  279. nvmath/linalg/generic/matmulmod.py +2094 -0
  280. nvmath/linalg/generic/solvermod.py +1301 -0
  281. nvmath/memory.py +279 -0
  282. nvmath/sparse/__init__.py +38 -0
  283. nvmath/sparse/_internal/__init__.py +21 -0
  284. nvmath/sparse/_internal/common_utils.py +147 -0
  285. nvmath/sparse/_internal/cudss_config_ifc.py +702 -0
  286. nvmath/sparse/_internal/cudss_data_ifc.py +399 -0
  287. nvmath/sparse/_internal/cudss_utils.py +506 -0
  288. nvmath/sparse/_internal/cusparse_utils.py +382 -0
  289. nvmath/sparse/_internal/sparse_bsc_ifc.py +303 -0
  290. nvmath/sparse/_internal/sparse_bsr_ifc.py +305 -0
  291. nvmath/sparse/_internal/sparse_coo_ifc.py +256 -0
  292. nvmath/sparse/_internal/sparse_csc_ifc.py +268 -0
  293. nvmath/sparse/_internal/sparse_csr_ifc.py +288 -0
  294. nvmath/sparse/_internal/sparse_dia_ifc.py +242 -0
  295. nvmath/sparse/_internal/sparse_format_helpers.py +601 -0
  296. nvmath/sparse/_internal/sparse_tensor_ifc.py +133 -0
  297. nvmath/sparse/_internal/sparse_ust_ifc.py +141 -0
  298. nvmath/sparse/_internal/utils.py +56 -0
  299. nvmath/sparse/advanced/__init__.py +7 -0
  300. nvmath/sparse/advanced/_configuration.py +227 -0
  301. nvmath/sparse/advanced/direct_solver.py +2069 -0
  302. nvmath/sparse/generic/__init__.py +7 -0
  303. nvmath/sparse/generic/_configuration.py +129 -0
  304. nvmath/sparse/generic/_helpers.py +137 -0
  305. nvmath/sparse/generic/_thunks.py +21 -0
  306. nvmath/sparse/generic/matmulmod.py +2353 -0
  307. nvmath/sparse/ust/__init__.py +7 -0
  308. nvmath/sparse/ust/_converters.py +422 -0
  309. nvmath/sparse/ust/_cpp.py +28 -0
  310. nvmath/sparse/ust/_drawer.py +565 -0
  311. nvmath/sparse/ust/_emitter.py +1033 -0
  312. nvmath/sparse/ust/_jit.py +188 -0
  313. nvmath/sparse/ust/_kernel.py +282 -0
  314. nvmath/sparse/ust/_utils.py +149 -0
  315. nvmath/sparse/ust/interfaces/__init__.py +0 -0
  316. nvmath/sparse/ust/interfaces/torch_interface.py +476 -0
  317. nvmath/sparse/ust/tensor.py +1016 -0
  318. nvmath/sparse/ust/tensor_format.py +957 -0
  319. nvmath/tensor/__init__.py +6 -0
  320. nvmath/tensor/_configuration.py +120 -0
  321. nvmath/tensor/_internal/__init__.py +3 -0
  322. nvmath/tensor/_internal/cutensor_config_ifc.py +279 -0
  323. nvmath/tensor/_internal/cutensor_utils.py +230 -0
  324. nvmath/tensor/_internal/data.py +43 -0
  325. nvmath/tensor/_internal/einsum_parser.py +444 -0
  326. nvmath/tensor/_internal/typemaps.py +96 -0
  327. nvmath/tensor/contract.py +1900 -0
  328. nvmath_python-1.0.0.dist-info/METADATA +134 -0
  329. nvmath_python-1.0.0.dist-info/RECORD +332 -0
  330. nvmath_python-1.0.0.dist-info/WHEEL +5 -0
  331. nvmath_python-1.0.0.dist-info/licenses/LICENSE +177 -0
  332. nvmath_python-1.0.0.dist-info/top_level.txt +2 -0
@@ -0,0 +1,1562 @@
1
+ # Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
2
+ #
3
+ # SPDX-License-Identifier: Apache-2.0
4
+
5
+ import _cython_3_2_5
6
+ import enum
7
+ from typing import Any, Callable, ClassVar
8
+
9
+ __pyx_capi__: dict
10
+ __test__: dict
11
+ check_status: _cython_3_2_5.cython_function_or_method
12
+ commondx_create_code: _cython_3_2_5.cython_function_or_method
13
+ commondx_destroy_code: _cython_3_2_5.cython_function_or_method
14
+ commondx_get_code_ltoir: _cython_3_2_5.cython_function_or_method
15
+ commondx_get_code_ltoir_size: _cython_3_2_5.cython_function_or_method
16
+ commondx_get_code_ltoir_sizes: _cython_3_2_5.cython_function_or_method
17
+ commondx_get_code_ltoirs: _cython_3_2_5.cython_function_or_method
18
+ commondx_get_code_num_ltoirs: _cython_3_2_5.cython_function_or_method
19
+ commondx_get_code_option_int64: _cython_3_2_5.cython_function_or_method
20
+ commondx_get_code_options_int64s: _cython_3_2_5.cython_function_or_method
21
+ commondx_get_code_ptx: _cython_3_2_5.cython_function_or_method
22
+ commondx_get_code_ptx_size: _cython_3_2_5.cython_function_or_method
23
+ commondx_set_code_option_int64: _cython_3_2_5.cython_function_or_method
24
+ commondx_set_code_option_int64s: _cython_3_2_5.cython_function_or_method
25
+ commondx_set_code_option_str: _cython_3_2_5.cython_function_or_method
26
+ commondx_status_to_str: _cython_3_2_5.cython_function_or_method
27
+ cublasdx_create_descriptor: _cython_3_2_5.cython_function_or_method
28
+ cublasdx_create_device_function: _cython_3_2_5.cython_function_or_method
29
+ cublasdx_create_device_function_with_pipelines: _cython_3_2_5.cython_function_or_method
30
+ cublasdx_create_device_pipeline: _cython_3_2_5.cython_function_or_method
31
+ cublasdx_create_tensor: _cython_3_2_5.cython_function_or_method
32
+ cublasdx_create_tensor_strided: _cython_3_2_5.cython_function_or_method
33
+ cublasdx_create_tile_pipeline: _cython_3_2_5.cython_function_or_method
34
+ cublasdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
35
+ cublasdx_destroy_device_function: _cython_3_2_5.cython_function_or_method
36
+ cublasdx_destroy_pipeline: _cython_3_2_5.cython_function_or_method
37
+ cublasdx_destroy_tensor: _cython_3_2_5.cython_function_or_method
38
+ cublasdx_finalize: _cython_3_2_5.cython_function_or_method
39
+ cublasdx_finalize_code: _cython_3_2_5.cython_function_or_method
40
+ cublasdx_finalize_device_functions: _cython_3_2_5.cython_function_or_method
41
+ cublasdx_finalize_pipelines: _cython_3_2_5.cython_function_or_method
42
+ cublasdx_finalize_tensors: _cython_3_2_5.cython_function_or_method
43
+ cublasdx_get_device_function_trait_str: _cython_3_2_5.cython_function_or_method
44
+ cublasdx_get_device_function_trait_str_size: _cython_3_2_5.cython_function_or_method
45
+ cublasdx_get_ltoir: _cython_3_2_5.cython_function_or_method
46
+ cublasdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
47
+ cublasdx_get_pipeline_trait_int64: _cython_3_2_5.cython_function_or_method
48
+ cublasdx_get_pipeline_trait_int64s: _cython_3_2_5.cython_function_or_method
49
+ cublasdx_get_pipeline_trait_str: _cython_3_2_5.cython_function_or_method
50
+ cublasdx_get_pipeline_trait_str_size: _cython_3_2_5.cython_function_or_method
51
+ cublasdx_get_tensor_trait_int64: _cython_3_2_5.cython_function_or_method
52
+ cublasdx_get_tensor_trait_str: _cython_3_2_5.cython_function_or_method
53
+ cublasdx_get_tensor_trait_str_size: _cython_3_2_5.cython_function_or_method
54
+ cublasdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
55
+ cublasdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
56
+ cublasdx_get_trait_str: _cython_3_2_5.cython_function_or_method
57
+ cublasdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
58
+ cublasdx_make_tensor_like: _cython_3_2_5.cython_function_or_method
59
+ cublasdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
60
+ cublasdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
61
+ cublasdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
62
+ cublasdx_set_option_str: _cython_3_2_5.cython_function_or_method
63
+ cublasdx_set_tensor_option_int64: _cython_3_2_5.cython_function_or_method
64
+ cublasdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
65
+ cufftdx_create_descriptor: _cython_3_2_5.cython_function_or_method
66
+ cufftdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
67
+ cufftdx_finalize_code: _cython_3_2_5.cython_function_or_method
68
+ cufftdx_get_knob_int64s: _cython_3_2_5.cython_function_or_method
69
+ cufftdx_get_knob_int64size: _cython_3_2_5.cython_function_or_method
70
+ cufftdx_get_ltoir: _cython_3_2_5.cython_function_or_method
71
+ cufftdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
72
+ cufftdx_get_trait_commondx_data_type: _cython_3_2_5.cython_function_or_method
73
+ cufftdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
74
+ cufftdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
75
+ cufftdx_get_trait_str: _cython_3_2_5.cython_function_or_method
76
+ cufftdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
77
+ cufftdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
78
+ cufftdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
79
+ cufftdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
80
+ cufftdx_set_option_str: _cython_3_2_5.cython_function_or_method
81
+ cufftdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
82
+ curanddx_create_descriptor: _cython_3_2_5.cython_function_or_method
83
+ curanddx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
84
+ curanddx_distribution_to_str: _cython_3_2_5.cython_function_or_method
85
+ curanddx_finalize_code: _cython_3_2_5.cython_function_or_method
86
+ curanddx_generate_method_to_str: _cython_3_2_5.cython_function_or_method
87
+ curanddx_generator_to_str: _cython_3_2_5.cython_function_or_method
88
+ curanddx_get_trait_int64: _cython_3_2_5.cython_function_or_method
89
+ curanddx_get_trait_str: _cython_3_2_5.cython_function_or_method
90
+ curanddx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
91
+ curanddx_get_version: _cython_3_2_5.cython_function_or_method
92
+ curanddx_normal_method_to_str: _cython_3_2_5.cython_function_or_method
93
+ curanddx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
94
+ curanddx_set_operator_doubles: _cython_3_2_5.cython_function_or_method
95
+ curanddx_set_operator_int64: _cython_3_2_5.cython_function_or_method
96
+ curanddx_set_option_str: _cython_3_2_5.cython_function_or_method
97
+ curanddx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
98
+ cusolverdx_create_descriptor: _cython_3_2_5.cython_function_or_method
99
+ cusolverdx_destroy_descriptor: _cython_3_2_5.cython_function_or_method
100
+ cusolverdx_finalize_code: _cython_3_2_5.cython_function_or_method
101
+ cusolverdx_get_ltoir: _cython_3_2_5.cython_function_or_method
102
+ cusolverdx_get_ltoir_size: _cython_3_2_5.cython_function_or_method
103
+ cusolverdx_get_trait_commondx_data_types: _cython_3_2_5.cython_function_or_method
104
+ cusolverdx_get_trait_int64: _cython_3_2_5.cython_function_or_method
105
+ cusolverdx_get_trait_int64s: _cython_3_2_5.cython_function_or_method
106
+ cusolverdx_get_trait_str: _cython_3_2_5.cython_function_or_method
107
+ cusolverdx_get_trait_str_size: _cython_3_2_5.cython_function_or_method
108
+ cusolverdx_get_universal_fatbin: _cython_3_2_5.cython_function_or_method
109
+ cusolverdx_get_universal_fatbin_size: _cython_3_2_5.cython_function_or_method
110
+ cusolverdx_operator_type_to_str: _cython_3_2_5.cython_function_or_method
111
+ cusolverdx_set_operator_int64: _cython_3_2_5.cython_function_or_method
112
+ cusolverdx_set_operator_int64s: _cython_3_2_5.cython_function_or_method
113
+ cusolverdx_set_option_str: _cython_3_2_5.cython_function_or_method
114
+ cusolverdx_trait_type_to_str: _cython_3_2_5.cython_function_or_method
115
+ get_version: _cython_3_2_5.cython_function_or_method
116
+ get_version_ex: _cython_3_2_5.cython_function_or_method
117
+
118
+ class CommondxArchModifier(enum.IntEnum):
119
+ """
120
+ The set of compute feature to target. See COMMONDX_OPTION_TARGET_SM
121
+ .See also https://docs.nvidia.com/cuda/cuda-c-programming-
122
+ guide/index.html#feature-set-compiler-targets
123
+
124
+ See `commondxArchModifier`.
125
+ """
126
+ __new__: ClassVar[Callable] = ...
127
+ ARCH_SPECIFIC: ClassVar[CommondxArchModifier] = ...
128
+ FAMILY_SPECIFIC: ClassVar[CommondxArchModifier] = ...
129
+ GENERIC: ClassVar[CommondxArchModifier] = ...
130
+ _generate_next_value_: ClassVar[Callable] = ...
131
+ _hashable_values_: ClassVar[list] = ...
132
+ _member_map_: ClassVar[dict] = ...
133
+ _member_names_: ClassVar[list] = ...
134
+ _member_type_: ClassVar[type[int]] = ...
135
+ _unhashable_values_: ClassVar[list] = ...
136
+ _unhashable_values_map_: ClassVar[dict] = ...
137
+ _use_args_: ClassVar[bool] = ...
138
+ _value2member_map_: ClassVar[dict] = ...
139
+ def __format__(self, *args, **kwargs) -> str:
140
+ """Convert to a string according to format_spec."""
141
+
142
+ class CommondxCodeContainer(enum.IntEnum):
143
+ """
144
+ The set of device code container types supported by the library. See
145
+ commondxOption_t::COMMONDX_OPTION_CODE_CONTAINER.
146
+
147
+ See `commondxCodeContainer`.
148
+ """
149
+ __new__: ClassVar[Callable] = ...
150
+ FATBIN: ClassVar[CommondxCodeContainer] = ...
151
+ LTOIR: ClassVar[CommondxCodeContainer] = ...
152
+ PTX: ClassVar[CommondxCodeContainer] = ...
153
+ _generate_next_value_: ClassVar[Callable] = ...
154
+ _hashable_values_: ClassVar[list] = ...
155
+ _member_map_: ClassVar[dict] = ...
156
+ _member_names_: ClassVar[list] = ...
157
+ _member_type_: ClassVar[type[int]] = ...
158
+ _unhashable_values_: ClassVar[list] = ...
159
+ _unhashable_values_map_: ClassVar[dict] = ...
160
+ _use_args_: ClassVar[bool] = ...
161
+ _value2member_map_: ClassVar[dict] = ...
162
+ def __format__(self, *args, **kwargs) -> str:
163
+ """Convert to a string according to format_spec."""
164
+
165
+ class CommondxCodeType(enum.IntEnum):
166
+ """
167
+ The set of code types supported by the library. See
168
+ commondxOption_t::COMMONDX_OPTION_TARGET_CODE.
169
+
170
+ See `commondxCodeType`.
171
+ """
172
+ __new__: ClassVar[Callable] = ...
173
+ LTO: ClassVar[CommondxCodeType] = ...
174
+ PTX: ClassVar[CommondxCodeType] = ...
175
+ _generate_next_value_: ClassVar[Callable] = ...
176
+ _hashable_values_: ClassVar[list] = ...
177
+ _member_map_: ClassVar[dict] = ...
178
+ _member_names_: ClassVar[list] = ...
179
+ _member_type_: ClassVar[type[int]] = ...
180
+ _unhashable_values_: ClassVar[list] = ...
181
+ _unhashable_values_map_: ClassVar[dict] = ...
182
+ _use_args_: ClassVar[bool] = ...
183
+ _value2member_map_: ClassVar[dict] = ...
184
+ def __format__(self, *args, **kwargs) -> str:
185
+ """Convert to a string according to format_spec."""
186
+
187
+ class CommondxExecution(enum.IntEnum):
188
+ """
189
+ The set of execution modes supported by the library.
190
+
191
+ See `commondxExecution`.
192
+ """
193
+ __new__: ClassVar[Callable] = ...
194
+ BLOCK: ClassVar[CommondxExecution] = ...
195
+ THREAD: ClassVar[CommondxExecution] = ...
196
+ _generate_next_value_: ClassVar[Callable] = ...
197
+ _hashable_values_: ClassVar[list] = ...
198
+ _member_map_: ClassVar[dict] = ...
199
+ _member_names_: ClassVar[list] = ...
200
+ _member_type_: ClassVar[type[int]] = ...
201
+ _unhashable_values_: ClassVar[list] = ...
202
+ _unhashable_values_map_: ClassVar[dict] = ...
203
+ _use_args_: ClassVar[bool] = ...
204
+ _value2member_map_: ClassVar[dict] = ...
205
+ def __format__(self, *args, **kwargs) -> str:
206
+ """Convert to a string according to format_spec."""
207
+
208
+ class CommondxOption(enum.IntEnum):
209
+ """
210
+ Options to tweak code generation.
211
+
212
+ See `commondxOption`.
213
+ """
214
+ __new__: ClassVar[Callable] = ...
215
+ CODE_CONTAINER: ClassVar[CommondxOption] = ...
216
+ CODE_ISA: ClassVar[CommondxOption] = ...
217
+ EXTRA_NVRTC_ARGS: ClassVar[CommondxOption] = ...
218
+ SYMBOL_NAME: ClassVar[CommondxOption] = ...
219
+ TARGET_CODE: ClassVar[CommondxOption] = ...
220
+ TARGET_SM: ClassVar[CommondxOption] = ...
221
+ _generate_next_value_: ClassVar[Callable] = ...
222
+ _hashable_values_: ClassVar[list] = ...
223
+ _member_map_: ClassVar[dict] = ...
224
+ _member_names_: ClassVar[list] = ...
225
+ _member_type_: ClassVar[type[int]] = ...
226
+ _unhashable_values_: ClassVar[list] = ...
227
+ _unhashable_values_map_: ClassVar[dict] = ...
228
+ _use_args_: ClassVar[bool] = ...
229
+ _value2member_map_: ClassVar[dict] = ...
230
+ def __format__(self, *args, **kwargs) -> str:
231
+ """Convert to a string according to format_spec."""
232
+
233
+ class CommondxPrecision(enum.IntEnum):
234
+ """
235
+ The set of precisions supported by the library.For the actual input and
236
+ output value types, see commondxValueType_t
237
+
238
+ See `commondxPrecision`.
239
+ """
240
+ __new__: ClassVar[Callable] = ...
241
+ BF16: ClassVar[CommondxPrecision] = ...
242
+ F16: ClassVar[CommondxPrecision] = ...
243
+ F32: ClassVar[CommondxPrecision] = ...
244
+ F64: ClassVar[CommondxPrecision] = ...
245
+ F8_E4M3: ClassVar[CommondxPrecision] = ...
246
+ F8_E5M2: ClassVar[CommondxPrecision] = ...
247
+ I16: ClassVar[CommondxPrecision] = ...
248
+ I32: ClassVar[CommondxPrecision] = ...
249
+ I64: ClassVar[CommondxPrecision] = ...
250
+ I8: ClassVar[CommondxPrecision] = ...
251
+ TF32: ClassVar[CommondxPrecision] = ...
252
+ UI16: ClassVar[CommondxPrecision] = ...
253
+ UI32: ClassVar[CommondxPrecision] = ...
254
+ UI64: ClassVar[CommondxPrecision] = ...
255
+ UI8: ClassVar[CommondxPrecision] = ...
256
+ _generate_next_value_: ClassVar[Callable] = ...
257
+ _hashable_values_: ClassVar[list] = ...
258
+ _member_map_: ClassVar[dict] = ...
259
+ _member_names_: ClassVar[list] = ...
260
+ _member_type_: ClassVar[type[int]] = ...
261
+ _unhashable_values_: ClassVar[list] = ...
262
+ _unhashable_values_map_: ClassVar[dict] = ...
263
+ _use_args_: ClassVar[bool] = ...
264
+ _value2member_map_: ClassVar[dict] = ...
265
+ def __format__(self, *args, **kwargs) -> str:
266
+ """Convert to a string according to format_spec."""
267
+
268
+ class CommondxStatusType(enum.IntEnum):
269
+ """
270
+ The set of status values that can be returned by APIs defined in the
271
+ library.
272
+
273
+ See `commondxStatusType`.
274
+ """
275
+ __new__: ClassVar[Callable] = ...
276
+ COMPILATION_ERROR: ClassVar[CommondxStatusType] = ...
277
+ CUFFT_ERROR: ClassVar[CommondxStatusType] = ...
278
+ INTERNAL_ERROR: ClassVar[CommondxStatusType] = ...
279
+ INVALID_VALUE: ClassVar[CommondxStatusType] = ...
280
+ SUCCESS: ClassVar[CommondxStatusType] = ...
281
+ _generate_next_value_: ClassVar[Callable] = ...
282
+ _hashable_values_: ClassVar[list] = ...
283
+ _member_map_: ClassVar[dict] = ...
284
+ _member_names_: ClassVar[list] = ...
285
+ _member_type_: ClassVar[type[int]] = ...
286
+ _unhashable_values_: ClassVar[list] = ...
287
+ _unhashable_values_map_: ClassVar[dict] = ...
288
+ _use_args_: ClassVar[bool] = ...
289
+ _value2member_map_: ClassVar[dict] = ...
290
+ def __format__(self, *args, **kwargs) -> str:
291
+ """Convert to a string according to format_spec."""
292
+
293
+ class CommondxValueType(enum.IntEnum):
294
+ """
295
+ The set of value types supported by the library.Note that, for complex
296
+ numbers, this combines real and imaginary part, and that all complex
297
+ numbers are overaligned on their size.
298
+
299
+ See `commondxValueType`.
300
+ """
301
+ __new__: ClassVar[Callable] = ...
302
+ C_16BF: ClassVar[CommondxValueType] = ...
303
+ C_16F: ClassVar[CommondxValueType] = ...
304
+ C_16F2: ClassVar[CommondxValueType] = ...
305
+ C_16I: ClassVar[CommondxValueType] = ...
306
+ C_16UI: ClassVar[CommondxValueType] = ...
307
+ C_32F: ClassVar[CommondxValueType] = ...
308
+ C_32I: ClassVar[CommondxValueType] = ...
309
+ C_32TF: ClassVar[CommondxValueType] = ...
310
+ C_32UI: ClassVar[CommondxValueType] = ...
311
+ C_64F: ClassVar[CommondxValueType] = ...
312
+ C_64I: ClassVar[CommondxValueType] = ...
313
+ C_64UI: ClassVar[CommondxValueType] = ...
314
+ C_8F_E4M3: ClassVar[CommondxValueType] = ...
315
+ C_8F_E5M2: ClassVar[CommondxValueType] = ...
316
+ C_8I: ClassVar[CommondxValueType] = ...
317
+ C_8UI: ClassVar[CommondxValueType] = ...
318
+ R_16BF: ClassVar[CommondxValueType] = ...
319
+ R_16F: ClassVar[CommondxValueType] = ...
320
+ R_16F2: ClassVar[CommondxValueType] = ...
321
+ R_16I: ClassVar[CommondxValueType] = ...
322
+ R_16UI: ClassVar[CommondxValueType] = ...
323
+ R_32F: ClassVar[CommondxValueType] = ...
324
+ R_32I: ClassVar[CommondxValueType] = ...
325
+ R_32TF: ClassVar[CommondxValueType] = ...
326
+ R_32UI: ClassVar[CommondxValueType] = ...
327
+ R_64F: ClassVar[CommondxValueType] = ...
328
+ R_64I: ClassVar[CommondxValueType] = ...
329
+ R_64UI: ClassVar[CommondxValueType] = ...
330
+ R_8F_E4M3: ClassVar[CommondxValueType] = ...
331
+ R_8F_E5M2: ClassVar[CommondxValueType] = ...
332
+ R_8I: ClassVar[CommondxValueType] = ...
333
+ R_8UI: ClassVar[CommondxValueType] = ...
334
+ _generate_next_value_: ClassVar[Callable] = ...
335
+ _hashable_values_: ClassVar[list] = ...
336
+ _member_map_: ClassVar[dict] = ...
337
+ _member_names_: ClassVar[list] = ...
338
+ _member_type_: ClassVar[type[int]] = ...
339
+ _unhashable_values_: ClassVar[list] = ...
340
+ _unhashable_values_map_: ClassVar[dict] = ...
341
+ _use_args_: ClassVar[bool] = ...
342
+ _value2member_map_: ClassVar[dict] = ...
343
+ def __format__(self, *args, **kwargs) -> str:
344
+ """Convert to a string according to format_spec."""
345
+
346
+ class CublasdxApi(enum.IntEnum):
347
+ """
348
+ Type of cublasDx API.Handling problems with default or custom/dynamic
349
+ leading dimensions. Check cublasdx::LeadingDimension operator
350
+ documentation for more details (https://docs.nvidia.com/cuda/cublasdx/a
351
+ pi/operators.html#leadingdimension-operator)
352
+
353
+ See `cublasdxApi`.
354
+ """
355
+ __new__: ClassVar[Callable] = ...
356
+ SMEM: ClassVar[CublasdxApi] = ...
357
+ SMEM_DYNAMIC_LD: ClassVar[CublasdxApi] = ...
358
+ TENSORS: ClassVar[CublasdxApi] = ...
359
+ _generate_next_value_: ClassVar[Callable] = ...
360
+ _hashable_values_: ClassVar[list] = ...
361
+ _member_map_: ClassVar[dict] = ...
362
+ _member_names_: ClassVar[list] = ...
363
+ _member_type_: ClassVar[type[int]] = ...
364
+ _unhashable_values_: ClassVar[list] = ...
365
+ _unhashable_values_map_: ClassVar[dict] = ...
366
+ _use_args_: ClassVar[bool] = ...
367
+ _value2member_map_: ClassVar[dict] = ...
368
+ def __format__(self, *args, **kwargs) -> str:
369
+ """Convert to a string according to format_spec."""
370
+
371
+ class CublasdxArrangement(enum.IntEnum):
372
+ """
373
+ Data arrangement mode.Defines data arrangements in tensors' taking part
374
+ in the calculation. Check cubladx::TransposeMode operator
375
+ documentation for more details
376
+ (https://docs.nvidia.com/cuda/cublasdx/api/operators.html#arrangement-
377
+ operator)
378
+
379
+ See `cublasdxArrangement`.
380
+ """
381
+ __new__: ClassVar[Callable] = ...
382
+ COL_MAJOR: ClassVar[CublasdxArrangement] = ...
383
+ ROW_MAJOR: ClassVar[CublasdxArrangement] = ...
384
+ _generate_next_value_: ClassVar[Callable] = ...
385
+ _hashable_values_: ClassVar[list] = ...
386
+ _member_map_: ClassVar[dict] = ...
387
+ _member_names_: ClassVar[list] = ...
388
+ _member_type_: ClassVar[type[int]] = ...
389
+ _unhashable_values_: ClassVar[list] = ...
390
+ _unhashable_values_map_: ClassVar[dict] = ...
391
+ _use_args_: ClassVar[bool] = ...
392
+ _value2member_map_: ClassVar[dict] = ...
393
+ def __format__(self, *args, **kwargs) -> str:
394
+ """Convert to a string according to format_spec."""
395
+
396
+ class CublasdxBlockSizeStrategy(enum.IntEnum):
397
+ """
398
+ Type of block size strategy used in device pipeline and tile pipeline
399
+ creation. This enum affects register usage in pipelining GEMM kernels
400
+ and should be considered for different architectures.
401
+
402
+ See `cublasdxBlockSizeStrategy`.
403
+ """
404
+ __new__: ClassVar[Callable] = ...
405
+ FIXED: ClassVar[CublasdxBlockSizeStrategy] = ...
406
+ HEURISTIC: ClassVar[CublasdxBlockSizeStrategy] = ...
407
+ _generate_next_value_: ClassVar[Callable] = ...
408
+ _hashable_values_: ClassVar[list] = ...
409
+ _member_map_: ClassVar[dict] = ...
410
+ _member_names_: ClassVar[list] = ...
411
+ _member_type_: ClassVar[type[int]] = ...
412
+ _unhashable_values_: ClassVar[list] = ...
413
+ _unhashable_values_map_: ClassVar[dict] = ...
414
+ _use_args_: ClassVar[bool] = ...
415
+ _value2member_map_: ClassVar[dict] = ...
416
+ def __format__(self, *args, **kwargs) -> str:
417
+ """Convert to a string according to format_spec."""
418
+
419
+ class CublasdxDeviceFunctionOption(enum.IntEnum):
420
+ """
421
+ Device function options.
422
+
423
+ See `cublasdxDeviceFunctionOption`.
424
+ """
425
+ __new__: ClassVar[Callable] = ...
426
+ CALLBACK: ClassVar[CublasdxDeviceFunctionOption] = ...
427
+ COPY_ALIGNMENT: ClassVar[CublasdxDeviceFunctionOption] = ...
428
+ NUM_THREADS: ClassVar[CublasdxDeviceFunctionOption] = ...
429
+ SYMBOL_NAME: ClassVar[CublasdxDeviceFunctionOption] = ...
430
+ _generate_next_value_: ClassVar[Callable] = ...
431
+ _hashable_values_: ClassVar[list] = ...
432
+ _member_map_: ClassVar[dict] = ...
433
+ _member_names_: ClassVar[list] = ...
434
+ _member_type_: ClassVar[type[int]] = ...
435
+ _unhashable_values_: ClassVar[list] = ...
436
+ _unhashable_values_map_: ClassVar[dict] = ...
437
+ _use_args_: ClassVar[bool] = ...
438
+ _value2member_map_: ClassVar[dict] = ...
439
+ def __format__(self, *args, **kwargs) -> str:
440
+ """Convert to a string according to format_spec."""
441
+
442
+ class CublasdxDeviceFunctionTrait(enum.IntEnum):
443
+ """
444
+ Device function traits, used to query informations.
445
+
446
+ See `cublasdxDeviceFunctionTrait`.
447
+ """
448
+ __new__: ClassVar[Callable] = ...
449
+ SYMBOL: ClassVar[CublasdxDeviceFunctionTrait] = ...
450
+ _generate_next_value_: ClassVar[Callable] = ...
451
+ _hashable_values_: ClassVar[list] = ...
452
+ _member_map_: ClassVar[dict] = ...
453
+ _member_names_: ClassVar[list] = ...
454
+ _member_type_: ClassVar[type[int]] = ...
455
+ _unhashable_values_: ClassVar[list] = ...
456
+ _unhashable_values_map_: ClassVar[dict] = ...
457
+ _use_args_: ClassVar[bool] = ...
458
+ _value2member_map_: ClassVar[dict] = ...
459
+ def __format__(self, *args, **kwargs) -> str:
460
+ """Convert to a string according to format_spec."""
461
+
462
+ class CublasdxDeviceFunctionType(enum.IntEnum):
463
+ """
464
+ Device functions supported by the library.
465
+
466
+ See `cublasdxDeviceFunctionType`.
467
+ """
468
+ __new__: ClassVar[Callable] = ...
469
+ AXPBY: ClassVar[CublasdxDeviceFunctionType] = ...
470
+ CLEAR: ClassVar[CublasdxDeviceFunctionType] = ...
471
+ COPY: ClassVar[CublasdxDeviceFunctionType] = ...
472
+ COPY_WAIT: ClassVar[CublasdxDeviceFunctionType] = ...
473
+ CREATE: ClassVar[CublasdxDeviceFunctionType] = ...
474
+ DESTROY: ClassVar[CublasdxDeviceFunctionType] = ...
475
+ EPILOGUE: ClassVar[CublasdxDeviceFunctionType] = ...
476
+ EXECUTE: ClassVar[CublasdxDeviceFunctionType] = ...
477
+ IS_INDEX_IN_BOUNDS: ClassVar[CublasdxDeviceFunctionType] = ...
478
+ IS_PREDICATED: ClassVar[CublasdxDeviceFunctionType] = ...
479
+ IS_THREAD_ACTIVE: ClassVar[CublasdxDeviceFunctionType] = ...
480
+ MAP_CRD2IDX: ClassVar[CublasdxDeviceFunctionType] = ...
481
+ MAP_IDX2CRD: ClassVar[CublasdxDeviceFunctionType] = ...
482
+ MAP_IDX2CRD_PARTITIONER: ClassVar[CublasdxDeviceFunctionType] = ...
483
+ RESET: ClassVar[CublasdxDeviceFunctionType] = ...
484
+ _generate_next_value_: ClassVar[Callable] = ...
485
+ _hashable_values_: ClassVar[list] = ...
486
+ _member_map_: ClassVar[dict] = ...
487
+ _member_names_: ClassVar[list] = ...
488
+ _member_type_: ClassVar[type[int]] = ...
489
+ _unhashable_values_: ClassVar[list] = ...
490
+ _unhashable_values_map_: ClassVar[dict] = ...
491
+ _use_args_: ClassVar[bool] = ...
492
+ _value2member_map_: ClassVar[dict] = ...
493
+ def __format__(self, *args, **kwargs) -> str:
494
+ """Convert to a string according to format_spec."""
495
+
496
+ class CublasdxDevicePipelineType(enum.IntEnum):
497
+ """
498
+ Type of device pipeline.
499
+
500
+ See `cublasdxDevicePipelineType`.
501
+ """
502
+ __new__: ClassVar[Callable] = ...
503
+ SUGGESTED: ClassVar[CublasdxDevicePipelineType] = ...
504
+ _generate_next_value_: ClassVar[Callable] = ...
505
+ _hashable_values_: ClassVar[list] = ...
506
+ _member_map_: ClassVar[dict] = ...
507
+ _member_names_: ClassVar[list] = ...
508
+ _member_type_: ClassVar[type[int]] = ...
509
+ _unhashable_values_: ClassVar[list] = ...
510
+ _unhashable_values_map_: ClassVar[dict] = ...
511
+ _use_args_: ClassVar[bool] = ...
512
+ _value2member_map_: ClassVar[dict] = ...
513
+ def __format__(self, *args, **kwargs) -> str:
514
+ """Convert to a string according to format_spec."""
515
+
516
+ class CublasdxFunction(enum.IntEnum):
517
+ """
518
+ BLAS function.Sets the BLAS function that will be executed. Check
519
+ cubladx::Function operator documentation for more details
520
+ (https://docs.nvidia.com/cuda/cublasdx/api/operators.html#function-
521
+ operator)
522
+
523
+ See `cublasdxFunction`.
524
+ """
525
+ __new__: ClassVar[Callable] = ...
526
+ MM: ClassVar[CublasdxFunction] = ...
527
+ _generate_next_value_: ClassVar[Callable] = ...
528
+ _hashable_values_: ClassVar[list] = ...
529
+ _member_map_: ClassVar[dict] = ...
530
+ _member_names_: ClassVar[list] = ...
531
+ _member_type_: ClassVar[type[int]] = ...
532
+ _unhashable_values_: ClassVar[list] = ...
533
+ _unhashable_values_map_: ClassVar[dict] = ...
534
+ _use_args_: ClassVar[bool] = ...
535
+ _value2member_map_: ClassVar[dict] = ...
536
+ def __format__(self, *args, **kwargs) -> str:
537
+ """Convert to a string according to format_spec."""
538
+
539
+ class CublasdxMemorySpace(enum.IntEnum):
540
+ """
541
+ Memory space.
542
+
543
+ See `cublasdxMemorySpace`.
544
+ """
545
+ __new__: ClassVar[Callable] = ...
546
+ ANY: ClassVar[CublasdxMemorySpace] = ...
547
+ GMEM: ClassVar[CublasdxMemorySpace] = ...
548
+ RMEM: ClassVar[CublasdxMemorySpace] = ...
549
+ SMEM: ClassVar[CublasdxMemorySpace] = ...
550
+ _generate_next_value_: ClassVar[Callable] = ...
551
+ _hashable_values_: ClassVar[list] = ...
552
+ _member_map_: ClassVar[dict] = ...
553
+ _member_names_: ClassVar[list] = ...
554
+ _member_type_: ClassVar[type[int]] = ...
555
+ _unhashable_values_: ClassVar[list] = ...
556
+ _unhashable_values_map_: ClassVar[dict] = ...
557
+ _use_args_: ClassVar[bool] = ...
558
+ _value2member_map_: ClassVar[dict] = ...
559
+ def __format__(self, *args, **kwargs) -> str:
560
+ """Convert to a string according to format_spec."""
561
+
562
+ class CublasdxOperatorType(enum.IntEnum):
563
+ """
564
+ cublasDx operatorsThe set of supported cublasDx operators. Check
565
+ cublaDx description operator documentation for more details
566
+ (https://docs.nvidia.com/cuda/cublasdx/api/operators.html#function-
567
+ operator) Check cublasDx execution operator documentation for more
568
+ details
569
+ (https://docs.nvidia.com/cuda/cublasdx/api/operators.html#execution-
570
+ operators)
571
+
572
+ See `cublasdxOperatorType`.
573
+ """
574
+ __new__: ClassVar[Callable] = ...
575
+ ALIGNMENT: ClassVar[CublasdxOperatorType] = ...
576
+ API: ClassVar[CublasdxOperatorType] = ...
577
+ ARRANGEMENT: ClassVar[CublasdxOperatorType] = ...
578
+ BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
579
+ ENABLE_INPUT_STREAMING: ClassVar[CublasdxOperatorType] = ...
580
+ EXECUTION: ClassVar[CublasdxOperatorType] = ...
581
+ FUNCTION: ClassVar[CublasdxOperatorType] = ...
582
+ LEADING_DIMENSION: ClassVar[CublasdxOperatorType] = ...
583
+ PRECISION: ClassVar[CublasdxOperatorType] = ...
584
+ SIZE: ClassVar[CublasdxOperatorType] = ...
585
+ SM: ClassVar[CublasdxOperatorType] = ...
586
+ STATIC_BLOCK_DIM: ClassVar[CublasdxOperatorType] = ...
587
+ TRANSPOSE_MODE: ClassVar[CublasdxOperatorType] = ...
588
+ TYPE: ClassVar[CublasdxOperatorType] = ...
589
+ WITH_PIPELINE: ClassVar[CublasdxOperatorType] = ...
590
+ _generate_next_value_: ClassVar[Callable] = ...
591
+ _hashable_values_: ClassVar[list] = ...
592
+ _member_map_: ClassVar[dict] = ...
593
+ _member_names_: ClassVar[list] = ...
594
+ _member_type_: ClassVar[type[int]] = ...
595
+ _unhashable_values_: ClassVar[list] = ...
596
+ _unhashable_values_map_: ClassVar[dict] = ...
597
+ _use_args_: ClassVar[bool] = ...
598
+ _value2member_map_: ClassVar[dict] = ...
599
+ def __format__(self, *args, **kwargs) -> str:
600
+ """Convert to a string according to format_spec."""
601
+
602
+ class CublasdxPipelineTrait(enum.IntEnum):
603
+ """
604
+ Pipeline traits, used to query informations.
605
+
606
+ See `cublasdxPipelineTrait`.
607
+ """
608
+ __new__: ClassVar[Callable] = ...
609
+ BLOCK_DIM: ClassVar[CublasdxPipelineTrait] = ...
610
+ BUFFER_ALIGNMENT_BYTES: ClassVar[CublasdxPipelineTrait] = ...
611
+ BUFFER_SIZE: ClassVar[CublasdxPipelineTrait] = ...
612
+ OPAQUE_NAME: ClassVar[CublasdxPipelineTrait] = ...
613
+ STORAGE_ALIGNMENT_BYTES: ClassVar[CublasdxPipelineTrait] = ...
614
+ STORAGE_BYTES: ClassVar[CublasdxPipelineTrait] = ...
615
+ _generate_next_value_: ClassVar[Callable] = ...
616
+ _hashable_values_: ClassVar[list] = ...
617
+ _member_map_: ClassVar[dict] = ...
618
+ _member_names_: ClassVar[list] = ...
619
+ _member_type_: ClassVar[type[int]] = ...
620
+ _unhashable_values_: ClassVar[list] = ...
621
+ _unhashable_values_map_: ClassVar[dict] = ...
622
+ _use_args_: ClassVar[bool] = ...
623
+ _value2member_map_: ClassVar[dict] = ...
624
+ def __format__(self, *args, **kwargs) -> str:
625
+ """Convert to a string according to format_spec."""
626
+
627
+ class CublasdxTensorOption(enum.IntEnum):
628
+ """
629
+ Tensor options.
630
+
631
+ See `cublasdxTensorOption`.
632
+ """
633
+ __new__: ClassVar[Callable] = ...
634
+ ALIGNMENT_BYTES: ClassVar[CublasdxTensorOption] = ...
635
+ _generate_next_value_: ClassVar[Callable] = ...
636
+ _hashable_values_: ClassVar[list] = ...
637
+ _member_map_: ClassVar[dict] = ...
638
+ _member_names_: ClassVar[list] = ...
639
+ _member_type_: ClassVar[type[int]] = ...
640
+ _unhashable_values_: ClassVar[list] = ...
641
+ _unhashable_values_map_: ClassVar[dict] = ...
642
+ _use_args_: ClassVar[bool] = ...
643
+ _value2member_map_: ClassVar[dict] = ...
644
+ def __format__(self, *args, **kwargs) -> str:
645
+ """Convert to a string according to format_spec."""
646
+
647
+ class CublasdxTensorTrait(enum.IntEnum):
648
+ """
649
+ Tensor traits, used to query informations.
650
+
651
+ See `cublasdxTensorTrait`.
652
+ """
653
+ __new__: ClassVar[Callable] = ...
654
+ ALIGNMENT_BYTES: ClassVar[CublasdxTensorTrait] = ...
655
+ LOGICAL_SIZE: ClassVar[CublasdxTensorTrait] = ...
656
+ MEMORY_SPACE: ClassVar[CublasdxTensorTrait] = ...
657
+ OPAQUE_NAME: ClassVar[CublasdxTensorTrait] = ...
658
+ STORAGE_BYTES: ClassVar[CublasdxTensorTrait] = ...
659
+ UID: ClassVar[CublasdxTensorTrait] = ...
660
+ _generate_next_value_: ClassVar[Callable] = ...
661
+ _hashable_values_: ClassVar[list] = ...
662
+ _member_map_: ClassVar[dict] = ...
663
+ _member_names_: ClassVar[list] = ...
664
+ _member_type_: ClassVar[type[int]] = ...
665
+ _unhashable_values_: ClassVar[list] = ...
666
+ _unhashable_values_map_: ClassVar[dict] = ...
667
+ _use_args_: ClassVar[bool] = ...
668
+ _value2member_map_: ClassVar[dict] = ...
669
+ def __format__(self, *args, **kwargs) -> str:
670
+ """Convert to a string according to format_spec."""
671
+
672
+ class CublasdxTensorType(enum.IntEnum):
673
+ """
674
+ cuBLASDx desired tensor typeTensor types are opaque (layout is
675
+ unspecified), non-owning, and defined by - Memory space (global,
676
+ shared or register memory) - Size & alignment (in bytes) Tensor's
677
+ representation in-memory and in-device depends on their memory space.
678
+ Shared & register tensors are defined as **View CUDA Toolkit
679
+ Documentation for a C++ code example** where `ptr` points to the
680
+ associated data. Global memory tensors have an associated runtime
681
+ leading dimension (64b signed integer), and their representation is
682
+ **View CUDA Toolkit Documentation for a C++ code example** where `ptr`
683
+ points to the associated data and `strides[0]` is the leading
684
+ dimension. In either case, `ptr` must point to some storage (with
685
+ appropriate size and alignment, see below) and is not owning. The user
686
+ is expected to keep memory allocated beyond any use of the tensor.
687
+ `strides[1]` should be a signed, 64bit integer (`long long`) equal to
688
+ the leading dimension of the global memory tensor. The leading
689
+ dimension is the number of `elements` between two successive rows or
690
+ columns (not bytes), depending on the context. All tensor APIs take
691
+ their argument by value (not by pointer) and expect the struct to be
692
+ passed as-is on the stack. Each opaque tensor type is uniquely
693
+ identified by a unique ID and name, see cublasdxTensorTrait_t .
694
+
695
+ See `cublasdxTensorType`.
696
+ """
697
+ __new__: ClassVar[Callable] = ...
698
+ ACCUMULATOR_C: ClassVar[CublasdxTensorType] = ...
699
+ GMEM_A: ClassVar[CublasdxTensorType] = ...
700
+ GMEM_B: ClassVar[CublasdxTensorType] = ...
701
+ GMEM_C: ClassVar[CublasdxTensorType] = ...
702
+ RMEM_C: ClassVar[CublasdxTensorType] = ...
703
+ SMEM_A: ClassVar[CublasdxTensorType] = ...
704
+ SMEM_B: ClassVar[CublasdxTensorType] = ...
705
+ SMEM_C: ClassVar[CublasdxTensorType] = ...
706
+ SUGGESTED_ACCUMULATOR_C: ClassVar[CublasdxTensorType] = ...
707
+ SUGGESTED_RMEM_C: ClassVar[CublasdxTensorType] = ...
708
+ SUGGESTED_SMEM_A: ClassVar[CublasdxTensorType] = ...
709
+ SUGGESTED_SMEM_B: ClassVar[CublasdxTensorType] = ...
710
+ SUGGESTED_SMEM_C: ClassVar[CublasdxTensorType] = ...
711
+ _generate_next_value_: ClassVar[Callable] = ...
712
+ _hashable_values_: ClassVar[list] = ...
713
+ _member_map_: ClassVar[dict] = ...
714
+ _member_names_: ClassVar[list] = ...
715
+ _member_type_: ClassVar[type[int]] = ...
716
+ _unhashable_values_: ClassVar[list] = ...
717
+ _unhashable_values_map_: ClassVar[dict] = ...
718
+ _use_args_: ClassVar[bool] = ...
719
+ _value2member_map_: ClassVar[dict] = ...
720
+ def __format__(self, *args, **kwargs) -> str:
721
+ """Convert to a string according to format_spec."""
722
+
723
+ class CublasdxTilePipelineType(enum.IntEnum):
724
+ """
725
+ Type of tile pipeline.
726
+
727
+ See `cublasdxTilePipelineType`.
728
+ """
729
+ __new__: ClassVar[Callable] = ...
730
+ PIPELINE_DEFAULT: ClassVar[CublasdxTilePipelineType] = ...
731
+ _generate_next_value_: ClassVar[Callable] = ...
732
+ _hashable_values_: ClassVar[list] = ...
733
+ _member_map_: ClassVar[dict] = ...
734
+ _member_names_: ClassVar[list] = ...
735
+ _member_type_: ClassVar[type[int]] = ...
736
+ _unhashable_values_: ClassVar[list] = ...
737
+ _unhashable_values_map_: ClassVar[dict] = ...
738
+ _use_args_: ClassVar[bool] = ...
739
+ _value2member_map_: ClassVar[dict] = ...
740
+ def __format__(self, *args, **kwargs) -> str:
741
+ """Convert to a string according to format_spec."""
742
+
743
+ class CublasdxTraitType(enum.IntEnum):
744
+ """
745
+ cublasDx traitsThe set of supported types of traits that can be
746
+ accessed from finalized sources that use cublasDx. Check cublasDx
747
+ Execution Block Traits documentation for more details
748
+ (https://docs.nvidia.com/cuda/cublasdx/api/traits.html#block-traits)
749
+
750
+ See `cublasdxTraitType`.
751
+ """
752
+ __new__: ClassVar[Callable] = ...
753
+ ALIGNMENT: ClassVar[CublasdxTraitType] = ...
754
+ ARRANGEMENT: ClassVar[CublasdxTraitType] = ...
755
+ BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
756
+ BLOCK_SIZE: ClassVar[CublasdxTraitType] = ...
757
+ LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
758
+ MAX_THREADS_PER_BLOCK: ClassVar[CublasdxTraitType] = ...
759
+ SIZE: ClassVar[CublasdxTraitType] = ...
760
+ SUGGESTED_BLOCK_DIM: ClassVar[CublasdxTraitType] = ...
761
+ SUGGESTED_LEADING_DIMENSION: ClassVar[CublasdxTraitType] = ...
762
+ SYMBOL_NAME: ClassVar[CublasdxTraitType] = ...
763
+ VALUE_TYPE: ClassVar[CublasdxTraitType] = ...
764
+ _generate_next_value_: ClassVar[Callable] = ...
765
+ _hashable_values_: ClassVar[list] = ...
766
+ _member_map_: ClassVar[dict] = ...
767
+ _member_names_: ClassVar[list] = ...
768
+ _member_type_: ClassVar[type[int]] = ...
769
+ _unhashable_values_: ClassVar[list] = ...
770
+ _unhashable_values_map_: ClassVar[dict] = ...
771
+ _use_args_: ClassVar[bool] = ...
772
+ _value2member_map_: ClassVar[dict] = ...
773
+ def __format__(self, *args, **kwargs) -> str:
774
+ """Convert to a string according to format_spec."""
775
+
776
+ class CublasdxTransposeMode(enum.IntEnum):
777
+ """
778
+ Tensor transpose mode.The transpose mode depends on
779
+ cubladx::TransposeMode operator which is deprecated since cublasDx
780
+ 0.2.0 and might be removed in future versions of mathDx libraries
781
+ Check cubladx::TransposeMode operator documentation for more details (h
782
+ ttps://docs.nvidia.com/cuda/cublasdx/api/operators.html#transposemode-
783
+ operator)
784
+
785
+ See `cublasdxTransposeMode`.
786
+ """
787
+ __new__: ClassVar[Callable] = ...
788
+ CONJ_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
789
+ NON_TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
790
+ TRANSPOSED: ClassVar[CublasdxTransposeMode] = ...
791
+ _generate_next_value_: ClassVar[Callable] = ...
792
+ _hashable_values_: ClassVar[list] = ...
793
+ _member_map_: ClassVar[dict] = ...
794
+ _member_names_: ClassVar[list] = ...
795
+ _member_type_: ClassVar[type[int]] = ...
796
+ _unhashable_values_: ClassVar[list] = ...
797
+ _unhashable_values_map_: ClassVar[dict] = ...
798
+ _use_args_: ClassVar[bool] = ...
799
+ _value2member_map_: ClassVar[dict] = ...
800
+ def __format__(self, *args, **kwargs) -> str:
801
+ """Convert to a string according to format_spec."""
802
+
803
+ class CublasdxType(enum.IntEnum):
804
+ """
805
+ Type of computation data.Check cubladx::Type operator documentation for
806
+ more details
807
+ (https://docs.nvidia.com/cuda/cublasdx/api/operators.html#type-
808
+ operator)
809
+
810
+ See `cublasdxType`.
811
+ """
812
+ __new__: ClassVar[Callable] = ...
813
+ COMPLEX: ClassVar[CublasdxType] = ...
814
+ REAL: ClassVar[CublasdxType] = ...
815
+ _generate_next_value_: ClassVar[Callable] = ...
816
+ _hashable_values_: ClassVar[list] = ...
817
+ _member_map_: ClassVar[dict] = ...
818
+ _member_names_: ClassVar[list] = ...
819
+ _member_type_: ClassVar[type[int]] = ...
820
+ _unhashable_values_: ClassVar[list] = ...
821
+ _unhashable_values_map_: ClassVar[dict] = ...
822
+ _use_args_: ClassVar[bool] = ...
823
+ _value2member_map_: ClassVar[dict] = ...
824
+ def __format__(self, *args, **kwargs) -> str:
825
+ """Convert to a string according to format_spec."""
826
+
827
+ class CufftdxApi(enum.IntEnum):
828
+ """
829
+ Type of cufftDx API.Handling problems with input in register or in
830
+ shared memory buffers. Check cufftdx::execute method documentation for
831
+ more details
832
+ (https://docs.nvidia.com/cuda/cufftdx/api/methods.html#block-execute-
833
+ method)
834
+
835
+ See `cufftdxApi`.
836
+ """
837
+ __new__: ClassVar[Callable] = ...
838
+ LMEM: ClassVar[CufftdxApi] = ...
839
+ SMEM: ClassVar[CufftdxApi] = ...
840
+ _generate_next_value_: ClassVar[Callable] = ...
841
+ _hashable_values_: ClassVar[list] = ...
842
+ _member_map_: ClassVar[dict] = ...
843
+ _member_names_: ClassVar[list] = ...
844
+ _member_type_: ClassVar[type[int]] = ...
845
+ _unhashable_values_: ClassVar[list] = ...
846
+ _unhashable_values_map_: ClassVar[dict] = ...
847
+ _use_args_: ClassVar[bool] = ...
848
+ _value2member_map_: ClassVar[dict] = ...
849
+ def __format__(self, *args, **kwargs) -> str:
850
+ """Convert to a string according to format_spec."""
851
+
852
+ class CufftdxCodeType(enum.IntEnum):
853
+ """
854
+ Code type.Check cufftdx code type documentation for more details
855
+ (https://docs.nvidia.com/cuda/1.4.0-ea/cufftdx/api/methods.html#code-
856
+ type)
857
+
858
+ See `cufftdxCodeType`.
859
+ """
860
+ __new__: ClassVar[Callable] = ...
861
+ LTOIR: ClassVar[CufftdxCodeType] = ...
862
+ PTX: ClassVar[CufftdxCodeType] = ...
863
+ _generate_next_value_: ClassVar[Callable] = ...
864
+ _hashable_values_: ClassVar[list] = ...
865
+ _member_map_: ClassVar[dict] = ...
866
+ _member_names_: ClassVar[list] = ...
867
+ _member_type_: ClassVar[type[int]] = ...
868
+ _unhashable_values_: ClassVar[list] = ...
869
+ _unhashable_values_map_: ClassVar[dict] = ...
870
+ _use_args_: ClassVar[bool] = ...
871
+ _value2member_map_: ClassVar[dict] = ...
872
+ def __format__(self, *args, **kwargs) -> str:
873
+ """Convert to a string according to format_spec."""
874
+
875
+ class CufftdxComplexLayout(enum.IntEnum):
876
+ """
877
+ Complex data layout.Check cufftdx layout documentation for more details
878
+ (https://docs.nvidia.com/cuda/cufftdx/api/methods.html#complex-element-
879
+ layouts)
880
+
881
+ See `cufftdxComplexLayout`.
882
+ """
883
+ __new__: ClassVar[Callable] = ...
884
+ FULL: ClassVar[CufftdxComplexLayout] = ...
885
+ NATURAL: ClassVar[CufftdxComplexLayout] = ...
886
+ PACKED: ClassVar[CufftdxComplexLayout] = ...
887
+ _generate_next_value_: ClassVar[Callable] = ...
888
+ _hashable_values_: ClassVar[list] = ...
889
+ _member_map_: ClassVar[dict] = ...
890
+ _member_names_: ClassVar[list] = ...
891
+ _member_type_: ClassVar[type[int]] = ...
892
+ _unhashable_values_: ClassVar[list] = ...
893
+ _unhashable_values_map_: ClassVar[dict] = ...
894
+ _use_args_: ClassVar[bool] = ...
895
+ _value2member_map_: ClassVar[dict] = ...
896
+ def __format__(self, *args, **kwargs) -> str:
897
+ """Convert to a string according to format_spec."""
898
+
899
+ class CufftdxDirection(enum.IntEnum):
900
+ """
901
+ FFT direction.Check cufftdx::Direction operator documentation for more
902
+ details
903
+ (https://docs.nvidia.com/cuda/cufftdx/api/operators.html#direction-
904
+ operator)
905
+
906
+ See `cufftdxDirection`.
907
+ """
908
+ __new__: ClassVar[Callable] = ...
909
+ FORWARD: ClassVar[CufftdxDirection] = ...
910
+ INVERSE: ClassVar[CufftdxDirection] = ...
911
+ _generate_next_value_: ClassVar[Callable] = ...
912
+ _hashable_values_: ClassVar[list] = ...
913
+ _member_map_: ClassVar[dict] = ...
914
+ _member_names_: ClassVar[list] = ...
915
+ _member_type_: ClassVar[type[int]] = ...
916
+ _unhashable_values_: ClassVar[list] = ...
917
+ _unhashable_values_map_: ClassVar[dict] = ...
918
+ _use_args_: ClassVar[bool] = ...
919
+ _value2member_map_: ClassVar[dict] = ...
920
+ def __format__(self, *args, **kwargs) -> str:
921
+ """Convert to a string according to format_spec."""
922
+
923
+ class CufftdxKnobType(enum.IntEnum):
924
+ """
925
+ cufftDx configuration knobsThe set of supported knobs used for
926
+ accessing cufftDx operator's performance configuration details. Check
927
+ cufftDx Execution Traits documentation for more details
928
+ (https://docs.nvidia.com/cuda/cufftdx/api/traits.html#execution-traits)
929
+
930
+ See `cufftdxKnobType`.
931
+ """
932
+ __new__: ClassVar[Callable] = ...
933
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxKnobType] = ...
934
+ FFTS_PER_BLOCK: ClassVar[CufftdxKnobType] = ...
935
+ _generate_next_value_: ClassVar[Callable] = ...
936
+ _hashable_values_: ClassVar[list] = ...
937
+ _member_map_: ClassVar[dict] = ...
938
+ _member_names_: ClassVar[list] = ...
939
+ _member_type_: ClassVar[type[int]] = ...
940
+ _unhashable_values_: ClassVar[list] = ...
941
+ _unhashable_values_map_: ClassVar[dict] = ...
942
+ _use_args_: ClassVar[bool] = ...
943
+ _value2member_map_: ClassVar[dict] = ...
944
+ def __format__(self, *args, **kwargs) -> str:
945
+ """Convert to a string according to format_spec."""
946
+
947
+ class CufftdxOperatorType(enum.IntEnum):
948
+ """
949
+ cufftDx operatorsThe set of supported cufftDx operators. Check cufftDx
950
+ description operators documentation for more details
951
+ (https://docs.nvidia.com/cuda/cufftdx/api/operators.html#description-
952
+ operators) Check cufftDx execution operators documentation for more
953
+ details
954
+ (https://docs.nvidia.com/cuda/cufftdx/api/operators.html#execution-
955
+ operators)
956
+
957
+ See `cufftdxOperatorType`.
958
+ """
959
+ __new__: ClassVar[Callable] = ...
960
+ API: ClassVar[CufftdxOperatorType] = ...
961
+ BLOCK_DIM: ClassVar[CufftdxOperatorType] = ...
962
+ CODE_TYPE: ClassVar[CufftdxOperatorType] = ...
963
+ DIRECTION: ClassVar[CufftdxOperatorType] = ...
964
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxOperatorType] = ...
965
+ EXECUTION: ClassVar[CufftdxOperatorType] = ...
966
+ FFTS_PER_BLOCK: ClassVar[CufftdxOperatorType] = ...
967
+ PRECISION: ClassVar[CufftdxOperatorType] = ...
968
+ REAL_FFT_OPTIONS: ClassVar[CufftdxOperatorType] = ...
969
+ SIZE: ClassVar[CufftdxOperatorType] = ...
970
+ SM: ClassVar[CufftdxOperatorType] = ...
971
+ TYPE: ClassVar[CufftdxOperatorType] = ...
972
+ _generate_next_value_: ClassVar[Callable] = ...
973
+ _hashable_values_: ClassVar[list] = ...
974
+ _member_map_: ClassVar[dict] = ...
975
+ _member_names_: ClassVar[list] = ...
976
+ _member_type_: ClassVar[type[int]] = ...
977
+ _unhashable_values_: ClassVar[list] = ...
978
+ _unhashable_values_map_: ClassVar[dict] = ...
979
+ _use_args_: ClassVar[bool] = ...
980
+ _value2member_map_: ClassVar[dict] = ...
981
+ def __format__(self, *args, **kwargs) -> str:
982
+ """Convert to a string according to format_spec."""
983
+
984
+ class CufftdxRealMode(enum.IntEnum):
985
+ """
986
+ Real data mode.Check cufftdx real data mode documentation for more
987
+ details (https://docs.nvidia.com/cuda/cufftdx/api/methods.html#real-
988
+ element-layouts)
989
+
990
+ See `cufftdxRealMode`.
991
+ """
992
+ __new__: ClassVar[Callable] = ...
993
+ FOLDED: ClassVar[CufftdxRealMode] = ...
994
+ NORMAL: ClassVar[CufftdxRealMode] = ...
995
+ _generate_next_value_: ClassVar[Callable] = ...
996
+ _hashable_values_: ClassVar[list] = ...
997
+ _member_map_: ClassVar[dict] = ...
998
+ _member_names_: ClassVar[list] = ...
999
+ _member_type_: ClassVar[type[int]] = ...
1000
+ _unhashable_values_: ClassVar[list] = ...
1001
+ _unhashable_values_map_: ClassVar[dict] = ...
1002
+ _use_args_: ClassVar[bool] = ...
1003
+ _value2member_map_: ClassVar[dict] = ...
1004
+ def __format__(self, *args, **kwargs) -> str:
1005
+ """Convert to a string according to format_spec."""
1006
+
1007
+ class CufftdxTraitType(enum.IntEnum):
1008
+ """
1009
+ cufftDx traitsThe set of supported types of traits that can be accessed
1010
+ from finalized sources that use cufftDx. Check cufftDx Execution
1011
+ Thread Traits documentation for more details
1012
+ (https://docs.nvidia.com/cuda/cufftdx/api/traits.html#thread-traits)
1013
+ Check cufftDx Execution Block Traits documentation for more details
1014
+ (https://docs.nvidia.com/cuda/cufftdx/api/traits.html#block-traits)
1015
+
1016
+ See `cufftdxTraitType`.
1017
+ """
1018
+ __new__: ClassVar[Callable] = ...
1019
+ BLOCK_DIM: ClassVar[CufftdxTraitType] = ...
1020
+ ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
1021
+ FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
1022
+ IMPLICIT_TYPE_BATCHING: ClassVar[CufftdxTraitType] = ...
1023
+ INPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
1024
+ INPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
1025
+ INPUT_TYPE: ClassVar[CufftdxTraitType] = ...
1026
+ OUTPUT_ELEMENTS_PER_THREAD: ClassVar[CufftdxTraitType] = ...
1027
+ OUTPUT_LENGTH: ClassVar[CufftdxTraitType] = ...
1028
+ OUTPUT_TYPE: ClassVar[CufftdxTraitType] = ...
1029
+ SHARED_MEMORY_SIZE: ClassVar[CufftdxTraitType] = ...
1030
+ STORAGE_SIZE: ClassVar[CufftdxTraitType] = ...
1031
+ STRIDE: ClassVar[CufftdxTraitType] = ...
1032
+ SUGGESTED_FFTS_PER_BLOCK: ClassVar[CufftdxTraitType] = ...
1033
+ SYMBOL_NAME: ClassVar[CufftdxTraitType] = ...
1034
+ VALUE_TYPE: ClassVar[CufftdxTraitType] = ...
1035
+ _generate_next_value_: ClassVar[Callable] = ...
1036
+ _hashable_values_: ClassVar[list] = ...
1037
+ _member_map_: ClassVar[dict] = ...
1038
+ _member_names_: ClassVar[list] = ...
1039
+ _member_type_: ClassVar[type[int]] = ...
1040
+ _unhashable_values_: ClassVar[list] = ...
1041
+ _unhashable_values_map_: ClassVar[dict] = ...
1042
+ _use_args_: ClassVar[bool] = ...
1043
+ _value2member_map_: ClassVar[dict] = ...
1044
+ def __format__(self, *args, **kwargs) -> str:
1045
+ """Convert to a string according to format_spec."""
1046
+
1047
+ class CufftdxType(enum.IntEnum):
1048
+ """
1049
+ Type of computation data.Check cufftdx::Type operator documentation for
1050
+ more details
1051
+ (https://docs.nvidia.com/cuda/cufftdx/api/methods.html#block-execute-
1052
+ method)
1053
+
1054
+ See `cufftdxType`.
1055
+ """
1056
+ __new__: ClassVar[Callable] = ...
1057
+ C2C: ClassVar[CufftdxType] = ...
1058
+ C2R: ClassVar[CufftdxType] = ...
1059
+ R2C: ClassVar[CufftdxType] = ...
1060
+ _generate_next_value_: ClassVar[Callable] = ...
1061
+ _hashable_values_: ClassVar[list] = ...
1062
+ _member_map_: ClassVar[dict] = ...
1063
+ _member_names_: ClassVar[list] = ...
1064
+ _member_type_: ClassVar[type[int]] = ...
1065
+ _unhashable_values_: ClassVar[list] = ...
1066
+ _unhashable_values_map_: ClassVar[dict] = ...
1067
+ _use_args_: ClassVar[bool] = ...
1068
+ _value2member_map_: ClassVar[dict] = ...
1069
+ def __format__(self, *args, **kwargs) -> str:
1070
+ """Convert to a string according to format_spec."""
1071
+
1072
+ class CuranddxDeviceFunctionType(enum.IntEnum):
1073
+ """
1074
+ Device functions.The set of supported device functions. Must be passed
1075
+ using CURANDDX_OPERATOR_DEVICE_FUNCTIONS and can be or'ed to indicate
1076
+ that libmathdx should generate multiple device functions.
1077
+
1078
+ See `curanddxDeviceFunctionType`.
1079
+ """
1080
+ __new__: ClassVar[Callable] = ...
1081
+ DESTROY_STATE: ClassVar[CuranddxDeviceFunctionType] = ...
1082
+ GENERATE: ClassVar[CuranddxDeviceFunctionType] = ...
1083
+ INIT_STATE: ClassVar[CuranddxDeviceFunctionType] = ...
1084
+ SKIP_OFFSET: ClassVar[CuranddxDeviceFunctionType] = ...
1085
+ SKIP_SEQUENCE: ClassVar[CuranddxDeviceFunctionType] = ...
1086
+ SKIP_SUBSEQUENCE: ClassVar[CuranddxDeviceFunctionType] = ...
1087
+ _generate_next_value_: ClassVar[Callable] = ...
1088
+ _hashable_values_: ClassVar[list] = ...
1089
+ _member_map_: ClassVar[dict] = ...
1090
+ _member_names_: ClassVar[list] = ...
1091
+ _member_type_: ClassVar[type[int]] = ...
1092
+ _unhashable_values_: ClassVar[list] = ...
1093
+ _unhashable_values_map_: ClassVar[dict] = ...
1094
+ _use_args_: ClassVar[bool] = ...
1095
+ _value2member_map_: ClassVar[dict] = ...
1096
+ def __format__(self, *args, **kwargs) -> str:
1097
+ """Convert to a string according to format_spec."""
1098
+
1099
+ class CuranddxDistribution(enum.IntEnum):
1100
+ """
1101
+ Distribution type.Indicate the random number generation to use. Can be
1102
+ set using CURANDDX_OPERATOR_DISTRIBUTION and further customized using
1103
+ CURANDDX_OPERATOR_NORMAL_METHOD. See cuRANDDx distributions in the
1104
+ documentation:
1105
+ https://docs.nvidia.com/cuda/curanddx/api/methods.html#random-number-
1106
+ generation-with-distributions.
1107
+
1108
+ See `curanddxDistribution`.
1109
+ """
1110
+ __new__: ClassVar[Callable] = ...
1111
+ LOG_NORMAL: ClassVar[CuranddxDistribution] = ...
1112
+ NORMAL: ClassVar[CuranddxDistribution] = ...
1113
+ POISSON: ClassVar[CuranddxDistribution] = ...
1114
+ UNIFORM: ClassVar[CuranddxDistribution] = ...
1115
+ UNIFORM_BITS: ClassVar[CuranddxDistribution] = ...
1116
+ _generate_next_value_: ClassVar[Callable] = ...
1117
+ _hashable_values_: ClassVar[list] = ...
1118
+ _member_map_: ClassVar[dict] = ...
1119
+ _member_names_: ClassVar[list] = ...
1120
+ _member_type_: ClassVar[type[int]] = ...
1121
+ _unhashable_values_: ClassVar[list] = ...
1122
+ _unhashable_values_map_: ClassVar[dict] = ...
1123
+ _use_args_: ClassVar[bool] = ...
1124
+ _value2member_map_: ClassVar[dict] = ...
1125
+ def __format__(self, *args, **kwargs) -> str:
1126
+ """Convert to a string according to format_spec."""
1127
+
1128
+ class CuranddxGenerateMethod(enum.IntEnum):
1129
+ """
1130
+ Generation method type.Specifies which generation method to use for
1131
+ cuRANDDx distributions. Set using CURANDDX_OPERATOR_GENERATE_METHOD .
1132
+ See cuRANDDx execution methods in the documentation:
1133
+ https://docs.nvidia.com/cuda/curanddx/api/methods.html
1134
+
1135
+ See `curanddxGenerateMethod`.
1136
+ """
1137
+ __new__: ClassVar[Callable] = ...
1138
+ PAIR: ClassVar[CuranddxGenerateMethod] = ...
1139
+ QUAD: ClassVar[CuranddxGenerateMethod] = ...
1140
+ SINGLE: ClassVar[CuranddxGenerateMethod] = ...
1141
+ _generate_next_value_: ClassVar[Callable] = ...
1142
+ _hashable_values_: ClassVar[list] = ...
1143
+ _member_map_: ClassVar[dict] = ...
1144
+ _member_names_: ClassVar[list] = ...
1145
+ _member_type_: ClassVar[type[int]] = ...
1146
+ _unhashable_values_: ClassVar[list] = ...
1147
+ _unhashable_values_map_: ClassVar[dict] = ...
1148
+ _use_args_: ClassVar[bool] = ...
1149
+ _value2member_map_: ClassVar[dict] = ...
1150
+ def __format__(self, *args, **kwargs) -> str:
1151
+ """Convert to a string according to format_spec."""
1152
+
1153
+ class CuranddxGenerator(enum.IntEnum):
1154
+ """
1155
+ Generator type.Matches cuRANDDx generator kinds. Set using
1156
+ CURANDDX_OPERATOR_GENERATOR . See cuRANDDx generators in the
1157
+ documentation: https://docs.nvidia.com/cuda/curanddx/api/description_op
1158
+ s.html#generator-operator-label
1159
+
1160
+ See `curanddxGenerator`.
1161
+ """
1162
+ __new__: ClassVar[Callable] = ...
1163
+ MRG32K3A: ClassVar[CuranddxGenerator] = ...
1164
+ PCG: ClassVar[CuranddxGenerator] = ...
1165
+ PHILOX4_32: ClassVar[CuranddxGenerator] = ...
1166
+ SCRAMBLED_SOBOL32: ClassVar[CuranddxGenerator] = ...
1167
+ SCRAMBLED_SOBOL64: ClassVar[CuranddxGenerator] = ...
1168
+ SOBOL32: ClassVar[CuranddxGenerator] = ...
1169
+ SOBOL64: ClassVar[CuranddxGenerator] = ...
1170
+ XORWOW: ClassVar[CuranddxGenerator] = ...
1171
+ _generate_next_value_: ClassVar[Callable] = ...
1172
+ _hashable_values_: ClassVar[list] = ...
1173
+ _member_map_: ClassVar[dict] = ...
1174
+ _member_names_: ClassVar[list] = ...
1175
+ _member_type_: ClassVar[type[int]] = ...
1176
+ _unhashable_values_: ClassVar[list] = ...
1177
+ _unhashable_values_map_: ClassVar[dict] = ...
1178
+ _use_args_: ClassVar[bool] = ...
1179
+ _value2member_map_: ClassVar[dict] = ...
1180
+ def __format__(self, *args, **kwargs) -> str:
1181
+ """Convert to a string according to format_spec."""
1182
+
1183
+ class CuranddxNormalMethod(enum.IntEnum):
1184
+ """
1185
+ Normal distribution method type.Specifies the method to use for normal
1186
+ and log-normal distributions. Set using CURANDDX_OPERATOR_NORMAL_METHOD
1187
+ . See cuRANDDx normal distribution documentation:
1188
+ https://docs.nvidia.com/cuda/curanddx/api/methods.html#normal-
1189
+ distribution
1190
+
1191
+ See `curanddxNormalMethod`.
1192
+ """
1193
+ __new__: ClassVar[Callable] = ...
1194
+ BOX_MULLER: ClassVar[CuranddxNormalMethod] = ...
1195
+ ICDF: ClassVar[CuranddxNormalMethod] = ...
1196
+ _generate_next_value_: ClassVar[Callable] = ...
1197
+ _hashable_values_: ClassVar[list] = ...
1198
+ _member_map_: ClassVar[dict] = ...
1199
+ _member_names_: ClassVar[list] = ...
1200
+ _member_type_: ClassVar[type[int]] = ...
1201
+ _unhashable_values_: ClassVar[list] = ...
1202
+ _unhashable_values_map_: ClassVar[dict] = ...
1203
+ _use_args_: ClassVar[bool] = ...
1204
+ _value2member_map_: ClassVar[dict] = ...
1205
+ def __format__(self, *args, **kwargs) -> str:
1206
+ """Convert to a string according to format_spec."""
1207
+
1208
+ class CuranddxOperatorType(enum.IntEnum):
1209
+ """
1210
+ Operators.The set of supported curandDx operators.
1211
+
1212
+ See `curanddxOperatorType`.
1213
+ """
1214
+ __new__: ClassVar[Callable] = ...
1215
+ DEVICE_FUNCTIONS: ClassVar[CuranddxOperatorType] = ...
1216
+ DISTRIBUTION: ClassVar[CuranddxOperatorType] = ...
1217
+ DISTRIBUTION_PARAMETERS: ClassVar[CuranddxOperatorType] = ...
1218
+ EXECUTION: ClassVar[CuranddxOperatorType] = ...
1219
+ GENERATE_METHOD: ClassVar[CuranddxOperatorType] = ...
1220
+ GENERATOR: ClassVar[CuranddxOperatorType] = ...
1221
+ NORMAL_METHOD: ClassVar[CuranddxOperatorType] = ...
1222
+ OUTPUT_TYPE: ClassVar[CuranddxOperatorType] = ...
1223
+ PHILOX_ROUNDS: ClassVar[CuranddxOperatorType] = ...
1224
+ SM: ClassVar[CuranddxOperatorType] = ...
1225
+ _generate_next_value_: ClassVar[Callable] = ...
1226
+ _hashable_values_: ClassVar[list] = ...
1227
+ _member_map_: ClassVar[dict] = ...
1228
+ _member_names_: ClassVar[list] = ...
1229
+ _member_type_: ClassVar[type[int]] = ...
1230
+ _unhashable_values_: ClassVar[list] = ...
1231
+ _unhashable_values_map_: ClassVar[dict] = ...
1232
+ _use_args_: ClassVar[bool] = ...
1233
+ _value2member_map_: ClassVar[dict] = ...
1234
+ def __format__(self, *args, **kwargs) -> str:
1235
+ """Convert to a string according to format_spec."""
1236
+
1237
+ class CuranddxTraitType(enum.IntEnum):
1238
+ """
1239
+ Traits.The set of supported types of traits that can be accessed from
1240
+ finalized sources that use curanddx.
1241
+
1242
+ See `curanddxTraitType`.
1243
+ """
1244
+ __new__: ClassVar[Callable] = ...
1245
+ STATE_ALIGNMENT: ClassVar[CuranddxTraitType] = ...
1246
+ STATE_SIZE: ClassVar[CuranddxTraitType] = ...
1247
+ SYMBOL_DESTROY_STATE_NAME: ClassVar[CuranddxTraitType] = ...
1248
+ SYMBOL_GENERATE_NAME: ClassVar[CuranddxTraitType] = ...
1249
+ SYMBOL_INIT_STATE_NAME: ClassVar[CuranddxTraitType] = ...
1250
+ SYMBOL_SKIP_OFFSET_NAME: ClassVar[CuranddxTraitType] = ...
1251
+ SYMBOL_SKIP_SEQUENCE_NAME: ClassVar[CuranddxTraitType] = ...
1252
+ SYMBOL_SKIP_SUBSEQUENCE_NAME: ClassVar[CuranddxTraitType] = ...
1253
+ _generate_next_value_: ClassVar[Callable] = ...
1254
+ _hashable_values_: ClassVar[list] = ...
1255
+ _member_map_: ClassVar[dict] = ...
1256
+ _member_names_: ClassVar[list] = ...
1257
+ _member_type_: ClassVar[type[int]] = ...
1258
+ _unhashable_values_: ClassVar[list] = ...
1259
+ _unhashable_values_map_: ClassVar[dict] = ...
1260
+ _use_args_: ClassVar[bool] = ...
1261
+ _value2member_map_: ClassVar[dict] = ...
1262
+ def __format__(self, *args, **kwargs) -> str:
1263
+ """Convert to a string according to format_spec."""
1264
+
1265
+ class CusolverdxApi(enum.IntEnum):
1266
+ """
1267
+ Type of cusolverdx API.
1268
+
1269
+ See `cusolverdxApi`.
1270
+ """
1271
+ __new__: ClassVar[Callable] = ...
1272
+ SMEM: ClassVar[CusolverdxApi] = ...
1273
+ SMEM_DYNAMIC_LD: ClassVar[CusolverdxApi] = ...
1274
+ _generate_next_value_: ClassVar[Callable] = ...
1275
+ _hashable_values_: ClassVar[list] = ...
1276
+ _member_map_: ClassVar[dict] = ...
1277
+ _member_names_: ClassVar[list] = ...
1278
+ _member_type_: ClassVar[type[int]] = ...
1279
+ _unhashable_values_: ClassVar[list] = ...
1280
+ _unhashable_values_map_: ClassVar[dict] = ...
1281
+ _use_args_: ClassVar[bool] = ...
1282
+ _value2member_map_: ClassVar[dict] = ...
1283
+ def __format__(self, *args, **kwargs) -> str:
1284
+ """Convert to a string according to format_spec."""
1285
+
1286
+ class CusolverdxArrangement(enum.IntEnum):
1287
+ """
1288
+ Data arrangement mode.Defines data arrangements in tensors' taking part
1289
+ in the calculation.
1290
+
1291
+ See `cusolverdxArrangement`.
1292
+ """
1293
+ __new__: ClassVar[Callable] = ...
1294
+ COL_MAJOR: ClassVar[CusolverdxArrangement] = ...
1295
+ ROW_MAJOR: ClassVar[CusolverdxArrangement] = ...
1296
+ _generate_next_value_: ClassVar[Callable] = ...
1297
+ _hashable_values_: ClassVar[list] = ...
1298
+ _member_map_: ClassVar[dict] = ...
1299
+ _member_names_: ClassVar[list] = ...
1300
+ _member_type_: ClassVar[type[int]] = ...
1301
+ _unhashable_values_: ClassVar[list] = ...
1302
+ _unhashable_values_map_: ClassVar[dict] = ...
1303
+ _use_args_: ClassVar[bool] = ...
1304
+ _value2member_map_: ClassVar[dict] = ...
1305
+ def __format__(self, *args, **kwargs) -> str:
1306
+ """Convert to a string according to format_spec."""
1307
+
1308
+ class CusolverdxDiag(enum.IntEnum):
1309
+ """
1310
+ Diag operator.Indicates whether the matrix diagonal is unit (all ones)
1311
+ or non-unit.
1312
+
1313
+ See `cusolverdxDiag`.
1314
+ """
1315
+ __new__: ClassVar[Callable] = ...
1316
+ NON_UNIT: ClassVar[CusolverdxDiag] = ...
1317
+ UNIT: ClassVar[CusolverdxDiag] = ...
1318
+ _generate_next_value_: ClassVar[Callable] = ...
1319
+ _hashable_values_: ClassVar[list] = ...
1320
+ _member_map_: ClassVar[dict] = ...
1321
+ _member_names_: ClassVar[list] = ...
1322
+ _member_type_: ClassVar[type[int]] = ...
1323
+ _unhashable_values_: ClassVar[list] = ...
1324
+ _unhashable_values_map_: ClassVar[dict] = ...
1325
+ _use_args_: ClassVar[bool] = ...
1326
+ _value2member_map_: ClassVar[dict] = ...
1327
+ def __format__(self, *args, **kwargs) -> str:
1328
+ """Convert to a string according to format_spec."""
1329
+
1330
+ class CusolverdxFillMode(enum.IntEnum):
1331
+ """
1332
+ Tensor fill mode.For symmetric matrix the fill mode can be upper or
1333
+ lower triangular
1334
+
1335
+ See `cusolverdxFillMode`.
1336
+ """
1337
+ __new__: ClassVar[Callable] = ...
1338
+ LOWER: ClassVar[CusolverdxFillMode] = ...
1339
+ UPPER: ClassVar[CusolverdxFillMode] = ...
1340
+ _generate_next_value_: ClassVar[Callable] = ...
1341
+ _hashable_values_: ClassVar[list] = ...
1342
+ _member_map_: ClassVar[dict] = ...
1343
+ _member_names_: ClassVar[list] = ...
1344
+ _member_type_: ClassVar[type[int]] = ...
1345
+ _unhashable_values_: ClassVar[list] = ...
1346
+ _unhashable_values_map_: ClassVar[dict] = ...
1347
+ _use_args_: ClassVar[bool] = ...
1348
+ _value2member_map_: ClassVar[dict] = ...
1349
+ def __format__(self, *args, **kwargs) -> str:
1350
+ """Convert to a string according to format_spec."""
1351
+
1352
+ class CusolverdxFunction(enum.IntEnum):
1353
+ '''
1354
+ Type of device function.The function signatures depend on the selected
1355
+ cusolverdxApi_t: - For CUSOLVERDX_API_SMEM: Leading dimensions are
1356
+ fixed at compile time. - For CUSOLVERDX_API_SMEM_DYNAMIC_LD: Leading
1357
+ dimensions are passed as pointers to `unsigned` at runtime. Type
1358
+ definitions used in function signatures: - `value_type`: The value
1359
+ type determined by CUSOLVERDX_OPERATOR_TYPE and
1360
+ CUSOLVERDX_OPERATOR_PRECISION. For real types: `float` or `double`. For
1361
+ complex types: `cuFloatComplex` or `cuDoubleComplex`. - `status_type`:
1362
+ A signed 32b integer (`int`) used to return status information. All
1363
+ functions are `extern "C"` and the symbol name can be queried using
1364
+ CUSOLVERDX_TRAIT_SYMBOL_NAME.
1365
+
1366
+ See `cusolverdxFunction`.
1367
+ '''
1368
+ __new__: ClassVar[Callable] = ...
1369
+ GELQF: ClassVar[CusolverdxFunction] = ...
1370
+ GELS: ClassVar[CusolverdxFunction] = ...
1371
+ GEQRF: ClassVar[CusolverdxFunction] = ...
1372
+ GESV_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
1373
+ GESV_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
1374
+ GETRF_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
1375
+ GETRF_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
1376
+ GETRS_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
1377
+ GETRS_PARTIAL_PIVOT: ClassVar[CusolverdxFunction] = ...
1378
+ GTSV_NO_PIVOT: ClassVar[CusolverdxFunction] = ...
1379
+ HEEV: ClassVar[CusolverdxFunction] = ...
1380
+ HTEV: ClassVar[CusolverdxFunction] = ...
1381
+ POSV: ClassVar[CusolverdxFunction] = ...
1382
+ POTRF: ClassVar[CusolverdxFunction] = ...
1383
+ POTRS: ClassVar[CusolverdxFunction] = ...
1384
+ TRSM: ClassVar[CusolverdxFunction] = ...
1385
+ UNGLQ: ClassVar[CusolverdxFunction] = ...
1386
+ UNGQR: ClassVar[CusolverdxFunction] = ...
1387
+ UNMLQ: ClassVar[CusolverdxFunction] = ...
1388
+ UNMQR: ClassVar[CusolverdxFunction] = ...
1389
+ _generate_next_value_: ClassVar[Callable] = ...
1390
+ _hashable_values_: ClassVar[list] = ...
1391
+ _member_map_: ClassVar[dict] = ...
1392
+ _member_names_: ClassVar[list] = ...
1393
+ _member_type_: ClassVar[type[int]] = ...
1394
+ _unhashable_values_: ClassVar[list] = ...
1395
+ _unhashable_values_map_: ClassVar[dict] = ...
1396
+ _use_args_: ClassVar[bool] = ...
1397
+ _value2member_map_: ClassVar[dict] = ...
1398
+ def __format__(self, *args, **kwargs) -> str:
1399
+ """Convert to a string according to format_spec."""
1400
+
1401
+ class CusolverdxJob(enum.IntEnum):
1402
+ """
1403
+ Job operator.Specifies whether and how eigenvectors are computed and
1404
+ stored in eigenvalue operations.
1405
+
1406
+ See `cusolverdxJob`.
1407
+ """
1408
+ __new__: ClassVar[Callable] = ...
1409
+ ALL_VECTORS: ClassVar[CusolverdxJob] = ...
1410
+ MULTIPLY_VECTORS: ClassVar[CusolverdxJob] = ...
1411
+ NO_VECTORS: ClassVar[CusolverdxJob] = ...
1412
+ OVERWRITE_VECTORS: ClassVar[CusolverdxJob] = ...
1413
+ _generate_next_value_: ClassVar[Callable] = ...
1414
+ _hashable_values_: ClassVar[list] = ...
1415
+ _member_map_: ClassVar[dict] = ...
1416
+ _member_names_: ClassVar[list] = ...
1417
+ _member_type_: ClassVar[type[int]] = ...
1418
+ _unhashable_values_: ClassVar[list] = ...
1419
+ _unhashable_values_map_: ClassVar[dict] = ...
1420
+ _use_args_: ClassVar[bool] = ...
1421
+ _value2member_map_: ClassVar[dict] = ...
1422
+ def __format__(self, *args, **kwargs) -> str:
1423
+ """Convert to a string according to format_spec."""
1424
+
1425
+ class CusolverdxOperatorType(enum.IntEnum):
1426
+ """
1427
+ Operators.The set of supported cusolverDx operators.
1428
+
1429
+ See `cusolverdxOperatorType`.
1430
+ """
1431
+ __new__: ClassVar[Callable] = ...
1432
+ API: ClassVar[CusolverdxOperatorType] = ...
1433
+ ARRANGEMENT: ClassVar[CusolverdxOperatorType] = ...
1434
+ BATCHES_PER_BLOCK: ClassVar[CusolverdxOperatorType] = ...
1435
+ BLOCK_DIM: ClassVar[CusolverdxOperatorType] = ...
1436
+ DIAG: ClassVar[CusolverdxOperatorType] = ...
1437
+ EXECUTION: ClassVar[CusolverdxOperatorType] = ...
1438
+ FILL_MODE: ClassVar[CusolverdxOperatorType] = ...
1439
+ FUNCTION: ClassVar[CusolverdxOperatorType] = ...
1440
+ JOB: ClassVar[CusolverdxOperatorType] = ...
1441
+ LEADING_DIMENSION: ClassVar[CusolverdxOperatorType] = ...
1442
+ PRECISION: ClassVar[CusolverdxOperatorType] = ...
1443
+ SIDE: ClassVar[CusolverdxOperatorType] = ...
1444
+ SIZE: ClassVar[CusolverdxOperatorType] = ...
1445
+ SM: ClassVar[CusolverdxOperatorType] = ...
1446
+ TRANSPOSE_MODE: ClassVar[CusolverdxOperatorType] = ...
1447
+ TYPE: ClassVar[CusolverdxOperatorType] = ...
1448
+ _generate_next_value_: ClassVar[Callable] = ...
1449
+ _hashable_values_: ClassVar[list] = ...
1450
+ _member_map_: ClassVar[dict] = ...
1451
+ _member_names_: ClassVar[list] = ...
1452
+ _member_type_: ClassVar[type[int]] = ...
1453
+ _unhashable_values_: ClassVar[list] = ...
1454
+ _unhashable_values_map_: ClassVar[dict] = ...
1455
+ _use_args_: ClassVar[bool] = ...
1456
+ _value2member_map_: ClassVar[dict] = ...
1457
+ def __format__(self, *args, **kwargs) -> str:
1458
+ """Convert to a string according to format_spec."""
1459
+
1460
+ class CusolverdxSide(enum.IntEnum):
1461
+ """
1462
+ Side operator.Indicates which side an operator is applied from in
1463
+ operations such as matrix multiplication.
1464
+
1465
+ See `cusolverdxSide`.
1466
+ """
1467
+ __new__: ClassVar[Callable] = ...
1468
+ LEFT: ClassVar[CusolverdxSide] = ...
1469
+ RIGHT: ClassVar[CusolverdxSide] = ...
1470
+ _generate_next_value_: ClassVar[Callable] = ...
1471
+ _hashable_values_: ClassVar[list] = ...
1472
+ _member_map_: ClassVar[dict] = ...
1473
+ _member_names_: ClassVar[list] = ...
1474
+ _member_type_: ClassVar[type[int]] = ...
1475
+ _unhashable_values_: ClassVar[list] = ...
1476
+ _unhashable_values_map_: ClassVar[dict] = ...
1477
+ _use_args_: ClassVar[bool] = ...
1478
+ _value2member_map_: ClassVar[dict] = ...
1479
+ def __format__(self, *args, **kwargs) -> str:
1480
+ """Convert to a string according to format_spec."""
1481
+
1482
+ class CusolverdxTraitType(enum.IntEnum):
1483
+ """
1484
+ Traits.The set of supported types of traits that can be accessed from
1485
+ finalized sources that use cusolverdx.
1486
+
1487
+ See `cusolverdxTraitType`.
1488
+ """
1489
+ __new__: ClassVar[Callable] = ...
1490
+ ARRANGEMENT: ClassVar[CusolverdxTraitType] = ...
1491
+ BATCHES_PER_BLOCK: ClassVar[CusolverdxTraitType] = ...
1492
+ BLOCK_DIM: ClassVar[CusolverdxTraitType] = ...
1493
+ LEADING_DIMENSION: ClassVar[CusolverdxTraitType] = ...
1494
+ SHARED_MEMORY_SIZE: ClassVar[CusolverdxTraitType] = ...
1495
+ SUGGESTED_BATCHES_PER_BLOCK: ClassVar[CusolverdxTraitType] = ...
1496
+ SUGGESTED_BLOCK_DIM: ClassVar[CusolverdxTraitType] = ...
1497
+ SUGGESTED_LEADING_DIMENSION: ClassVar[CusolverdxTraitType] = ...
1498
+ SYMBOL_NAME: ClassVar[CusolverdxTraitType] = ...
1499
+ TYPE: ClassVar[CusolverdxTraitType] = ...
1500
+ VALUE_TYPE: ClassVar[CusolverdxTraitType] = ...
1501
+ WORKSPACE_SIZE: ClassVar[CusolverdxTraitType] = ...
1502
+ _generate_next_value_: ClassVar[Callable] = ...
1503
+ _hashable_values_: ClassVar[list] = ...
1504
+ _member_map_: ClassVar[dict] = ...
1505
+ _member_names_: ClassVar[list] = ...
1506
+ _member_type_: ClassVar[type[int]] = ...
1507
+ _unhashable_values_: ClassVar[list] = ...
1508
+ _unhashable_values_map_: ClassVar[dict] = ...
1509
+ _use_args_: ClassVar[bool] = ...
1510
+ _value2member_map_: ClassVar[dict] = ...
1511
+ def __format__(self, *args, **kwargs) -> str:
1512
+ """Convert to a string according to format_spec."""
1513
+
1514
+ class CusolverdxTransposeMode(enum.IntEnum):
1515
+ """
1516
+ Transpose mode.Indicates inputs or outputs must be considered
1517
+ transposed.
1518
+
1519
+ See `cusolverdxTransposeMode`.
1520
+ """
1521
+ __new__: ClassVar[Callable] = ...
1522
+ CONJ_TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
1523
+ NON_TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
1524
+ TRANSPOSED: ClassVar[CusolverdxTransposeMode] = ...
1525
+ _generate_next_value_: ClassVar[Callable] = ...
1526
+ _hashable_values_: ClassVar[list] = ...
1527
+ _member_map_: ClassVar[dict] = ...
1528
+ _member_names_: ClassVar[list] = ...
1529
+ _member_type_: ClassVar[type[int]] = ...
1530
+ _unhashable_values_: ClassVar[list] = ...
1531
+ _unhashable_values_map_: ClassVar[dict] = ...
1532
+ _use_args_: ClassVar[bool] = ...
1533
+ _value2member_map_: ClassVar[dict] = ...
1534
+ def __format__(self, *args, **kwargs) -> str:
1535
+ """Convert to a string according to format_spec."""
1536
+
1537
+ class CusolverdxType(enum.IntEnum):
1538
+ """
1539
+ Type of input values.
1540
+
1541
+ See `cusolverdxType`.
1542
+ """
1543
+ __new__: ClassVar[Callable] = ...
1544
+ COMPLEX: ClassVar[CusolverdxType] = ...
1545
+ REAL: ClassVar[CusolverdxType] = ...
1546
+ _generate_next_value_: ClassVar[Callable] = ...
1547
+ _hashable_values_: ClassVar[list] = ...
1548
+ _member_map_: ClassVar[dict] = ...
1549
+ _member_names_: ClassVar[list] = ...
1550
+ _member_type_: ClassVar[type[int]] = ...
1551
+ _unhashable_values_: ClassVar[list] = ...
1552
+ _unhashable_values_map_: ClassVar[dict] = ...
1553
+ _use_args_: ClassVar[bool] = ...
1554
+ _value2member_map_: ClassVar[dict] = ...
1555
+ def __format__(self, *args, **kwargs) -> str:
1556
+ """Convert to a string according to format_spec."""
1557
+
1558
+ class LibMathDxError(Exception):
1559
+ def __init__(self, status) -> Any:
1560
+ """LibMathDxError.__init__(self, status)"""
1561
+ def __reduce__(self) -> Any:
1562
+ """LibMathDxError.__reduce__(self)"""