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
nvmath/bindings/cusolverSp.pxd ADDED
@@ -0,0 +1,95 @@
1
+ # Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
2
+ #
3
+ # SPDX-License-Identifier: Apache-2.0
4
+ #
5
+ # This code was automatically generated across versions from 12.0.1 to 13.2.1, generator version 0.3.1.dev1471+g7122059e9. Do not modify it directly.
6
+
7
+ from libc.stdint cimport intptr_t
8
+
9
+ from .cycusolverSp cimport *
10
+
11
+
12
+ ###############################################################################
13
+ # Types
14
+ ###############################################################################
15
+
16
+ ctypedef cusolverSpHandle_t Handle
17
+ ctypedef csrqrInfo_t csrqrInfo
18
+
19
+ ctypedef cudaStream_t Stream
20
+ ctypedef cusparseMatDescr_t MatDescr
21
+
22
+
23
+ ###############################################################################
24
+ # Enum
25
+ ###############################################################################
26
+
27
+
28
+
29
+
30
+ ###############################################################################
31
+ # Functions
32
+ ###############################################################################
33
+
34
+ cpdef intptr_t create() except? 0
35
+ cpdef destroy(intptr_t handle)
36
+ cpdef set_stream(intptr_t handle, intptr_t stream_id)
37
+ cpdef intptr_t get_stream(intptr_t handle) except? 0
38
+ cpdef xcsrissym_host(intptr_t handle, int m, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_end_ptr_a, intptr_t csr_col_ind_a, intptr_t issym)
39
+ cpdef scsrlsvlu_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
40
+ cpdef dcsrlsvlu_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
41
+ cpdef ccsrlsvlu_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
42
+ cpdef zcsrlsvlu_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
43
+ cpdef scsrlsvqr(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
44
+ cpdef dcsrlsvqr(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
45
+ cpdef ccsrlsvqr(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
46
+ cpdef zcsrlsvqr(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
47
+ cpdef scsrlsvqr_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
48
+ cpdef dcsrlsvqr_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
49
+ cpdef ccsrlsvqr_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
50
+ cpdef zcsrlsvqr_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
51
+ cpdef scsrlsvchol_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
52
+ cpdef dcsrlsvchol_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
53
+ cpdef ccsrlsvchol_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
54
+ cpdef zcsrlsvchol_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
55
+ cpdef scsrlsvchol(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
56
+ cpdef dcsrlsvchol(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
57
+ cpdef ccsrlsvchol(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, float tol, int reorder, intptr_t x, intptr_t singularity)
58
+ cpdef zcsrlsvchol(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, intptr_t b, double tol, int reorder, intptr_t x, intptr_t singularity)
59
+ cpdef scsrlsqvqr_host(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, intptr_t rank_a, intptr_t x, intptr_t p, intptr_t min_norm)
60
+ cpdef dcsrlsqvqr_host(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, intptr_t rank_a, intptr_t x, intptr_t p, intptr_t min_norm)
61
+ cpdef ccsrlsqvqr_host(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, float tol, intptr_t rank_a, intptr_t x, intptr_t p, intptr_t min_norm)
62
+ cpdef zcsrlsqvqr_host(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, double tol, intptr_t rank_a, intptr_t x, intptr_t p, intptr_t min_norm)
63
+ cpdef scsreigvsi_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, float mu0, intptr_t x0, int maxite, float tol, intptr_t mu, intptr_t x)
64
+ cpdef dcsreigvsi_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, double mu0, intptr_t x0, int maxite, double tol, intptr_t mu, intptr_t x)
65
+ cpdef ccsreigvsi_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex mu0, intptr_t x0, int maxite, float tol, intptr_t mu, intptr_t x)
66
+ cpdef zcsreigvsi_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex mu0, intptr_t x0, int maxite, double tol, intptr_t mu, intptr_t x)
67
+ cpdef scsreigvsi(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, float mu0, intptr_t x0, int maxite, float eps, intptr_t mu, intptr_t x)
68
+ cpdef dcsreigvsi(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, double mu0, intptr_t x0, int maxite, double eps, intptr_t mu, intptr_t x)
69
+ cpdef ccsreigvsi(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex mu0, intptr_t x0, int maxite, float eps, intptr_t mu, intptr_t x)
70
+ cpdef zcsreigvsi(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex mu0, intptr_t x0, int maxite, double eps, intptr_t mu, intptr_t x)
71
+ cpdef scsreigs_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex left_bottom_corner, complex right_upper_corner, intptr_t num_eigs)
72
+ cpdef dcsreigs_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex left_bottom_corner, complex right_upper_corner, intptr_t num_eigs)
73
+ cpdef ccsreigs_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex left_bottom_corner, complex right_upper_corner, intptr_t num_eigs)
74
+ cpdef zcsreigs_host(intptr_t handle, int m, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, complex left_bottom_corner, complex right_upper_corner, intptr_t num_eigs)
75
+ cpdef xcsrsymrcm_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p)
76
+ cpdef xcsrsymmdq_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p)
77
+ cpdef xcsrsymamd_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p)
78
+ cpdef xcsrmetisnd_host(intptr_t handle, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t options, intptr_t p)
79
+ cpdef scsrzfd_host(intptr_t handle, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t numnz)
80
+ cpdef dcsrzfd_host(intptr_t handle, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t numnz)
81
+ cpdef ccsrzfd_host(intptr_t handle, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t numnz)
82
+ cpdef zcsrzfd_host(intptr_t handle, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t numnz)
83
+ cpdef size_t xcsrperm_buffer_size_host(intptr_t handle, int m, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t q) except? 0
84
+ cpdef xcsrperm_host(intptr_t handle, int m, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t p, intptr_t q, intptr_t map, intptr_t p_buffer)
85
+ cpdef intptr_t create_csrqr_info() except? 0
86
+ cpdef destroy_csrqr_info(intptr_t info)
87
+ cpdef xcsrqr_analysis_batched(intptr_t handle, int m, int n, int nnz_a, intptr_t descr_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t info)
88
+ cpdef tuple scsrqr_buffer_info_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, int batch_size, intptr_t info)
89
+ cpdef tuple dcsrqr_buffer_info_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, int batch_size, intptr_t info)
90
+ cpdef tuple ccsrqr_buffer_info_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, int batch_size, intptr_t info)
91
+ cpdef tuple zcsrqr_buffer_info_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val, intptr_t csr_row_ptr, intptr_t csr_col_ind, int batch_size, intptr_t info)
92
+ cpdef scsrqrsv_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, intptr_t x, int batch_size, intptr_t info, intptr_t p_buffer)
93
+ cpdef dcsrqrsv_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, intptr_t x, int batch_size, intptr_t info, intptr_t p_buffer)
94
+ cpdef ccsrqrsv_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, intptr_t x, int batch_size, intptr_t info, intptr_t p_buffer)
95
+ cpdef zcsrqrsv_batched(intptr_t handle, int m, int n, int nnz, intptr_t descr_a, intptr_t csr_val_a, intptr_t csr_row_ptr_a, intptr_t csr_col_ind_a, intptr_t b, intptr_t x, int batch_size, intptr_t info, intptr_t p_buffer)
nvmath/bindings/cusolverSp.pyi ADDED
@@ -0,0 +1,70 @@
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
+
7
+ __pyx_capi__: dict
8
+ __test__: dict
9
+ ccsreigs_host: _cython_3_2_5.cython_function_or_method
10
+ ccsreigvsi: _cython_3_2_5.cython_function_or_method
11
+ ccsreigvsi_host: _cython_3_2_5.cython_function_or_method
12
+ ccsrlsqvqr_host: _cython_3_2_5.cython_function_or_method
13
+ ccsrlsvchol: _cython_3_2_5.cython_function_or_method
14
+ ccsrlsvchol_host: _cython_3_2_5.cython_function_or_method
15
+ ccsrlsvlu_host: _cython_3_2_5.cython_function_or_method
16
+ ccsrlsvqr: _cython_3_2_5.cython_function_or_method
17
+ ccsrlsvqr_host: _cython_3_2_5.cython_function_or_method
18
+ ccsrqr_buffer_info_batched: _cython_3_2_5.cython_function_or_method
19
+ ccsrqrsv_batched: _cython_3_2_5.cython_function_or_method
20
+ ccsrzfd_host: _cython_3_2_5.cython_function_or_method
21
+ create: _cython_3_2_5.cython_function_or_method
22
+ create_csrqr_info: _cython_3_2_5.cython_function_or_method
23
+ dcsreigs_host: _cython_3_2_5.cython_function_or_method
24
+ dcsreigvsi: _cython_3_2_5.cython_function_or_method
25
+ dcsreigvsi_host: _cython_3_2_5.cython_function_or_method
26
+ dcsrlsqvqr_host: _cython_3_2_5.cython_function_or_method
27
+ dcsrlsvchol: _cython_3_2_5.cython_function_or_method
28
+ dcsrlsvchol_host: _cython_3_2_5.cython_function_or_method
29
+ dcsrlsvlu_host: _cython_3_2_5.cython_function_or_method
30
+ dcsrlsvqr: _cython_3_2_5.cython_function_or_method
31
+ dcsrlsvqr_host: _cython_3_2_5.cython_function_or_method
32
+ dcsrqr_buffer_info_batched: _cython_3_2_5.cython_function_or_method
33
+ dcsrqrsv_batched: _cython_3_2_5.cython_function_or_method
34
+ dcsrzfd_host: _cython_3_2_5.cython_function_or_method
35
+ destroy: _cython_3_2_5.cython_function_or_method
36
+ destroy_csrqr_info: _cython_3_2_5.cython_function_or_method
37
+ get_stream: _cython_3_2_5.cython_function_or_method
38
+ scsreigs_host: _cython_3_2_5.cython_function_or_method
39
+ scsreigvsi: _cython_3_2_5.cython_function_or_method
40
+ scsreigvsi_host: _cython_3_2_5.cython_function_or_method
41
+ scsrlsqvqr_host: _cython_3_2_5.cython_function_or_method
42
+ scsrlsvchol: _cython_3_2_5.cython_function_or_method
43
+ scsrlsvchol_host: _cython_3_2_5.cython_function_or_method
44
+ scsrlsvlu_host: _cython_3_2_5.cython_function_or_method
45
+ scsrlsvqr: _cython_3_2_5.cython_function_or_method
46
+ scsrlsvqr_host: _cython_3_2_5.cython_function_or_method
47
+ scsrqr_buffer_info_batched: _cython_3_2_5.cython_function_or_method
48
+ scsrqrsv_batched: _cython_3_2_5.cython_function_or_method
49
+ scsrzfd_host: _cython_3_2_5.cython_function_or_method
50
+ set_stream: _cython_3_2_5.cython_function_or_method
51
+ xcsrissym_host: _cython_3_2_5.cython_function_or_method
52
+ xcsrmetisnd_host: _cython_3_2_5.cython_function_or_method
53
+ xcsrperm_buffer_size_host: _cython_3_2_5.cython_function_or_method
54
+ xcsrperm_host: _cython_3_2_5.cython_function_or_method
55
+ xcsrqr_analysis_batched: _cython_3_2_5.cython_function_or_method
56
+ xcsrsymamd_host: _cython_3_2_5.cython_function_or_method
57
+ xcsrsymmdq_host: _cython_3_2_5.cython_function_or_method
58
+ xcsrsymrcm_host: _cython_3_2_5.cython_function_or_method
59
+ zcsreigs_host: _cython_3_2_5.cython_function_or_method
60
+ zcsreigvsi: _cython_3_2_5.cython_function_or_method
61
+ zcsreigvsi_host: _cython_3_2_5.cython_function_or_method
62
+ zcsrlsqvqr_host: _cython_3_2_5.cython_function_or_method
63
+ zcsrlsvchol: _cython_3_2_5.cython_function_or_method
64
+ zcsrlsvchol_host: _cython_3_2_5.cython_function_or_method
65
+ zcsrlsvlu_host: _cython_3_2_5.cython_function_or_method
66
+ zcsrlsvqr: _cython_3_2_5.cython_function_or_method
67
+ zcsrlsvqr_host: _cython_3_2_5.cython_function_or_method
68
+ zcsrqr_buffer_info_batched: _cython_3_2_5.cython_function_or_method
69
+ zcsrqrsv_batched: _cython_3_2_5.cython_function_or_method
70
+ zcsrzfd_host: _cython_3_2_5.cython_function_or_method