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,552 @@
1
+ # This code was automatically generated across versions from 0.3.1 to 0.3.2, generator version 0.3.1.dev1418+g63712a33a.d20260318. Do not modify it directly.
2
+ # This layer exposes the C header to Cython as-is.
3
+
4
+ from libc.stdint cimport int64_t
5
+
6
+
7
+ ###############################################################################
8
+ # Types (structs, enums, ...)
9
+ ###############################################################################
10
+
11
+ # enums
12
+ ctypedef enum commondxValueType "commondxValueType":
13
+ COMMONDX_R_8F_E5M2 "COMMONDX_R_8F_E5M2" = 0
14
+ COMMONDX_C_8F_E5M2 "COMMONDX_C_8F_E5M2" = 1
15
+ COMMONDX_R_8F_E4M3 "COMMONDX_R_8F_E4M3" = 2
16
+ COMMONDX_C_8F_E4M3 "COMMONDX_C_8F_E4M3" = 3
17
+ COMMONDX_R_16BF "COMMONDX_R_16BF" = 4
18
+ COMMONDX_C_16BF "COMMONDX_C_16BF" = 5
19
+ COMMONDX_R_16F2 "COMMONDX_R_16F2" = 6
20
+ COMMONDX_R_16F "COMMONDX_R_16F" = 7
21
+ COMMONDX_C_16F "COMMONDX_C_16F" = 8
22
+ COMMONDX_C_16F2 "COMMONDX_C_16F2" = 9
23
+ COMMONDX_R_32TF "COMMONDX_R_32TF" = 10
24
+ COMMONDX_C_32TF "COMMONDX_C_32TF" = 11
25
+ COMMONDX_R_32F "COMMONDX_R_32F" = 12
26
+ COMMONDX_C_32F "COMMONDX_C_32F" = 13
27
+ COMMONDX_R_64F "COMMONDX_R_64F" = 14
28
+ COMMONDX_C_64F "COMMONDX_C_64F" = 15
29
+ COMMONDX_R_8I "COMMONDX_R_8I" = 16
30
+ COMMONDX_C_8I "COMMONDX_C_8I" = 17
31
+ COMMONDX_R_16I "COMMONDX_R_16I" = 18
32
+ COMMONDX_C_16I "COMMONDX_C_16I" = 19
33
+ COMMONDX_R_32I "COMMONDX_R_32I" = 20
34
+ COMMONDX_C_32I "COMMONDX_C_32I" = 21
35
+ COMMONDX_R_64I "COMMONDX_R_64I" = 22
36
+ COMMONDX_C_64I "COMMONDX_C_64I" = 23
37
+ COMMONDX_R_8UI "COMMONDX_R_8UI" = 24
38
+ COMMONDX_C_8UI "COMMONDX_C_8UI" = 25
39
+ COMMONDX_R_16UI "COMMONDX_R_16UI" = 26
40
+ COMMONDX_C_16UI "COMMONDX_C_16UI" = 27
41
+ COMMONDX_R_32UI "COMMONDX_R_32UI" = 28
42
+ COMMONDX_C_32UI "COMMONDX_C_32UI" = 29
43
+ COMMONDX_R_64UI "COMMONDX_R_64UI" = 30
44
+ COMMONDX_C_64UI "COMMONDX_C_64UI" = 31
45
+
46
+ ctypedef enum commondxStatusType "commondxStatusType":
47
+ COMMONDX_SUCCESS "COMMONDX_SUCCESS" = 0
48
+ COMMONDX_INVALID_VALUE "COMMONDX_INVALID_VALUE" = 1
49
+ COMMONDX_INTERNAL_ERROR "COMMONDX_INTERNAL_ERROR" = 2
50
+ COMMONDX_COMPILATION_ERROR "COMMONDX_COMPILATION_ERROR" = 3
51
+ COMMONDX_CUFFT_ERROR "COMMONDX_CUFFT_ERROR" = 4
52
+ _COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR "_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR" = -42
53
+
54
+ ctypedef enum commondxPrecision "commondxPrecision":
55
+ COMMONDX_PRECISION_F8_E5M2 "COMMONDX_PRECISION_F8_E5M2" = 0
56
+ COMMONDX_PRECISION_F8_E4M3 "COMMONDX_PRECISION_F8_E4M3" = 1
57
+ COMMONDX_PRECISION_BF16 "COMMONDX_PRECISION_BF16" = 2
58
+ COMMONDX_PRECISION_F16 "COMMONDX_PRECISION_F16" = 3
59
+ COMMONDX_PRECISION_TF32 "COMMONDX_PRECISION_TF32" = 4
60
+ COMMONDX_PRECISION_F32 "COMMONDX_PRECISION_F32" = 5
61
+ COMMONDX_PRECISION_F64 "COMMONDX_PRECISION_F64" = 6
62
+ COMMONDX_PRECISION_I8 "COMMONDX_PRECISION_I8" = 7
63
+ COMMONDX_PRECISION_I16 "COMMONDX_PRECISION_I16" = 8
64
+ COMMONDX_PRECISION_I32 "COMMONDX_PRECISION_I32" = 9
65
+ COMMONDX_PRECISION_I64 "COMMONDX_PRECISION_I64" = 10
66
+ COMMONDX_PRECISION_UI8 "COMMONDX_PRECISION_UI8" = 11
67
+ COMMONDX_PRECISION_UI16 "COMMONDX_PRECISION_UI16" = 12
68
+ COMMONDX_PRECISION_UI32 "COMMONDX_PRECISION_UI32" = 13
69
+ COMMONDX_PRECISION_UI64 "COMMONDX_PRECISION_UI64" = 14
70
+
71
+ ctypedef enum commondxArchModifier "commondxArchModifier":
72
+ COMMONDX_ARCH_MODIFIER_GENERIC "COMMONDX_ARCH_MODIFIER_GENERIC" = 0
73
+ COMMONDX_ARCH_MODIFIER_ARCH_SPECIFIC "COMMONDX_ARCH_MODIFIER_ARCH_SPECIFIC" = 1
74
+ COMMONDX_ARCH_MODIFIER_FAMILY_SPECIFIC "COMMONDX_ARCH_MODIFIER_FAMILY_SPECIFIC" = 2
75
+
76
+ ctypedef enum commondxOption "commondxOption":
77
+ COMMONDX_OPTION_SYMBOL_NAME "COMMONDX_OPTION_SYMBOL_NAME" = 0
78
+ COMMONDX_OPTION_TARGET_SM "COMMONDX_OPTION_TARGET_SM" = 1
79
+ COMMONDX_OPTION_CODE_CONTAINER "COMMONDX_OPTION_CODE_CONTAINER" = 2
80
+ COMMONDX_OPTION_CODE_ISA "COMMONDX_OPTION_CODE_ISA" = 3
81
+ COMMONDX_OPTION_EXTRA_NVRTC_ARGS "COMMONDX_OPTION_EXTRA_NVRTC_ARGS" = 4
82
+ COMMONDX_OPTION_TARGET_CODE "COMMONDX_OPTION_TARGET_CODE" = 5
83
+
84
+ ctypedef enum commondxExecution "commondxExecution":
85
+ COMMONDX_EXECUTION_THREAD "COMMONDX_EXECUTION_THREAD" = 0
86
+ COMMONDX_EXECUTION_BLOCK "COMMONDX_EXECUTION_BLOCK" = 1
87
+
88
+ ctypedef enum commondxCodeContainer "commondxCodeContainer":
89
+ COMMONDX_CODE_CONTAINER_LTOIR "COMMONDX_CODE_CONTAINER_LTOIR" = 0
90
+ COMMONDX_CODE_CONTAINER_FATBIN "COMMONDX_CODE_CONTAINER_FATBIN" = 1
91
+ COMMONDX_CODE_CONTAINER_PTX "COMMONDX_CODE_CONTAINER_PTX" = 2
92
+
93
+ ctypedef enum cublasdxApi "cublasdxApi":
94
+ CUBLASDX_API_SMEM "CUBLASDX_API_SMEM" = 0
95
+ CUBLASDX_API_SMEM_DYNAMIC_LD "CUBLASDX_API_SMEM_DYNAMIC_LD" = 1
96
+ CUBLASDX_API_TENSORS "CUBLASDX_API_TENSORS" = 2
97
+
98
+ ctypedef enum cublasdxDevicePipelineType "cublasdxDevicePipelineType":
99
+ CUBLASDX_DEVICE_PIPELINE_SUGGESTED "CUBLASDX_DEVICE_PIPELINE_SUGGESTED" = 0
100
+
101
+ ctypedef enum cublasdxTilePipelineType "cublasdxTilePipelineType":
102
+ CUBLASDX_TILE_PIPELINE_DEFAULT "CUBLASDX_TILE_PIPELINE_DEFAULT" = 0
103
+
104
+ ctypedef enum cublasdxBlockSizeStrategy "cublasdxBlockSizeStrategy":
105
+ CUBLASDX_BLOCK_SIZE_STRATEGY_HEURISTIC "CUBLASDX_BLOCK_SIZE_STRATEGY_HEURISTIC" = 0
106
+ CUBLASDX_BLOCK_SIZE_STRATEGY_FIXED "CUBLASDX_BLOCK_SIZE_STRATEGY_FIXED" = 1
107
+
108
+ ctypedef enum cublasdxType "cublasdxType":
109
+ CUBLASDX_TYPE_REAL "CUBLASDX_TYPE_REAL" = 0
110
+ CUBLASDX_TYPE_COMPLEX "CUBLASDX_TYPE_COMPLEX" = 1
111
+
112
+ ctypedef enum cublasdxMemorySpace "cublasdxMemorySpace":
113
+ CUBLASDX_MEMORY_SPACE_RMEM "CUBLASDX_MEMORY_SPACE_RMEM" = 0
114
+ CUBLASDX_MEMORY_SPACE_SMEM "CUBLASDX_MEMORY_SPACE_SMEM" = 1
115
+ CUBLASDX_MEMORY_SPACE_GMEM "CUBLASDX_MEMORY_SPACE_GMEM" = 2
116
+ CUBLASDX_MEMORY_SPACE_ANY "CUBLASDX_MEMORY_SPACE_ANY" = 3
117
+
118
+ ctypedef enum cublasdxTransposeMode "cublasdxTransposeMode":
119
+ CUBLASDX_TRANSPOSE_MODE_NON_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_NON_TRANSPOSED" = 0
120
+ CUBLASDX_TRANSPOSE_MODE_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_TRANSPOSED" = 1
121
+ CUBLASDX_TRANSPOSE_MODE_CONJ_TRANSPOSED "CUBLASDX_TRANSPOSE_MODE_CONJ_TRANSPOSED" = 2
122
+
123
+ ctypedef enum cublasdxArrangement "cublasdxArrangement":
124
+ CUBLASDX_ARRANGEMENT_COL_MAJOR "CUBLASDX_ARRANGEMENT_COL_MAJOR" = 0
125
+ CUBLASDX_ARRANGEMENT_ROW_MAJOR "CUBLASDX_ARRANGEMENT_ROW_MAJOR" = 1
126
+
127
+ ctypedef enum cublasdxFunction "cublasdxFunction":
128
+ CUBLASDX_FUNCTION_MM "CUBLASDX_FUNCTION_MM" = 0
129
+
130
+ ctypedef enum cublasdxOperatorType "cublasdxOperatorType":
131
+ CUBLASDX_OPERATOR_FUNCTION "CUBLASDX_OPERATOR_FUNCTION" = 0
132
+ CUBLASDX_OPERATOR_SIZE "CUBLASDX_OPERATOR_SIZE" = 1
133
+ CUBLASDX_OPERATOR_TYPE "CUBLASDX_OPERATOR_TYPE" = 2
134
+ CUBLASDX_OPERATOR_PRECISION "CUBLASDX_OPERATOR_PRECISION" = 3
135
+ CUBLASDX_OPERATOR_SM "CUBLASDX_OPERATOR_SM" = 4
136
+ CUBLASDX_OPERATOR_EXECUTION "CUBLASDX_OPERATOR_EXECUTION" = 5
137
+ CUBLASDX_OPERATOR_BLOCK_DIM "CUBLASDX_OPERATOR_BLOCK_DIM" = 6
138
+ CUBLASDX_OPERATOR_LEADING_DIMENSION "CUBLASDX_OPERATOR_LEADING_DIMENSION" = 7
139
+ CUBLASDX_OPERATOR_TRANSPOSE_MODE "CUBLASDX_OPERATOR_TRANSPOSE_MODE" = 8
140
+ CUBLASDX_OPERATOR_API "CUBLASDX_OPERATOR_API" = 9
141
+ CUBLASDX_OPERATOR_ARRANGEMENT "CUBLASDX_OPERATOR_ARRANGEMENT" = 10
142
+ CUBLASDX_OPERATOR_ALIGNMENT "CUBLASDX_OPERATOR_ALIGNMENT" = 11
143
+ CUBLASDX_OPERATOR_STATIC_BLOCK_DIM "CUBLASDX_OPERATOR_STATIC_BLOCK_DIM" = 12
144
+ CUBLASDX_OPERATOR_ENABLE_INPUT_STREAMING "CUBLASDX_OPERATOR_ENABLE_INPUT_STREAMING" = 13
145
+ CUBLASDX_OPERATOR_WITH_PIPELINE "CUBLASDX_OPERATOR_WITH_PIPELINE" = 14
146
+
147
+ ctypedef enum cublasdxTraitType "cublasdxTraitType":
148
+ CUBLASDX_TRAIT_VALUE_TYPE "CUBLASDX_TRAIT_VALUE_TYPE" = 0
149
+ CUBLASDX_TRAIT_SIZE "CUBLASDX_TRAIT_SIZE" = 1
150
+ CUBLASDX_TRAIT_BLOCK_SIZE "CUBLASDX_TRAIT_BLOCK_SIZE" = 2
151
+ CUBLASDX_TRAIT_BLOCK_DIM "CUBLASDX_TRAIT_BLOCK_DIM" = 3
152
+ CUBLASDX_TRAIT_LEADING_DIMENSION "CUBLASDX_TRAIT_LEADING_DIMENSION" = 4
153
+ CUBLASDX_TRAIT_SYMBOL_NAME "CUBLASDX_TRAIT_SYMBOL_NAME" = 5
154
+ CUBLASDX_TRAIT_ARRANGEMENT "CUBLASDX_TRAIT_ARRANGEMENT" = 6
155
+ CUBLASDX_TRAIT_ALIGNMENT "CUBLASDX_TRAIT_ALIGNMENT" = 7
156
+ CUBLASDX_TRAIT_SUGGESTED_LEADING_DIMENSION "CUBLASDX_TRAIT_SUGGESTED_LEADING_DIMENSION" = 8
157
+ CUBLASDX_TRAIT_SUGGESTED_BLOCK_DIM "CUBLASDX_TRAIT_SUGGESTED_BLOCK_DIM" = 9
158
+ CUBLASDX_TRAIT_MAX_THREADS_PER_BLOCK "CUBLASDX_TRAIT_MAX_THREADS_PER_BLOCK" = 10
159
+
160
+ ctypedef enum cublasdxTensorType "cublasdxTensorType":
161
+ CUBLASDX_TENSOR_SMEM_A "CUBLASDX_TENSOR_SMEM_A" = 0
162
+ CUBLASDX_TENSOR_SMEM_B "CUBLASDX_TENSOR_SMEM_B" = 1
163
+ CUBLASDX_TENSOR_SMEM_C "CUBLASDX_TENSOR_SMEM_C" = 2
164
+ CUBLASDX_TENSOR_SUGGESTED_SMEM_A "CUBLASDX_TENSOR_SUGGESTED_SMEM_A" = 3
165
+ CUBLASDX_TENSOR_SUGGESTED_SMEM_B "CUBLASDX_TENSOR_SUGGESTED_SMEM_B" = 4
166
+ CUBLASDX_TENSOR_SUGGESTED_SMEM_C "CUBLASDX_TENSOR_SUGGESTED_SMEM_C" = 5
167
+ CUBLASDX_TENSOR_SUGGESTED_RMEM_C "CUBLASDX_TENSOR_SUGGESTED_RMEM_C" = 6
168
+ CUBLASDX_TENSOR_GMEM_A "CUBLASDX_TENSOR_GMEM_A" = 7
169
+ CUBLASDX_TENSOR_GMEM_B "CUBLASDX_TENSOR_GMEM_B" = 8
170
+ CUBLASDX_TENSOR_GMEM_C "CUBLASDX_TENSOR_GMEM_C" = 9
171
+ CUBLASDX_TENSOR_SUGGESTED_ACCUMULATOR_C "CUBLASDX_TENSOR_SUGGESTED_ACCUMULATOR_C" = 10
172
+ CUBLASDX_TENSOR_RMEM_C "CUBLASDX_TENSOR_RMEM_C" = 11
173
+ CUBLASDX_TENSOR_ACCUMULATOR_C "CUBLASDX_TENSOR_ACCUMULATOR_C" = 12
174
+
175
+ ctypedef enum cublasdxTensorOption "cublasdxTensorOption":
176
+ CUBLASDX_TENSOR_OPTION_ALIGNMENT_BYTES "CUBLASDX_TENSOR_OPTION_ALIGNMENT_BYTES" = 0
177
+
178
+ ctypedef enum cublasdxTensorTrait "cublasdxTensorTrait":
179
+ CUBLASDX_TENSOR_TRAIT_STORAGE_BYTES "CUBLASDX_TENSOR_TRAIT_STORAGE_BYTES" = 0
180
+ CUBLASDX_TENSOR_TRAIT_ALIGNMENT_BYTES "CUBLASDX_TENSOR_TRAIT_ALIGNMENT_BYTES" = 1
181
+ CUBLASDX_TENSOR_TRAIT_UID "CUBLASDX_TENSOR_TRAIT_UID" = 2
182
+ CUBLASDX_TENSOR_TRAIT_OPAQUE_NAME "CUBLASDX_TENSOR_TRAIT_OPAQUE_NAME" = 4
183
+ CUBLASDX_TENSOR_TRAIT_LOGICAL_SIZE "CUBLASDX_TENSOR_TRAIT_LOGICAL_SIZE" = 5
184
+ CUBLASDX_TENSOR_TRAIT_MEMORY_SPACE "CUBLASDX_TENSOR_TRAIT_MEMORY_SPACE" = 6
185
+
186
+ ctypedef enum cublasdxPipelineTrait "cublasdxPipelineTrait":
187
+ CUBLASDX_PIPELINE_TRAIT_STORAGE_BYTES "CUBLASDX_PIPELINE_TRAIT_STORAGE_BYTES" = 0
188
+ CUBLASDX_PIPELINE_TRAIT_STORAGE_ALIGNMENT_BYTES "CUBLASDX_PIPELINE_TRAIT_STORAGE_ALIGNMENT_BYTES" = 1
189
+ CUBLASDX_PIPELINE_TRAIT_BUFFER_SIZE "CUBLASDX_PIPELINE_TRAIT_BUFFER_SIZE" = 2
190
+ CUBLASDX_PIPELINE_TRAIT_BUFFER_ALIGNMENT_BYTES "CUBLASDX_PIPELINE_TRAIT_BUFFER_ALIGNMENT_BYTES" = 3
191
+ CUBLASDX_PIPELINE_TRAIT_OPAQUE_NAME "CUBLASDX_PIPELINE_TRAIT_OPAQUE_NAME" = 4
192
+ CUBLASDX_PIPELINE_TRAIT_BLOCK_DIM "CUBLASDX_PIPELINE_TRAIT_BLOCK_DIM" = 5
193
+
194
+ ctypedef enum cublasdxDeviceFunctionTrait "cublasdxDeviceFunctionTrait":
195
+ CUBLASDX_DEVICE_FUNCTION_TRAIT_SYMBOL "CUBLASDX_DEVICE_FUNCTION_TRAIT_SYMBOL" = 1
196
+
197
+ ctypedef enum cublasdxDeviceFunctionOption "cublasdxDeviceFunctionOption":
198
+ CUBLASDX_DEVICE_FUNCTION_OPTION_SYMBOL_NAME "CUBLASDX_DEVICE_FUNCTION_OPTION_SYMBOL_NAME" = 0
199
+ CUBLASDX_DEVICE_FUNCTION_OPTION_COPY_ALIGNMENT "CUBLASDX_DEVICE_FUNCTION_OPTION_COPY_ALIGNMENT" = 1
200
+ CUBLASDX_DEVICE_FUNCTION_OPTION_CALLBACK "CUBLASDX_DEVICE_FUNCTION_OPTION_CALLBACK" = 2
201
+ CUBLASDX_DEVICE_FUNCTION_OPTION_NUM_THREADS "CUBLASDX_DEVICE_FUNCTION_OPTION_NUM_THREADS" = 3
202
+
203
+ ctypedef enum cublasdxDeviceFunctionType "cublasdxDeviceFunctionType":
204
+ CUBLASDX_DEVICE_FUNCTION_EXECUTE "CUBLASDX_DEVICE_FUNCTION_EXECUTE" = 0
205
+ CUBLASDX_DEVICE_FUNCTION_COPY "CUBLASDX_DEVICE_FUNCTION_COPY" = 1
206
+ CUBLASDX_DEVICE_FUNCTION_COPY_WAIT "CUBLASDX_DEVICE_FUNCTION_COPY_WAIT" = 2
207
+ CUBLASDX_DEVICE_FUNCTION_CLEAR "CUBLASDX_DEVICE_FUNCTION_CLEAR" = 3
208
+ CUBLASDX_DEVICE_FUNCTION_AXPBY "CUBLASDX_DEVICE_FUNCTION_AXPBY" = 4
209
+ CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD "CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD" = 5
210
+ CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD_PARTITIONER "CUBLASDX_DEVICE_FUNCTION_MAP_IDX2CRD_PARTITIONER" = 6
211
+ CUBLASDX_DEVICE_FUNCTION_MAP_CRD2IDX "CUBLASDX_DEVICE_FUNCTION_MAP_CRD2IDX" = 7
212
+ CUBLASDX_DEVICE_FUNCTION_IS_THREAD_ACTIVE "CUBLASDX_DEVICE_FUNCTION_IS_THREAD_ACTIVE" = 8
213
+ CUBLASDX_DEVICE_FUNCTION_IS_PREDICATED "CUBLASDX_DEVICE_FUNCTION_IS_PREDICATED" = 9
214
+ CUBLASDX_DEVICE_FUNCTION_IS_INDEX_IN_BOUNDS "CUBLASDX_DEVICE_FUNCTION_IS_INDEX_IN_BOUNDS" = 10
215
+ CUBLASDX_DEVICE_FUNCTION_CREATE "CUBLASDX_DEVICE_FUNCTION_CREATE" = 11
216
+ CUBLASDX_DEVICE_FUNCTION_DESTROY "CUBLASDX_DEVICE_FUNCTION_DESTROY" = 12
217
+ CUBLASDX_DEVICE_FUNCTION_RESET "CUBLASDX_DEVICE_FUNCTION_RESET" = 13
218
+ CUBLASDX_DEVICE_FUNCTION_EPILOGUE "CUBLASDX_DEVICE_FUNCTION_EPILOGUE" = 14
219
+
220
+ ctypedef enum cufftdxApi "cufftdxApi":
221
+ CUFFTDX_API_LMEM "CUFFTDX_API_LMEM" = 0
222
+ CUFFTDX_API_SMEM "CUFFTDX_API_SMEM" = 1
223
+
224
+ ctypedef enum cufftdxType "cufftdxType":
225
+ CUFFTDX_TYPE_C2C "CUFFTDX_TYPE_C2C" = 0
226
+ CUFFTDX_TYPE_R2C "CUFFTDX_TYPE_R2C" = 1
227
+ CUFFTDX_TYPE_C2R "CUFFTDX_TYPE_C2R" = 2
228
+
229
+ ctypedef enum cufftdxDirection "cufftdxDirection":
230
+ CUFFTDX_DIRECTION_FORWARD "CUFFTDX_DIRECTION_FORWARD" = 0
231
+ CUFFTDX_DIRECTION_INVERSE "CUFFTDX_DIRECTION_INVERSE" = 1
232
+
233
+ ctypedef enum cufftdxComplexLayout "cufftdxComplexLayout":
234
+ CUFFTDX_COMPLEX_LAYOUT_NATURAL "CUFFTDX_COMPLEX_LAYOUT_NATURAL" = 0
235
+ CUFFTDX_COMPLEX_LAYOUT_PACKED "CUFFTDX_COMPLEX_LAYOUT_PACKED" = 1
236
+ CUFFTDX_COMPLEX_LAYOUT_FULL "CUFFTDX_COMPLEX_LAYOUT_FULL" = 2
237
+
238
+ ctypedef enum cufftdxRealMode "cufftdxRealMode":
239
+ CUFFTDX_REAL_MODE_NORMAL "CUFFTDX_REAL_MODE_NORMAL" = 0
240
+ CUFFTDX_REAL_MODE_FOLDED "CUFFTDX_REAL_MODE_FOLDED" = 1
241
+
242
+ ctypedef enum cufftdxCodeType "cufftdxCodeType":
243
+ CUFFTDX_CODE_TYPE_PTX "CUFFTDX_CODE_TYPE_PTX" = 0
244
+ CUFFTDX_CODE_TYPE_LTOIR "CUFFTDX_CODE_TYPE_LTOIR" = 1
245
+
246
+ ctypedef enum cufftdxOperatorType "cufftdxOperatorType":
247
+ CUFFTDX_OPERATOR_SIZE "CUFFTDX_OPERATOR_SIZE" = 0
248
+ CUFFTDX_OPERATOR_DIRECTION "CUFFTDX_OPERATOR_DIRECTION" = 1
249
+ CUFFTDX_OPERATOR_TYPE "CUFFTDX_OPERATOR_TYPE" = 2
250
+ CUFFTDX_OPERATOR_PRECISION "CUFFTDX_OPERATOR_PRECISION" = 3
251
+ CUFFTDX_OPERATOR_SM "CUFFTDX_OPERATOR_SM" = 4
252
+ CUFFTDX_OPERATOR_EXECUTION "CUFFTDX_OPERATOR_EXECUTION" = 5
253
+ CUFFTDX_OPERATOR_FFTS_PER_BLOCK "CUFFTDX_OPERATOR_FFTS_PER_BLOCK" = 6
254
+ CUFFTDX_OPERATOR_ELEMENTS_PER_THREAD "CUFFTDX_OPERATOR_ELEMENTS_PER_THREAD" = 7
255
+ CUFFTDX_OPERATOR_BLOCK_DIM "CUFFTDX_OPERATOR_BLOCK_DIM" = 8
256
+ CUFFTDX_OPERATOR_REAL_FFT_OPTIONS "CUFFTDX_OPERATOR_REAL_FFT_OPTIONS" = 9
257
+ CUFFTDX_OPERATOR_API "CUFFTDX_OPERATOR_API" = 10
258
+ CUFFTDX_OPERATOR_CODE_TYPE "CUFFTDX_OPERATOR_CODE_TYPE" = 11
259
+
260
+ ctypedef enum cufftdxKnobType "cufftdxKnobType":
261
+ CUFFTDX_KNOB_ELEMENTS_PER_THREAD "CUFFTDX_KNOB_ELEMENTS_PER_THREAD" = 0
262
+ CUFFTDX_KNOB_FFTS_PER_BLOCK "CUFFTDX_KNOB_FFTS_PER_BLOCK" = 1
263
+
264
+ ctypedef enum cufftdxTraitType "cufftdxTraitType":
265
+ CUFFTDX_TRAIT_VALUE_TYPE "CUFFTDX_TRAIT_VALUE_TYPE" = 0
266
+ CUFFTDX_TRAIT_INPUT_TYPE "CUFFTDX_TRAIT_INPUT_TYPE" = 1
267
+ CUFFTDX_TRAIT_OUTPUT_TYPE "CUFFTDX_TRAIT_OUTPUT_TYPE" = 2
268
+ CUFFTDX_TRAIT_IMPLICIT_TYPE_BATCHING "CUFFTDX_TRAIT_IMPLICIT_TYPE_BATCHING" = 3
269
+ CUFFTDX_TRAIT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_ELEMENTS_PER_THREAD" = 4
270
+ CUFFTDX_TRAIT_STORAGE_SIZE "CUFFTDX_TRAIT_STORAGE_SIZE" = 5
271
+ CUFFTDX_TRAIT_STRIDE "CUFFTDX_TRAIT_STRIDE" = 6
272
+ CUFFTDX_TRAIT_BLOCK_DIM "CUFFTDX_TRAIT_BLOCK_DIM" = 7
273
+ CUFFTDX_TRAIT_SHARED_MEMORY_SIZE "CUFFTDX_TRAIT_SHARED_MEMORY_SIZE" = 8
274
+ CUFFTDX_TRAIT_FFTS_PER_BLOCK "CUFFTDX_TRAIT_FFTS_PER_BLOCK" = 9
275
+ CUFFTDX_TRAIT_SYMBOL_NAME "CUFFTDX_TRAIT_SYMBOL_NAME" = 10
276
+ CUFFTDX_TRAIT_INPUT_LENGTH "CUFFTDX_TRAIT_INPUT_LENGTH" = 11
277
+ CUFFTDX_TRAIT_OUTPUT_LENGTH "CUFFTDX_TRAIT_OUTPUT_LENGTH" = 12
278
+ CUFFTDX_TRAIT_INPUT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_INPUT_ELEMENTS_PER_THREAD" = 13
279
+ CUFFTDX_TRAIT_OUTPUT_ELEMENTS_PER_THREAD "CUFFTDX_TRAIT_OUTPUT_ELEMENTS_PER_THREAD" = 14
280
+ CUFFTDX_TRAIT_SUGGESTED_FFTS_PER_BLOCK "CUFFTDX_TRAIT_SUGGESTED_FFTS_PER_BLOCK" = 15
281
+
282
+ ctypedef enum cusolverdxApi "cusolverdxApi":
283
+ CUSOLVERDX_API_SMEM "CUSOLVERDX_API_SMEM" = 0
284
+ CUSOLVERDX_API_SMEM_DYNAMIC_LD "CUSOLVERDX_API_SMEM_DYNAMIC_LD" = 1
285
+
286
+ ctypedef enum cusolverdxType "cusolverdxType":
287
+ CUSOLVERDX_TYPE_REAL "CUSOLVERDX_TYPE_REAL" = 0
288
+ CUSOLVERDX_TYPE_COMPLEX "CUSOLVERDX_TYPE_COMPLEX" = 1
289
+
290
+ ctypedef enum cusolverdxFunction "cusolverdxFunction":
291
+ CUSOLVERDX_FUNCTION_GETRF_NO_PIVOT "CUSOLVERDX_FUNCTION_GETRF_NO_PIVOT" = 0
292
+ CUSOLVERDX_FUNCTION_GETRS_NO_PIVOT "CUSOLVERDX_FUNCTION_GETRS_NO_PIVOT" = 1
293
+ CUSOLVERDX_FUNCTION_POTRF "CUSOLVERDX_FUNCTION_POTRF" = 2
294
+ CUSOLVERDX_FUNCTION_POTRS "CUSOLVERDX_FUNCTION_POTRS" = 3
295
+ CUSOLVERDX_FUNCTION_TRSM "CUSOLVERDX_FUNCTION_TRSM" = 4
296
+ CUSOLVERDX_FUNCTION_GETRF_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GETRF_PARTIAL_PIVOT" = 5
297
+ CUSOLVERDX_FUNCTION_GETRS_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GETRS_PARTIAL_PIVOT" = 6
298
+ CUSOLVERDX_FUNCTION_GEQRF "CUSOLVERDX_FUNCTION_GEQRF" = 7
299
+ CUSOLVERDX_FUNCTION_UNMQR "CUSOLVERDX_FUNCTION_UNMQR" = 8
300
+ CUSOLVERDX_FUNCTION_GELQF "CUSOLVERDX_FUNCTION_GELQF" = 9
301
+ CUSOLVERDX_FUNCTION_UNMLQ "CUSOLVERDX_FUNCTION_UNMLQ" = 10
302
+ CUSOLVERDX_FUNCTION_POSV "CUSOLVERDX_FUNCTION_POSV" = 11
303
+ CUSOLVERDX_FUNCTION_GESV_NO_PIVOT "CUSOLVERDX_FUNCTION_GESV_NO_PIVOT" = 12
304
+ CUSOLVERDX_FUNCTION_GESV_PARTIAL_PIVOT "CUSOLVERDX_FUNCTION_GESV_PARTIAL_PIVOT" = 13
305
+ CUSOLVERDX_FUNCTION_GELS "CUSOLVERDX_FUNCTION_GELS" = 14
306
+ CUSOLVERDX_FUNCTION_UNGQR "CUSOLVERDX_FUNCTION_UNGQR" = 15
307
+ CUSOLVERDX_FUNCTION_UNGLQ "CUSOLVERDX_FUNCTION_UNGLQ" = 16
308
+ CUSOLVERDX_FUNCTION_GTSV_NO_PIVOT "CUSOLVERDX_FUNCTION_GTSV_NO_PIVOT" = 17
309
+ CUSOLVERDX_FUNCTION_HTEV "CUSOLVERDX_FUNCTION_HTEV" = 18
310
+ CUSOLVERDX_FUNCTION_HEEV "CUSOLVERDX_FUNCTION_HEEV" = 19
311
+
312
+ ctypedef enum cusolverdxArrangement "cusolverdxArrangement":
313
+ CUSOLVERDX_ARRANGEMENT_COL_MAJOR "CUSOLVERDX_ARRANGEMENT_COL_MAJOR" = 0
314
+ CUSOLVERDX_ARRANGEMENT_ROW_MAJOR "CUSOLVERDX_ARRANGEMENT_ROW_MAJOR" = 1
315
+
316
+ ctypedef enum cusolverdxTransposeMode "cusolverdxTransposeMode":
317
+ CUSOLVERDX_TRANSPOSE_MODE_NON_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_NON_TRANSPOSED" = 0
318
+ CUSOLVERDX_TRANSPOSE_MODE_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_TRANSPOSED" = 1
319
+ CUSOLVERDX_TRANSPOSE_MODE_CONJ_TRANSPOSED "CUSOLVERDX_TRANSPOSE_MODE_CONJ_TRANSPOSED" = 2
320
+
321
+ ctypedef enum cusolverdxFillMode "cusolverdxFillMode":
322
+ CUSOLVERDX_FILL_MODE_UPPER "CUSOLVERDX_FILL_MODE_UPPER" = 0
323
+ CUSOLVERDX_FILL_MODE_LOWER "CUSOLVERDX_FILL_MODE_LOWER" = 1
324
+
325
+ ctypedef enum cusolverdxSide "cusolverdxSide":
326
+ CUSOLVERDX_SIDE_LEFT "CUSOLVERDX_SIDE_LEFT" = 0
327
+ CUSOLVERDX_SIDE_RIGHT "CUSOLVERDX_SIDE_RIGHT" = 1
328
+
329
+ ctypedef enum cusolverdxDiag "cusolverdxDiag":
330
+ CUSOLVERDX_DIAG_UNIT "CUSOLVERDX_DIAG_UNIT" = 0
331
+ CUSOLVERDX_DIAG_NON_UNIT "CUSOLVERDX_DIAG_NON_UNIT" = 1
332
+
333
+ ctypedef enum cusolverdxOperatorType "cusolverdxOperatorType":
334
+ CUSOLVERDX_OPERATOR_SIZE "CUSOLVERDX_OPERATOR_SIZE" = 0
335
+ CUSOLVERDX_OPERATOR_TYPE "CUSOLVERDX_OPERATOR_TYPE" = 1
336
+ CUSOLVERDX_OPERATOR_PRECISION "CUSOLVERDX_OPERATOR_PRECISION" = 2
337
+ CUSOLVERDX_OPERATOR_SM "CUSOLVERDX_OPERATOR_SM" = 3
338
+ CUSOLVERDX_OPERATOR_EXECUTION "CUSOLVERDX_OPERATOR_EXECUTION" = 4
339
+ CUSOLVERDX_OPERATOR_BLOCK_DIM "CUSOLVERDX_OPERATOR_BLOCK_DIM" = 5
340
+ CUSOLVERDX_OPERATOR_API "CUSOLVERDX_OPERATOR_API" = 6
341
+ CUSOLVERDX_OPERATOR_FUNCTION "CUSOLVERDX_OPERATOR_FUNCTION" = 7
342
+ CUSOLVERDX_OPERATOR_ARRANGEMENT "CUSOLVERDX_OPERATOR_ARRANGEMENT" = 8
343
+ CUSOLVERDX_OPERATOR_FILL_MODE "CUSOLVERDX_OPERATOR_FILL_MODE" = 9
344
+ CUSOLVERDX_OPERATOR_SIDE "CUSOLVERDX_OPERATOR_SIDE" = 10
345
+ CUSOLVERDX_OPERATOR_DIAG "CUSOLVERDX_OPERATOR_DIAG" = 11
346
+ CUSOLVERDX_OPERATOR_TRANSPOSE_MODE "CUSOLVERDX_OPERATOR_TRANSPOSE_MODE" = 12
347
+ CUSOLVERDX_OPERATOR_LEADING_DIMENSION "CUSOLVERDX_OPERATOR_LEADING_DIMENSION" = 13
348
+ CUSOLVERDX_OPERATOR_BATCHES_PER_BLOCK "CUSOLVERDX_OPERATOR_BATCHES_PER_BLOCK" = 14
349
+ CUSOLVERDX_OPERATOR_JOB "CUSOLVERDX_OPERATOR_JOB" = 15
350
+
351
+ ctypedef enum cusolverdxTraitType "cusolverdxTraitType":
352
+ CUSOLVERDX_TRAIT_SHARED_MEMORY_SIZE "CUSOLVERDX_TRAIT_SHARED_MEMORY_SIZE" = 1
353
+ CUSOLVERDX_TRAIT_SYMBOL_NAME "CUSOLVERDX_TRAIT_SYMBOL_NAME" = 2
354
+ CUSOLVERDX_TRAIT_BLOCK_DIM "CUSOLVERDX_TRAIT_BLOCK_DIM" = 3
355
+ CUSOLVERDX_TRAIT_SUGGESTED_BLOCK_DIM "CUSOLVERDX_TRAIT_SUGGESTED_BLOCK_DIM" = 5
356
+ CUSOLVERDX_TRAIT_SUGGESTED_BATCHES_PER_BLOCK "CUSOLVERDX_TRAIT_SUGGESTED_BATCHES_PER_BLOCK" = 6
357
+ CUSOLVERDX_TRAIT_ARRANGEMENT "CUSOLVERDX_TRAIT_ARRANGEMENT" = 7
358
+ CUSOLVERDX_TRAIT_BATCHES_PER_BLOCK "CUSOLVERDX_TRAIT_BATCHES_PER_BLOCK" = 8
359
+ CUSOLVERDX_TRAIT_TYPE "CUSOLVERDX_TRAIT_TYPE" = 9
360
+ CUSOLVERDX_TRAIT_LEADING_DIMENSION "CUSOLVERDX_TRAIT_LEADING_DIMENSION" = 10
361
+ CUSOLVERDX_TRAIT_VALUE_TYPE "CUSOLVERDX_TRAIT_VALUE_TYPE" = 11
362
+ CUSOLVERDX_TRAIT_WORKSPACE_SIZE "CUSOLVERDX_TRAIT_WORKSPACE_SIZE" = 12
363
+ CUSOLVERDX_TRAIT_SUGGESTED_LEADING_DIMENSION "CUSOLVERDX_TRAIT_SUGGESTED_LEADING_DIMENSION" = 4
364
+
365
+ ctypedef enum curanddxDistribution "curanddxDistribution":
366
+ CURANDDX_DISTRIBUTION_UNIFORM_BITS "CURANDDX_DISTRIBUTION_UNIFORM_BITS" = 0
367
+ CURANDDX_DISTRIBUTION_UNIFORM "CURANDDX_DISTRIBUTION_UNIFORM" = 1
368
+ CURANDDX_DISTRIBUTION_NORMAL "CURANDDX_DISTRIBUTION_NORMAL" = 2
369
+ CURANDDX_DISTRIBUTION_LOG_NORMAL "CURANDDX_DISTRIBUTION_LOG_NORMAL" = 3
370
+ CURANDDX_DISTRIBUTION_POISSON "CURANDDX_DISTRIBUTION_POISSON" = 4
371
+
372
+ ctypedef enum curanddxNormalMethod "curanddxNormalMethod":
373
+ CURANDDX_NORMAL_METHOD_ICDF "CURANDDX_NORMAL_METHOD_ICDF" = 0
374
+ CURANDDX_NORMAL_METHOD_BOX_MULLER "CURANDDX_NORMAL_METHOD_BOX_MULLER" = 1
375
+
376
+ ctypedef enum curanddxGenerateMethod "curanddxGenerateMethod":
377
+ CURANDDX_GENERATE_METHOD_SINGLE "CURANDDX_GENERATE_METHOD_SINGLE" = 0
378
+ CURANDDX_GENERATE_METHOD_PAIR "CURANDDX_GENERATE_METHOD_PAIR" = 1
379
+ CURANDDX_GENERATE_METHOD_QUAD "CURANDDX_GENERATE_METHOD_QUAD" = 2
380
+
381
+ ctypedef enum curanddxGenerator "curanddxGenerator":
382
+ CURANDDX_GENERATOR_XORWOW "CURANDDX_GENERATOR_XORWOW" = 0
383
+ CURANDDX_GENERATOR_MRG32K3A "CURANDDX_GENERATOR_MRG32K3A" = 1
384
+ CURANDDX_GENERATOR_PHILOX4_32 "CURANDDX_GENERATOR_PHILOX4_32" = 2
385
+ CURANDDX_GENERATOR_PCG "CURANDDX_GENERATOR_PCG" = 3
386
+ CURANDDX_GENERATOR_SOBOL32 "CURANDDX_GENERATOR_SOBOL32" = 4
387
+ CURANDDX_GENERATOR_SCRAMBLED_SOBOL32 "CURANDDX_GENERATOR_SCRAMBLED_SOBOL32" = 5
388
+ CURANDDX_GENERATOR_SOBOL64 "CURANDDX_GENERATOR_SOBOL64" = 6
389
+ CURANDDX_GENERATOR_SCRAMBLED_SOBOL64 "CURANDDX_GENERATOR_SCRAMBLED_SOBOL64" = 7
390
+
391
+ ctypedef enum curanddxOperatorType "curanddxOperatorType":
392
+ CURANDDX_OPERATOR_GENERATOR "CURANDDX_OPERATOR_GENERATOR" = 0
393
+ CURANDDX_OPERATOR_PHILOX_ROUNDS "CURANDDX_OPERATOR_PHILOX_ROUNDS" = 1
394
+ CURANDDX_OPERATOR_SM "CURANDDX_OPERATOR_SM" = 2
395
+ CURANDDX_OPERATOR_EXECUTION "CURANDDX_OPERATOR_EXECUTION" = 3
396
+ CURANDDX_OPERATOR_DISTRIBUTION "CURANDDX_OPERATOR_DISTRIBUTION" = 4
397
+ CURANDDX_OPERATOR_OUTPUT_TYPE "CURANDDX_OPERATOR_OUTPUT_TYPE" = 5
398
+ CURANDDX_OPERATOR_GENERATE_METHOD "CURANDDX_OPERATOR_GENERATE_METHOD" = 6
399
+ CURANDDX_OPERATOR_NORMAL_METHOD "CURANDDX_OPERATOR_NORMAL_METHOD" = 7
400
+ CURANDDX_OPERATOR_DEVICE_FUNCTIONS "CURANDDX_OPERATOR_DEVICE_FUNCTIONS" = 8
401
+ CURANDDX_OPERATOR_DISTRIBUTION_PARAMETERS "CURANDDX_OPERATOR_DISTRIBUTION_PARAMETERS" = 9
402
+
403
+ ctypedef enum curanddxDeviceFunctionType "curanddxDeviceFunctionType":
404
+ CURANDDX_DEVICE_FUNCTION_GENERATE "CURANDDX_DEVICE_FUNCTION_GENERATE" = (1 << 1)
405
+ CURANDDX_DEVICE_FUNCTION_INIT_STATE "CURANDDX_DEVICE_FUNCTION_INIT_STATE" = (1 << 2)
406
+ CURANDDX_DEVICE_FUNCTION_DESTROY_STATE "CURANDDX_DEVICE_FUNCTION_DESTROY_STATE" = (1 << 3)
407
+ CURANDDX_DEVICE_FUNCTION_SKIP_OFFSET "CURANDDX_DEVICE_FUNCTION_SKIP_OFFSET" = (1 << 4)
408
+ CURANDDX_DEVICE_FUNCTION_SKIP_SUBSEQUENCE "CURANDDX_DEVICE_FUNCTION_SKIP_SUBSEQUENCE" = (1 << 5)
409
+ CURANDDX_DEVICE_FUNCTION_SKIP_SEQUENCE "CURANDDX_DEVICE_FUNCTION_SKIP_SEQUENCE" = (1 << 6)
410
+
411
+ ctypedef enum curanddxTraitType "curanddxTraitType":
412
+ CURANDDX_TRAIT_SYMBOL_GENERATE_NAME "CURANDDX_TRAIT_SYMBOL_GENERATE_NAME" = 0
413
+ CURANDDX_TRAIT_SYMBOL_INIT_STATE_NAME "CURANDDX_TRAIT_SYMBOL_INIT_STATE_NAME" = 1
414
+ CURANDDX_TRAIT_SYMBOL_DESTROY_STATE_NAME "CURANDDX_TRAIT_SYMBOL_DESTROY_STATE_NAME" = 2
415
+ CURANDDX_TRAIT_SYMBOL_SKIP_OFFSET_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_OFFSET_NAME" = 3
416
+ CURANDDX_TRAIT_SYMBOL_SKIP_SUBSEQUENCE_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_SUBSEQUENCE_NAME" = 4
417
+ CURANDDX_TRAIT_SYMBOL_SKIP_SEQUENCE_NAME "CURANDDX_TRAIT_SYMBOL_SKIP_SEQUENCE_NAME" = 5
418
+ CURANDDX_TRAIT_STATE_SIZE "CURANDDX_TRAIT_STATE_SIZE" = 6
419
+ CURANDDX_TRAIT_STATE_ALIGNMENT "CURANDDX_TRAIT_STATE_ALIGNMENT" = 7
420
+
421
+ ctypedef enum commondxCodeType "commondxCodeType":
422
+ COMMONDX_CODE_TYPE_LTO "COMMONDX_CODE_TYPE_LTO" = 0
423
+ COMMONDX_CODE_TYPE_PTX "COMMONDX_CODE_TYPE_PTX" = 1
424
+
425
+ ctypedef enum cusolverdxJob "cusolverdxJob":
426
+ CUSOLVERDX_JOB_NO_VECTORS "CUSOLVERDX_JOB_NO_VECTORS" = 0
427
+ CUSOLVERDX_JOB_ALL_VECTORS "CUSOLVERDX_JOB_ALL_VECTORS" = 1
428
+ CUSOLVERDX_JOB_MULTIPLY_VECTORS "CUSOLVERDX_JOB_MULTIPLY_VECTORS" = 2
429
+ CUSOLVERDX_JOB_OVERWRITE_VECTORS "CUSOLVERDX_JOB_OVERWRITE_VECTORS" = 3
430
+
431
+ # types
432
+ ctypedef long long int commondxCode 'commondxCode'
433
+ ctypedef long long int cublasdxDescriptor 'cublasdxDescriptor'
434
+ ctypedef long long int cublasdxTensor 'cublasdxTensor'
435
+ ctypedef long long int cublasdxDeviceFunction 'cublasdxDeviceFunction'
436
+ ctypedef long long int cublasdxPipeline 'cublasdxPipeline'
437
+ ctypedef long long int cufftdxDescriptor 'cufftdxDescriptor'
438
+ ctypedef long long int cusolverdxDescriptor 'cusolverdxDescriptor'
439
+ ctypedef long long int curanddxDescriptor 'curanddxDescriptor'
440
+
441
+
442
+ ###############################################################################
443
+ # Functions
444
+ ###############################################################################
445
+
446
+ cdef commondxStatusType commondxCreateCode(commondxCode* code) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
447
+ cdef commondxStatusType commondxSetCodeOptionInt64(commondxCode code, commondxOption option, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
448
+ cdef commondxStatusType commondxSetCodeOptionInt64s(commondxCode code, commondxOption option, size_t count, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
449
+ cdef commondxStatusType commondxSetCodeOptionStr(commondxCode code, commondxOption option, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
450
+ cdef commondxStatusType commondxGetCodeOptionInt64(commondxCode code, commondxOption option, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
451
+ cdef commondxStatusType commondxGetCodeOptionsInt64s(commondxCode code, commondxOption option, size_t size, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
452
+ cdef commondxStatusType commondxGetCodeLTOIRSize(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
453
+ cdef commondxStatusType commondxGetCodeLTOIR(commondxCode code, size_t size, void* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
454
+ cdef commondxStatusType commondxGetCodeNumLTOIRs(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
455
+ cdef commondxStatusType commondxGetCodeLTOIRSizes(commondxCode code, size_t size, size_t* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
456
+ cdef commondxStatusType commondxGetCodeLTOIRs(commondxCode code, size_t size, void** out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
457
+ cdef commondxStatusType commondxDestroyCode(commondxCode code) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
458
+ cdef const char* commondxStatusToStr(commondxStatusType status) except?NULL nogil
459
+ cdef commondxStatusType commondxGetLastErrorStrSize(size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
460
+ cdef commondxStatusType commondxGetLastErrorStr(commondxStatusType* code, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
461
+ cdef commondxStatusType mathdxGetVersion(int* version) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
462
+ cdef commondxStatusType mathdxGetVersionEx(int* major, int* minor, int* patch) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
463
+ cdef commondxStatusType cublasdxCreateDescriptor(cublasdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
464
+ cdef commondxStatusType cublasdxSetOptionStr(cublasdxDescriptor handle, commondxOption option, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
465
+ cdef commondxStatusType cublasdxSetOperatorInt64(cublasdxDescriptor handle, cublasdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
466
+ cdef commondxStatusType cublasdxSetOperatorInt64s(cublasdxDescriptor handle, cublasdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
467
+ cdef commondxStatusType cublasdxCreateTensor(cublasdxDescriptor handle, cublasdxTensorType tensor_type, cublasdxTensor* tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
468
+ cdef commondxStatusType cublasdxCreateTensorStrided(cublasdxMemorySpace memory_space, commondxValueType value_type, void* ptr, long long int rank, long long int* shape, long long int* stride, cublasdxTensor* tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
469
+ cdef commondxStatusType cublasdxMakeTensorLike(cublasdxTensor input, commondxValueType value_type, cublasdxTensor* output) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
470
+ cdef commondxStatusType cublasdxDestroyTensor(cublasdxTensor tensor) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
471
+ cdef commondxStatusType cublasdxDestroyPipeline(cublasdxPipeline pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
472
+ cdef commondxStatusType cublasdxCreateDevicePipeline(cublasdxDescriptor handle, cublasdxDevicePipelineType device_pipeline_type, long long int pipeline_depth, cublasdxBlockSizeStrategy block_size_strategy, cublasdxTensor tensor_a, cublasdxTensor tensor_b, cublasdxPipeline* device_pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
473
+ cdef commondxStatusType cublasdxCreateTilePipeline(cublasdxDescriptor handle, cublasdxTilePipelineType tile_pipeline_type, cublasdxPipeline device_pipeline, cublasdxPipeline* tile_pipeline) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
474
+ cdef commondxStatusType cublasdxSetTensorOptionInt64(cublasdxTensor tensor, cublasdxTensorOption option, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
475
+ cdef commondxStatusType cublasdxFinalizeTensors(size_t count, const cublasdxTensor* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
476
+ cdef commondxStatusType cublasdxFinalizePipelines(size_t count, const cublasdxPipeline* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
477
+ cdef commondxStatusType cublasdxFinalize(size_t countTensors, const cublasdxTensor* tensors, size_t countPipelines, const cublasdxPipeline* pipelines) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
478
+ cdef commondxStatusType cublasdxGetTensorTraitInt64(cublasdxTensor tensor, cublasdxTensorTrait trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
479
+ cdef commondxStatusType cublasdxGetPipelineTraitInt64(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
480
+ cdef commondxStatusType cublasdxGetPipelineTraitInt64s(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
481
+ cdef commondxStatusType cublasdxGetTensorTraitStrSize(cublasdxTensor tensor, cublasdxTensorTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
482
+ cdef commondxStatusType cublasdxGetPipelineTraitStrSize(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
483
+ cdef commondxStatusType cublasdxGetTensorTraitStr(cublasdxTensor tensor, cublasdxTensorTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
484
+ cdef commondxStatusType cublasdxGetPipelineTraitStr(cublasdxPipeline pipeline, cublasdxPipelineTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
485
+ cdef commondxStatusType cublasdxCreateDeviceFunction(cublasdxDescriptor handle, cublasdxDeviceFunctionType device_function_type, size_t count, const cublasdxTensor* array, cublasdxDeviceFunction* device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
486
+ cdef commondxStatusType cublasdxDestroyDeviceFunction(cublasdxDeviceFunction device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
487
+ cdef commondxStatusType cublasdxCreateDeviceFunctionWithPipelines(cublasdxDescriptor handle, cublasdxDeviceFunctionType device_function_type, size_t tensor_count, const cublasdxTensor* tensors, size_t pipeline_count, const cublasdxPipeline* pipelines, cublasdxDeviceFunction* device_function) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
488
+ cdef commondxStatusType cublasdxFinalizeDeviceFunctions(commondxCode code, size_t count, const cublasdxDeviceFunction* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
489
+ cdef commondxStatusType cublasdxGetDeviceFunctionTraitStrSize(cublasdxDeviceFunction device_function, cublasdxDeviceFunctionTrait trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
490
+ cdef commondxStatusType cublasdxGetDeviceFunctionTraitStr(cublasdxDeviceFunction device_function, cublasdxDeviceFunctionTrait trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
491
+ cdef commondxStatusType cublasdxGetLTOIRSize(cublasdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
492
+ cdef commondxStatusType cublasdxGetLTOIR(cublasdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
493
+ cdef commondxStatusType cublasdxGetTraitStrSize(cublasdxDescriptor handle, cublasdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
494
+ cdef commondxStatusType cublasdxGetTraitStr(cublasdxDescriptor handle, cublasdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
495
+ cdef commondxStatusType cublasdxGetTraitInt64(cublasdxDescriptor handle, cublasdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
496
+ cdef commondxStatusType cublasdxGetTraitInt64s(cublasdxDescriptor handle, cublasdxTraitType trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
497
+ cdef const char* cublasdxOperatorTypeToStr(cublasdxOperatorType op) except?NULL nogil
498
+ cdef const char* cublasdxTraitTypeToStr(cublasdxTraitType trait) except?NULL nogil
499
+ cdef commondxStatusType cublasdxFinalizeCode(commondxCode code, cublasdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
500
+ cdef commondxStatusType cublasdxDestroyDescriptor(cublasdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
501
+ cdef commondxStatusType cufftdxCreateDescriptor(cufftdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
502
+ cdef commondxStatusType cufftdxSetOptionStr(cufftdxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
503
+ cdef commondxStatusType cufftdxGetKnobInt64Size(cufftdxDescriptor handle, size_t num_knobs, cufftdxKnobType* knobs_ptr, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
504
+ cdef commondxStatusType cufftdxGetKnobInt64s(cufftdxDescriptor handle, size_t num_knobs, cufftdxKnobType* knobs_ptr, size_t size, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
505
+ cdef commondxStatusType cufftdxSetOperatorInt64(cufftdxDescriptor handle, cufftdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
506
+ cdef commondxStatusType cufftdxSetOperatorInt64s(cufftdxDescriptor handle, cufftdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
507
+ cdef commondxStatusType cufftdxGetLTOIRSize(cufftdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
508
+ cdef commondxStatusType cufftdxGetLTOIR(cufftdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
509
+ cdef commondxStatusType cufftdxGetTraitStrSize(cufftdxDescriptor handle, cufftdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
510
+ cdef commondxStatusType cufftdxGetTraitStr(cufftdxDescriptor handle, cufftdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
511
+ cdef commondxStatusType cufftdxGetTraitInt64(cufftdxDescriptor handle, cufftdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
512
+ cdef commondxStatusType cufftdxGetTraitInt64s(cufftdxDescriptor handle, cufftdxTraitType trait, size_t count, long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
513
+ cdef commondxStatusType cufftdxGetTraitCommondxDataType(cufftdxDescriptor handle, cufftdxTraitType trait, commondxValueType* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
514
+ cdef commondxStatusType cufftdxFinalizeCode(commondxCode code, cufftdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
515
+ cdef commondxStatusType cufftdxDestroyDescriptor(cufftdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
516
+ cdef const char* cufftdxOperatorTypeToStr(cufftdxOperatorType op) except?NULL nogil
517
+ cdef const char* cufftdxTraitTypeToStr(cufftdxTraitType op) except?NULL nogil
518
+ cdef commondxStatusType cusolverdxCreateDescriptor(cusolverdxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
519
+ cdef commondxStatusType cusolverdxSetOptionStr(cusolverdxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
520
+ cdef commondxStatusType cusolverdxSetOperatorInt64(cusolverdxDescriptor handle, cusolverdxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
521
+ cdef commondxStatusType cusolverdxSetOperatorInt64s(cusolverdxDescriptor handle, cusolverdxOperatorType op, size_t count, const long long int* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
522
+ cdef commondxStatusType cusolverdxGetLTOIRSize(cusolverdxDescriptor handle, size_t* lto_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
523
+ cdef commondxStatusType cusolverdxGetLTOIR(cusolverdxDescriptor handle, size_t size, void* lto) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
524
+ cdef commondxStatusType cusolverdxGetUniversalFATBINSize(cusolverdxDescriptor handle, size_t* fatbin_size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
525
+ cdef commondxStatusType cusolverdxGetUniversalFATBIN(cusolverdxDescriptor handle, size_t fatbin_size, void* fatbin) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
526
+ cdef commondxStatusType cusolverdxGetTraitStrSize(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
527
+ cdef commondxStatusType cusolverdxGetTraitStr(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
528
+ cdef commondxStatusType cusolverdxGetTraitInt64(cusolverdxDescriptor handle, cusolverdxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
529
+ cdef commondxStatusType cusolverdxGetTraitInt64s(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t count, long long int* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
530
+ cdef commondxStatusType cusolverdxFinalizeCode(commondxCode code, cusolverdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
531
+ cdef commondxStatusType cusolverdxDestroyDescriptor(cusolverdxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
532
+ cdef const char* cusolverdxOperatorTypeToStr(cusolverdxOperatorType op) except?NULL nogil
533
+ cdef const char* cusolverdxTraitTypeToStr(cusolverdxTraitType trait) except?NULL nogil
534
+ cdef commondxStatusType curanddxGetVersion(int* major, int* minor, int* patch) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
535
+ cdef commondxStatusType curanddxCreateDescriptor(curanddxDescriptor* handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
536
+ cdef commondxStatusType curanddxSetOptionStr(curanddxDescriptor handle, commondxOption opt, const char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
537
+ cdef commondxStatusType curanddxSetOperatorInt64(curanddxDescriptor handle, curanddxOperatorType op, long long int value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
538
+ cdef commondxStatusType curanddxSetOperatorDoubles(curanddxDescriptor handle, curanddxOperatorType op, size_t count, double* values) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
539
+ cdef commondxStatusType curanddxGetTraitStrSize(curanddxDescriptor handle, curanddxTraitType trait, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
540
+ cdef commondxStatusType curanddxGetTraitStr(curanddxDescriptor handle, curanddxTraitType trait, size_t size, char* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
541
+ cdef commondxStatusType curanddxGetTraitInt64(curanddxDescriptor handle, curanddxTraitType trait, long long int* value) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
542
+ cdef commondxStatusType curanddxFinalizeCode(commondxCode code, curanddxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
543
+ cdef commondxStatusType curanddxDestroyDescriptor(curanddxDescriptor handle) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
544
+ cdef const char* curanddxOperatorTypeToStr(curanddxOperatorType op) except?NULL nogil
545
+ cdef const char* curanddxDistributionToStr(curanddxDistribution dist) except?NULL nogil
546
+ cdef const char* curanddxGeneratorToStr(curanddxGenerator generator) except?NULL nogil
547
+ cdef const char* curanddxGenerateMethodToStr(curanddxGenerateMethod generate_method) except?NULL nogil
548
+ cdef const char* curanddxNormalMethodToStr(curanddxNormalMethod normal_method) except?NULL nogil
549
+ cdef const char* curanddxTraitTypeToStr(curanddxTraitType trait) except?NULL nogil
550
+ cdef commondxStatusType commondxGetCodePTXSize(commondxCode code, size_t* size) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
551
+ cdef commondxStatusType commondxGetCodePTX(commondxCode code, size_t size, void* out) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil
552
+ cdef commondxStatusType cusolverdxGetTraitCommondxDataTypes(cusolverdxDescriptor handle, cusolverdxTraitType trait, size_t count, commondxValueType* array) except?_COMMONDXSTATUSTYPE_INTERNAL_LOADING_ERROR nogil