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,320 @@
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
+ get_property: _cython_3_2_5.cython_function_or_method
13
+ get_version: _cython_3_2_5.cython_function_or_method
14
+
15
+ class AlgMode(enum.IntEnum):
16
+ """
17
+ See `cusolverAlgMode_t`.
18
+ """
19
+ __new__: ClassVar[Callable] = ...
20
+ ALG_0: ClassVar[AlgMode] = ...
21
+ ALG_1: ClassVar[AlgMode] = ...
22
+ ALG_2: ClassVar[AlgMode] = ...
23
+ _generate_next_value_: ClassVar[Callable] = ...
24
+ _hashable_values_: ClassVar[list] = ...
25
+ _member_map_: ClassVar[dict] = ...
26
+ _member_names_: ClassVar[list] = ...
27
+ _member_type_: ClassVar[type[int]] = ...
28
+ _unhashable_values_: ClassVar[list] = ...
29
+ _unhashable_values_map_: ClassVar[dict] = ...
30
+ _use_args_: ClassVar[bool] = ...
31
+ _value2member_map_: ClassVar[dict] = ...
32
+ def __format__(self, *args, **kwargs) -> str:
33
+ """Convert to a string according to format_spec."""
34
+
35
+ class DeterministicMode(enum.IntEnum):
36
+ """
37
+ See `cusolverDeterministicMode_t`.
38
+ """
39
+ __new__: ClassVar[Callable] = ...
40
+ ALLOW_NON_DETERMINISTIC_RESULTS: ClassVar[DeterministicMode] = ...
41
+ DETERMINISTIC_RESULTS: ClassVar[DeterministicMode] = ...
42
+ _generate_next_value_: ClassVar[Callable] = ...
43
+ _hashable_values_: ClassVar[list] = ...
44
+ _member_map_: ClassVar[dict] = ...
45
+ _member_names_: ClassVar[list] = ...
46
+ _member_type_: ClassVar[type[int]] = ...
47
+ _unhashable_values_: ClassVar[list] = ...
48
+ _unhashable_values_map_: ClassVar[dict] = ...
49
+ _use_args_: ClassVar[bool] = ...
50
+ _value2member_map_: ClassVar[dict] = ...
51
+ def __format__(self, *args, **kwargs) -> str:
52
+ """Convert to a string according to format_spec."""
53
+
54
+ class DirectMode(enum.IntEnum):
55
+ """
56
+ See `cusolverDirectMode_t`.
57
+ """
58
+ __new__: ClassVar[Callable] = ...
59
+ BACKWARD: ClassVar[DirectMode] = ...
60
+ FORWARD: ClassVar[DirectMode] = ...
61
+ _generate_next_value_: ClassVar[Callable] = ...
62
+ _hashable_values_: ClassVar[list] = ...
63
+ _member_map_: ClassVar[dict] = ...
64
+ _member_names_: ClassVar[list] = ...
65
+ _member_type_: ClassVar[type[int]] = ...
66
+ _unhashable_values_: ClassVar[list] = ...
67
+ _unhashable_values_map_: ClassVar[dict] = ...
68
+ _use_args_: ClassVar[bool] = ...
69
+ _value2member_map_: ClassVar[dict] = ...
70
+ def __format__(self, *args, **kwargs) -> str:
71
+ """Convert to a string according to format_spec."""
72
+
73
+ class EigComp(enum.IntEnum):
74
+ """
75
+ See `cusolverEigComp_t`.
76
+ """
77
+ __new__: ClassVar[Callable] = ...
78
+ I: ClassVar[EigComp] = ...
79
+ N: ClassVar[EigComp] = ...
80
+ V: ClassVar[EigComp] = ...
81
+ _generate_next_value_: ClassVar[Callable] = ...
82
+ _hashable_values_: ClassVar[list] = ...
83
+ _member_map_: ClassVar[dict] = ...
84
+ _member_names_: ClassVar[list] = ...
85
+ _member_type_: ClassVar[type[int]] = ...
86
+ _unhashable_values_: ClassVar[list] = ...
87
+ _unhashable_values_map_: ClassVar[dict] = ...
88
+ _use_args_: ClassVar[bool] = ...
89
+ _value2member_map_: ClassVar[dict] = ...
90
+ def __format__(self, *args, **kwargs) -> str:
91
+ """Convert to a string according to format_spec."""
92
+
93
+ class EigMode(enum.IntEnum):
94
+ """
95
+ See `cusolverEigMode_t`.
96
+ """
97
+ __new__: ClassVar[Callable] = ...
98
+ NOVECTOR: ClassVar[EigMode] = ...
99
+ VECTOR: ClassVar[EigMode] = ...
100
+ _generate_next_value_: ClassVar[Callable] = ...
101
+ _hashable_values_: ClassVar[list] = ...
102
+ _member_map_: ClassVar[dict] = ...
103
+ _member_names_: ClassVar[list] = ...
104
+ _member_type_: ClassVar[type[int]] = ...
105
+ _unhashable_values_: ClassVar[list] = ...
106
+ _unhashable_values_map_: ClassVar[dict] = ...
107
+ _use_args_: ClassVar[bool] = ...
108
+ _value2member_map_: ClassVar[dict] = ...
109
+ def __format__(self, *args, **kwargs) -> str:
110
+ """Convert to a string according to format_spec."""
111
+
112
+ class EigRange(enum.IntEnum):
113
+ """
114
+ See `cusolverEigRange_t`.
115
+ """
116
+ __new__: ClassVar[Callable] = ...
117
+ ALL: ClassVar[EigRange] = ...
118
+ I: ClassVar[EigRange] = ...
119
+ V: ClassVar[EigRange] = ...
120
+ _generate_next_value_: ClassVar[Callable] = ...
121
+ _hashable_values_: ClassVar[list] = ...
122
+ _member_map_: ClassVar[dict] = ...
123
+ _member_names_: ClassVar[list] = ...
124
+ _member_type_: ClassVar[type[int]] = ...
125
+ _unhashable_values_: ClassVar[list] = ...
126
+ _unhashable_values_map_: ClassVar[dict] = ...
127
+ _use_args_: ClassVar[bool] = ...
128
+ _value2member_map_: ClassVar[dict] = ...
129
+ def __format__(self, *args, **kwargs) -> str:
130
+ """Convert to a string according to format_spec."""
131
+
132
+ class EigType(enum.IntEnum):
133
+ """
134
+ See `cusolverEigType_t`.
135
+ """
136
+ __new__: ClassVar[Callable] = ...
137
+ TYPE_1: ClassVar[EigType] = ...
138
+ TYPE_2: ClassVar[EigType] = ...
139
+ TYPE_3: ClassVar[EigType] = ...
140
+ _generate_next_value_: ClassVar[Callable] = ...
141
+ _hashable_values_: ClassVar[list] = ...
142
+ _member_map_: ClassVar[dict] = ...
143
+ _member_names_: ClassVar[list] = ...
144
+ _member_type_: ClassVar[type[int]] = ...
145
+ _unhashable_values_: ClassVar[list] = ...
146
+ _unhashable_values_map_: ClassVar[dict] = ...
147
+ _use_args_: ClassVar[bool] = ...
148
+ _value2member_map_: ClassVar[dict] = ...
149
+ def __format__(self, *args, **kwargs) -> str:
150
+ """Convert to a string according to format_spec."""
151
+
152
+ class IRSRefinement(enum.IntEnum):
153
+ """
154
+ See `cusolverIRSRefinement_t`.
155
+ """
156
+ __new__: ClassVar[Callable] = ...
157
+ IRS_REFINE_CLASSICAL: ClassVar[IRSRefinement] = ...
158
+ IRS_REFINE_CLASSICAL_GMRES: ClassVar[IRSRefinement] = ...
159
+ IRS_REFINE_GMRES: ClassVar[IRSRefinement] = ...
160
+ IRS_REFINE_GMRES_GMRES: ClassVar[IRSRefinement] = ...
161
+ IRS_REFINE_GMRES_NOPCOND: ClassVar[IRSRefinement] = ...
162
+ IRS_REFINE_NONE: ClassVar[IRSRefinement] = ...
163
+ IRS_REFINE_NOT_SET: ClassVar[IRSRefinement] = ...
164
+ PREC_DD: ClassVar[IRSRefinement] = ...
165
+ PREC_SHT: ClassVar[IRSRefinement] = ...
166
+ PREC_SS: ClassVar[IRSRefinement] = ...
167
+ _generate_next_value_: ClassVar[Callable] = ...
168
+ _hashable_values_: ClassVar[list] = ...
169
+ _member_map_: ClassVar[dict] = ...
170
+ _member_names_: ClassVar[list] = ...
171
+ _member_type_: ClassVar[type[int]] = ...
172
+ _unhashable_values_: ClassVar[list] = ...
173
+ _unhashable_values_map_: ClassVar[dict] = ...
174
+ _use_args_: ClassVar[bool] = ...
175
+ _value2member_map_: ClassVar[dict] = ...
176
+ def __format__(self, *args, **kwargs) -> str:
177
+ """Convert to a string according to format_spec."""
178
+
179
+ class MathMode(enum.IntEnum):
180
+ """
181
+ See `cusolverMathMode_t`.
182
+ """
183
+ __new__: ClassVar[Callable] = ...
184
+ DEFAULT_MATH: ClassVar[MathMode] = ...
185
+ FP32_EMULATED_BF16X9_MATH: ClassVar[MathMode] = ...
186
+ FP32_FP64_EMULATED_MATH: ClassVar[MathMode] = ...
187
+ FP64_EMULATED_FIXEDPOINT_MATH: ClassVar[MathMode] = ...
188
+ _generate_next_value_: ClassVar[Callable] = ...
189
+ _hashable_values_: ClassVar[list] = ...
190
+ _member_map_: ClassVar[dict] = ...
191
+ _member_names_: ClassVar[list] = ...
192
+ _member_type_: ClassVar[type[int]] = ...
193
+ _unhashable_values_: ClassVar[list] = ...
194
+ _unhashable_values_map_: ClassVar[dict] = ...
195
+ _use_args_: ClassVar[bool] = ...
196
+ _value2member_map_: ClassVar[dict] = ...
197
+ def __format__(self, *args, **kwargs) -> str:
198
+ """Convert to a string according to format_spec."""
199
+
200
+ class Norm(enum.IntEnum):
201
+ """
202
+ See `cusolverNorm_t`.
203
+ """
204
+ __new__: ClassVar[Callable] = ...
205
+ FRO_NORM: ClassVar[Norm] = ...
206
+ INF_NORM: ClassVar[Norm] = ...
207
+ MAX_NORM: ClassVar[Norm] = ...
208
+ ONE_NORM: ClassVar[Norm] = ...
209
+ _generate_next_value_: ClassVar[Callable] = ...
210
+ _hashable_values_: ClassVar[list] = ...
211
+ _member_map_: ClassVar[dict] = ...
212
+ _member_names_: ClassVar[list] = ...
213
+ _member_type_: ClassVar[type[int]] = ...
214
+ _unhashable_values_: ClassVar[list] = ...
215
+ _unhashable_values_map_: ClassVar[dict] = ...
216
+ _use_args_: ClassVar[bool] = ...
217
+ _value2member_map_: ClassVar[dict] = ...
218
+ def __format__(self, *args, **kwargs) -> str:
219
+ """Convert to a string according to format_spec."""
220
+
221
+ class PrecType(enum.IntEnum):
222
+ """
223
+ See `cusolverPrecType_t`.
224
+ """
225
+ __new__: ClassVar[Callable] = ...
226
+ C_16BF: ClassVar[PrecType] = ...
227
+ C_16F: ClassVar[PrecType] = ...
228
+ C_32F: ClassVar[PrecType] = ...
229
+ C_64F: ClassVar[PrecType] = ...
230
+ C_8I: ClassVar[PrecType] = ...
231
+ C_8U: ClassVar[PrecType] = ...
232
+ C_AP: ClassVar[PrecType] = ...
233
+ C_TF32: ClassVar[PrecType] = ...
234
+ R_16BF: ClassVar[PrecType] = ...
235
+ R_16F: ClassVar[PrecType] = ...
236
+ R_32F: ClassVar[PrecType] = ...
237
+ R_64F: ClassVar[PrecType] = ...
238
+ R_8I: ClassVar[PrecType] = ...
239
+ R_8U: ClassVar[PrecType] = ...
240
+ R_AP: ClassVar[PrecType] = ...
241
+ R_TF32: ClassVar[PrecType] = ...
242
+ _generate_next_value_: ClassVar[Callable] = ...
243
+ _hashable_values_: ClassVar[list] = ...
244
+ _member_map_: ClassVar[dict] = ...
245
+ _member_names_: ClassVar[list] = ...
246
+ _member_type_: ClassVar[type[int]] = ...
247
+ _unhashable_values_: ClassVar[list] = ...
248
+ _unhashable_values_map_: ClassVar[dict] = ...
249
+ _use_args_: ClassVar[bool] = ...
250
+ _value2member_map_: ClassVar[dict] = ...
251
+ def __format__(self, *args, **kwargs) -> str:
252
+ """Convert to a string according to format_spec."""
253
+
254
+ class Status(enum.IntEnum):
255
+ """
256
+ See `cusolverStatus_t`.
257
+ """
258
+ __new__: ClassVar[Callable] = ...
259
+ ALLOC_FAILED: ClassVar[Status] = ...
260
+ ARCH_MISMATCH: ClassVar[Status] = ...
261
+ EXECUTION_FAILED: ClassVar[Status] = ...
262
+ INTERNAL_ERROR: ClassVar[Status] = ...
263
+ INVALID_LICENSE: ClassVar[Status] = ...
264
+ INVALID_VALUE: ClassVar[Status] = ...
265
+ INVALID_WORKSPACE: ClassVar[Status] = ...
266
+ IRS_INFOS_NOT_DESTROYED: ClassVar[Status] = ...
267
+ IRS_INFOS_NOT_INITIALIZED: ClassVar[Status] = ...
268
+ IRS_INTERNAL_ERROR: ClassVar[Status] = ...
269
+ IRS_MATRIX_SINGULAR: ClassVar[Status] = ...
270
+ IRS_NOT_SUPPORTED: ClassVar[Status] = ...
271
+ IRS_NRHS_NOT_SUPPORTED_FOR_REFINE_GMRES: ClassVar[Status] = ...
272
+ IRS_OUT_OF_RANGE: ClassVar[Status] = ...
273
+ IRS_PARAMS_INVALID: ClassVar[Status] = ...
274
+ IRS_PARAMS_INVALID_MAXITER: ClassVar[Status] = ...
275
+ IRS_PARAMS_INVALID_PREC: ClassVar[Status] = ...
276
+ IRS_PARAMS_INVALID_REFINE: ClassVar[Status] = ...
277
+ IRS_PARAMS_NOT_INITIALIZED: ClassVar[Status] = ...
278
+ MAPPING_ERROR: ClassVar[Status] = ...
279
+ MATRIX_TYPE_NOT_SUPPORTED: ClassVar[Status] = ...
280
+ NOT_INITIALIZED: ClassVar[Status] = ...
281
+ NOT_SUPPORTED: ClassVar[Status] = ...
282
+ SUCCESS: ClassVar[Status] = ...
283
+ ZERO_PIVOT: ClassVar[Status] = ...
284
+ _generate_next_value_: ClassVar[Callable] = ...
285
+ _hashable_values_: ClassVar[list] = ...
286
+ _member_map_: ClassVar[dict] = ...
287
+ _member_names_: ClassVar[list] = ...
288
+ _member_type_: ClassVar[type[int]] = ...
289
+ _unhashable_values_: ClassVar[list] = ...
290
+ _unhashable_values_map_: ClassVar[dict] = ...
291
+ _use_args_: ClassVar[bool] = ...
292
+ _value2member_map_: ClassVar[dict] = ...
293
+ def __format__(self, *args, **kwargs) -> str:
294
+ """Convert to a string according to format_spec."""
295
+
296
+ class StorevMode(enum.IntEnum):
297
+ """
298
+ See `cusolverStorevMode_t`.
299
+ """
300
+ __new__: ClassVar[Callable] = ...
301
+ COLUMNWISE: ClassVar[StorevMode] = ...
302
+ ROWWISE: ClassVar[StorevMode] = ...
303
+ _generate_next_value_: ClassVar[Callable] = ...
304
+ _hashable_values_: ClassVar[list] = ...
305
+ _member_map_: ClassVar[dict] = ...
306
+ _member_names_: ClassVar[list] = ...
307
+ _member_type_: ClassVar[type[int]] = ...
308
+ _unhashable_values_: ClassVar[list] = ...
309
+ _unhashable_values_map_: ClassVar[dict] = ...
310
+ _use_args_: ClassVar[bool] = ...
311
+ _value2member_map_: ClassVar[dict] = ...
312
+ def __format__(self, *args, **kwargs) -> str:
313
+ """Convert to a string according to format_spec."""
314
+
315
+ class cuSOLVERError(Exception):
316
+ """cuSOLVERError(status)"""
317
+ def __init__(self, status) -> Any:
318
+ """Initialize self. See help(type(self)) for accurate signature."""
319
+ def __reduce__(self) -> Any:
320
+ """cuSOLVERError.__reduce__(self)"""