nvmath-python 0.9.0__cp314-cp314-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 (309) 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 +383 -0
  7. nvmath/_utils.py +150 -0
  8. nvmath/bindings/__init__.py +51 -0
  9. nvmath/bindings/_internal/__init__.pxd +0 -0
  10. nvmath/bindings/_internal/__init__.py +0 -0
  11. nvmath/bindings/_internal/cublas.cp314-win_amd64.pyd +0 -0
  12. nvmath/bindings/_internal/cublas.pxd +530 -0
  13. nvmath/bindings/_internal/cublasLt.cp314-win_amd64.pyd +0 -0
  14. nvmath/bindings/_internal/cublasLt.pxd +59 -0
  15. nvmath/bindings/_internal/cublasMp.pxd +52 -0
  16. nvmath/bindings/_internal/cudss.cp314-win_amd64.pyd +0 -0
  17. nvmath/bindings/_internal/cudss.pxd +47 -0
  18. nvmath/bindings/_internal/cufft.cp314-win_amd64.pyd +0 -0
  19. nvmath/bindings/_internal/cufft.pxd +70 -0
  20. nvmath/bindings/_internal/cufftMp.pxd +77 -0
  21. nvmath/bindings/_internal/curand.cp314-win_amd64.pyd +0 -0
  22. nvmath/bindings/_internal/curand.pxd +42 -0
  23. nvmath/bindings/_internal/cusolver.cp314-win_amd64.pyd +0 -0
  24. nvmath/bindings/_internal/cusolver.pxd +15 -0
  25. nvmath/bindings/_internal/cusolverDn.cp314-win_amd64.pyd +0 -0
  26. nvmath/bindings/_internal/cusolverDn.pxd +406 -0
  27. nvmath/bindings/_internal/cusparse.cp314-win_amd64.pyd +0 -0
  28. nvmath/bindings/_internal/cusparse.pxd +277 -0
  29. nvmath/bindings/_internal/cusparseLt.cp314-win_amd64.pyd +0 -0
  30. nvmath/bindings/_internal/cusparseLt.pxd +48 -0
  31. nvmath/bindings/_internal/cutensor.cp314-win_amd64.pyd +0 -0
  32. nvmath/bindings/_internal/cutensor.pxd +58 -0
  33. nvmath/bindings/_internal/mathdx.cp314-win_amd64.pyd +0 -0
  34. nvmath/bindings/_internal/mathdx.pxd +116 -0
  35. nvmath/bindings/_internal/nvshmem.pxd +29 -0
  36. nvmath/bindings/_internal/utils.cp314-win_amd64.pyd +0 -0
  37. nvmath/bindings/_internal/utils.pxd +179 -0
  38. nvmath/bindings/_internal/utils.pyi +10 -0
  39. nvmath/bindings/cublas.cp314-win_amd64.pyd +0 -0
  40. nvmath/bindings/cublas.pxd +558 -0
  41. nvmath/bindings/cublas.pyi +746 -0
  42. nvmath/bindings/cublasLt.cp314-win_amd64.pyd +0 -0
  43. nvmath/bindings/cublasLt.pxd +109 -0
  44. nvmath/bindings/cublasLt.pyi +1337 -0
  45. nvmath/bindings/cublasMp.pxd +85 -0
  46. nvmath/bindings/cublasMp.pyi +219 -0
  47. nvmath/bindings/cudss.cp314-win_amd64.pyd +0 -0
  48. nvmath/bindings/cudss.pxd +86 -0
  49. nvmath/bindings/cudss.pyi +282 -0
  50. nvmath/bindings/cufft.cp314-win_amd64.pyd +0 -0
  51. nvmath/bindings/cufft.pxd +118 -0
  52. nvmath/bindings/cufft.pyi +241 -0
  53. nvmath/bindings/cufftMp.pxd +124 -0
  54. nvmath/bindings/cufftMp.pyi +260 -0
  55. nvmath/bindings/curand.cp314-win_amd64.pyd +0 -0
  56. nvmath/bindings/curand.pxd +71 -0
  57. nvmath/bindings/curand.pyi +159 -0
  58. nvmath/bindings/cusolver.cp314-win_amd64.pyd +0 -0
  59. nvmath/bindings/cusolver.pxd +62 -0
  60. nvmath/bindings/cusolver.pyi +242 -0
  61. nvmath/bindings/cusolverDn.cp314-win_amd64.pyd +0 -0
  62. nvmath/bindings/cusolverDn.pxd +430 -0
  63. nvmath/bindings/cusolverDn.pyi +416 -0
  64. nvmath/bindings/cusparse.cp314-win_amd64.pyd +0 -0
  65. nvmath/bindings/cusparse.pxd +338 -0
  66. nvmath/bindings/cusparse.pyi +659 -0
  67. nvmath/bindings/cusparseLt.cp314-win_amd64.pyd +0 -0
  68. nvmath/bindings/cusparseLt.pxd +99 -0
  69. nvmath/bindings/cusparseLt.pyi +198 -0
  70. nvmath/bindings/cutensor.cp314-win_amd64.pyd +0 -0
  71. nvmath/bindings/cutensor.pxd +98 -0
  72. nvmath/bindings/cutensor.pyi +294 -0
  73. nvmath/bindings/cycublas.cp314-win_amd64.pyd +0 -0
  74. nvmath/bindings/cycublas.pxd +684 -0
  75. nvmath/bindings/cycublasLt.cp314-win_amd64.pyd +0 -0
  76. nvmath/bindings/cycublasLt.pxd +1055 -0
  77. nvmath/bindings/cycublasMp.pxd +183 -0
  78. nvmath/bindings/cycudss.cp314-win_amd64.pyd +0 -0
  79. nvmath/bindings/cycudss.pxd +224 -0
  80. nvmath/bindings/cycufft.cp314-win_amd64.pyd +0 -0
  81. nvmath/bindings/cycufft.pxd +326 -0
  82. nvmath/bindings/cycufftMp.pxd +334 -0
  83. nvmath/bindings/cycurand.cp314-win_amd64.pyd +0 -0
  84. nvmath/bindings/cycurand.pxd +146 -0
  85. nvmath/bindings/cycusolver.cp314-win_amd64.pyd +0 -0
  86. nvmath/bindings/cycusolver.pxd +154 -0
  87. nvmath/bindings/cycusolverDn.cp314-win_amd64.pyd +0 -0
  88. nvmath/bindings/cycusolverDn.pxd +446 -0
  89. nvmath/bindings/cycusparse.cp314-win_amd64.pyd +0 -0
  90. nvmath/bindings/cycusparse.pxd +470 -0
  91. nvmath/bindings/cycusparseLt.cp314-win_amd64.pyd +0 -0
  92. nvmath/bindings/cycusparseLt.pxd +150 -0
  93. nvmath/bindings/cycutensor.cp314-win_amd64.pyd +0 -0
  94. nvmath/bindings/cycutensor.pxd +192 -0
  95. nvmath/bindings/cymathdx.cp314-win_amd64.pyd +0 -0
  96. nvmath/bindings/cymathdx.pxd +552 -0
  97. nvmath/bindings/cynvshmem.pxd +126 -0
  98. nvmath/bindings/mathdx.cp314-win_amd64.pyd +0 -0
  99. nvmath/bindings/mathdx.pxd +182 -0
  100. nvmath/bindings/mathdx.pyi +1244 -0
  101. nvmath/bindings/nvpl/__init__.pxd +0 -0
  102. nvmath/bindings/nvpl/__init__.py +13 -0
  103. nvmath/bindings/nvpl/_internal/__init__.pxd +0 -0
  104. nvmath/bindings/nvpl/_internal/__init__.py +0 -0
  105. nvmath/bindings/nvpl/_internal/blas.cp314-win_amd64.pyd +0 -0
  106. nvmath/bindings/nvpl/_internal/blas.pxd +125 -0
  107. nvmath/bindings/nvpl/_internal/fft.cp314-win_amd64.pyd +0 -0
  108. nvmath/bindings/nvpl/_internal/fft.pxd +36 -0
  109. nvmath/bindings/nvpl/blas.cp314-win_amd64.pyd +0 -0
  110. nvmath/bindings/nvpl/blas.pxd +85 -0
  111. nvmath/bindings/nvpl/blas.pyi +122 -0
  112. nvmath/bindings/nvpl/cyblas.cp314-win_amd64.pyd +0 -0
  113. nvmath/bindings/nvpl/cyblas.pxd +166 -0
  114. nvmath/bindings/nvpl/cyfft.cp314-win_amd64.pyd +0 -0
  115. nvmath/bindings/nvpl/cyfft.pxd +92 -0
  116. nvmath/bindings/nvpl/fft.cp314-win_amd64.pyd +0 -0
  117. nvmath/bindings/nvpl/fft.pxd +100 -0
  118. nvmath/bindings/nvpl/fft.pyi +100 -0
  119. nvmath/bindings/nvshmem.pxd +54 -0
  120. nvmath/bindings/nvshmem.pyi +179 -0
  121. nvmath/device/__init__.py +21 -0
  122. nvmath/device/_deprecated.py +33 -0
  123. nvmath/device/common.py +313 -0
  124. nvmath/device/common_backend.py +131 -0
  125. nvmath/device/common_cuda.py +201 -0
  126. nvmath/device/common_numba.py +310 -0
  127. nvmath/device/common_opaque_tensor.py +201 -0
  128. nvmath/device/cublasdx.py +1805 -0
  129. nvmath/device/cublasdx_backend.py +807 -0
  130. nvmath/device/cublasdx_numba.py +1612 -0
  131. nvmath/device/cufftdx.py +510 -0
  132. nvmath/device/cufftdx_backend.py +196 -0
  133. nvmath/device/cufftdx_numba.py +196 -0
  134. nvmath/device/curand_kernel.py +9147 -0
  135. nvmath/device/cusolverdx.py +2690 -0
  136. nvmath/device/cusolverdx_backend.py +440 -0
  137. nvmath/device/cusolverdx_numba.py +1624 -0
  138. nvmath/device/llvm_array.py +29 -0
  139. nvmath/device/random.py +445 -0
  140. nvmath/device/random_helpers.py +23 -0
  141. nvmath/device/random_states.py +187 -0
  142. nvmath/device/types.py +147 -0
  143. nvmath/device/vector_types_numba.py +203 -0
  144. nvmath/distributed/__init__.py +205 -0
  145. nvmath/distributed/_internal/__init__.py +0 -0
  146. nvmath/distributed/_internal/nvshmem.py +302 -0
  147. nvmath/distributed/_internal/tensor_ifc.py +67 -0
  148. nvmath/distributed/_internal/tensor_ifc_cupy.py +62 -0
  149. nvmath/distributed/_internal/tensor_ifc_host_device.py +165 -0
  150. nvmath/distributed/_internal/tensor_ifc_numpy.py +41 -0
  151. nvmath/distributed/_internal/tensor_ifc_torch.py +141 -0
  152. nvmath/distributed/_internal/tensor_wrapper.py +78 -0
  153. nvmath/distributed/_utils.py +167 -0
  154. nvmath/distributed/distribution.py +770 -0
  155. nvmath/distributed/fft/__init__.py +7 -0
  156. nvmath/distributed/fft/_configuration.py +79 -0
  157. nvmath/distributed/fft/fft.py +2801 -0
  158. nvmath/distributed/linalg/__init__.py +12 -0
  159. nvmath/distributed/linalg/_internal/__init__.py +3 -0
  160. nvmath/distributed/linalg/_internal/epilog_protocol.py +496 -0
  161. nvmath/distributed/linalg/_internal/matmul_desc_ifc.py +28 -0
  162. nvmath/distributed/linalg/advanced/__init__.py +8 -0
  163. nvmath/distributed/linalg/advanced/_configuration.py +166 -0
  164. nvmath/distributed/linalg/advanced/matmulmod.py +2908 -0
  165. nvmath/distributed/process_group.py +408 -0
  166. nvmath/distributed/reshape/__init__.py +6 -0
  167. nvmath/distributed/reshape/_configuration.py +39 -0
  168. nvmath/distributed/reshape/reshape.py +1256 -0
  169. nvmath/fft/__init__.py +7 -0
  170. nvmath/fft/_configuration.py +208 -0
  171. nvmath/fft/_exec_utils.py +109 -0
  172. nvmath/fft/_helpers.py +237 -0
  173. nvmath/fft/fft.py +2903 -0
  174. nvmath/internal/__init__.pxd +3 -0
  175. nvmath/internal/__init__.py +10 -0
  176. nvmath/internal/_bindings.cp314-win_amd64.pyd +0 -0
  177. nvmath/internal/_bindings.pxd +12 -0
  178. nvmath/internal/_device_utils.py +54 -0
  179. nvmath/internal/enum_utils.py +142 -0
  180. nvmath/internal/formatters.py +87 -0
  181. nvmath/internal/mem_limit.py +51 -0
  182. nvmath/internal/memory.cp314-win_amd64.pyd +0 -0
  183. nvmath/internal/memory.pxd +10 -0
  184. nvmath/internal/memory.pyi +46 -0
  185. nvmath/internal/ndbuffer/__init__.pxd +3 -0
  186. nvmath/internal/ndbuffer/__init__.py +7 -0
  187. nvmath/internal/ndbuffer/copy_kernel/args.h +34 -0
  188. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/array_view.h +52 -0
  189. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/elementwise.h +68 -0
  190. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/grid_indexer.h +69 -0
  191. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/transposed.h +240 -0
  192. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/type_utils.h +39 -0
  193. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/utils.h +132 -0
  194. nvmath/internal/ndbuffer/copy_kernel/copy_kernel_impl/vec.h +159 -0
  195. nvmath/internal/ndbuffer/copy_kernel/elementwise.h +53 -0
  196. nvmath/internal/ndbuffer/copy_kernel/transposed.h +58 -0
  197. nvmath/internal/ndbuffer/copy_kernel.cp314-win_amd64.pyd +0 -0
  198. nvmath/internal/ndbuffer/copy_kernel.pxd +9 -0
  199. nvmath/internal/ndbuffer/copy_kernel.pyi +9 -0
  200. nvmath/internal/ndbuffer/data_layout.cp314-win_amd64.pyd +0 -0
  201. nvmath/internal/ndbuffer/data_layout.pxd +72 -0
  202. nvmath/internal/ndbuffer/data_layout.pyi +23 -0
  203. nvmath/internal/ndbuffer/jit.cp314-win_amd64.pyd +0 -0
  204. nvmath/internal/ndbuffer/jit.pxd +7 -0
  205. nvmath/internal/ndbuffer/jit.pyi +6 -0
  206. nvmath/internal/ndbuffer/ndbuffer.cp314-win_amd64.pyd +0 -0
  207. nvmath/internal/ndbuffer/ndbuffer.pxd +36 -0
  208. nvmath/internal/ndbuffer/ndbuffer.pyi +41 -0
  209. nvmath/internal/ndbuffer/package_utils.cp314-win_amd64.pyd +0 -0
  210. nvmath/internal/ndbuffer/package_utils.pxd +11 -0
  211. nvmath/internal/ndbuffer/package_utils.pyi +12 -0
  212. nvmath/internal/package_ifc.py +147 -0
  213. nvmath/internal/package_ifc_cuda.py +49 -0
  214. nvmath/internal/package_ifc_cupy.py +49 -0
  215. nvmath/internal/package_ifc_torch.py +31 -0
  216. nvmath/internal/package_wrapper.py +14 -0
  217. nvmath/internal/tensor_ifc.py +194 -0
  218. nvmath/internal/tensor_ifc_cupy.py +241 -0
  219. nvmath/internal/tensor_ifc_ndbuffer.py +153 -0
  220. nvmath/internal/tensor_ifc_numpy.py +183 -0
  221. nvmath/internal/tensor_ifc_torch.py +176 -0
  222. nvmath/internal/tensor_wrapper.py +157 -0
  223. nvmath/internal/typemaps.py +162 -0
  224. nvmath/internal/utils.py +771 -0
  225. nvmath/linalg/__init__.py +46 -0
  226. nvmath/linalg/_internal/__init__.py +3 -0
  227. nvmath/linalg/_internal/algo_cap_ifc.py +82 -0
  228. nvmath/linalg/_internal/algo_config_ifc.py +43 -0
  229. nvmath/linalg/_internal/batch.py +217 -0
  230. nvmath/linalg/_internal/enum_to_tuples.py +64 -0
  231. nvmath/linalg/_internal/epilog_protocol.py +766 -0
  232. nvmath/linalg/_internal/layout.py +624 -0
  233. nvmath/linalg/_internal/matmul_desc_ifc.py +28 -0
  234. nvmath/linalg/_internal/matmul_pref_ifc.py +27 -0
  235. nvmath/linalg/_internal/matrix_layout_ifc.py +26 -0
  236. nvmath/linalg/_internal/typemaps.py +144 -0
  237. nvmath/linalg/_internal/utils.py +134 -0
  238. nvmath/linalg/advanced/__init__.py +8 -0
  239. nvmath/linalg/advanced/_algorithmmod.py +149 -0
  240. nvmath/linalg/advanced/_configuration.py +349 -0
  241. nvmath/linalg/advanced/helpers/__init__.py +5 -0
  242. nvmath/linalg/advanced/helpers/matmul.py +1348 -0
  243. nvmath/linalg/advanced/matmulmod.py +3937 -0
  244. nvmath/linalg/generic/__init__.py +43 -0
  245. nvmath/linalg/generic/_configuration/__init__.py +37 -0
  246. nvmath/linalg/generic/_configuration/layout.py +263 -0
  247. nvmath/linalg/generic/_configuration/match.py +609 -0
  248. nvmath/linalg/generic/_configuration/qualifiers.py +494 -0
  249. nvmath/linalg/generic/_configuration/wrap.py +217 -0
  250. nvmath/linalg/generic/_dtype.py +15 -0
  251. nvmath/linalg/generic/matmulmod.py +1060 -0
  252. nvmath/memory.py +282 -0
  253. nvmath/sparse/__init__.py +38 -0
  254. nvmath/sparse/_internal/__init__.py +21 -0
  255. nvmath/sparse/_internal/common_utils.py +144 -0
  256. nvmath/sparse/_internal/cudss_config_ifc.py +741 -0
  257. nvmath/sparse/_internal/cudss_data_ifc.py +399 -0
  258. nvmath/sparse/_internal/cudss_utils.py +445 -0
  259. nvmath/sparse/_internal/cusparse_utils.py +382 -0
  260. nvmath/sparse/_internal/sparse_bsc_ifc.py +300 -0
  261. nvmath/sparse/_internal/sparse_bsr_ifc.py +303 -0
  262. nvmath/sparse/_internal/sparse_coo_ifc.py +254 -0
  263. nvmath/sparse/_internal/sparse_csc_ifc.py +266 -0
  264. nvmath/sparse/_internal/sparse_csr_ifc.py +266 -0
  265. nvmath/sparse/_internal/sparse_dia_ifc.py +239 -0
  266. nvmath/sparse/_internal/sparse_format_helpers.py +601 -0
  267. nvmath/sparse/_internal/sparse_tensor_ifc.py +121 -0
  268. nvmath/sparse/_internal/sparse_ust_ifc.py +146 -0
  269. nvmath/sparse/_internal/utils.py +56 -0
  270. nvmath/sparse/advanced/__init__.py +7 -0
  271. nvmath/sparse/advanced/_configuration.py +230 -0
  272. nvmath/sparse/advanced/direct_solver.py +1884 -0
  273. nvmath/sparse/generic/__init__.py +7 -0
  274. nvmath/sparse/generic/_configuration.py +129 -0
  275. nvmath/sparse/generic/_helpers.py +137 -0
  276. nvmath/sparse/generic/_thunks.py +21 -0
  277. nvmath/sparse/generic/matmulmod.py +2426 -0
  278. nvmath/sparse/ust/__init__.py +7 -0
  279. nvmath/sparse/ust/_converters.py +378 -0
  280. nvmath/sparse/ust/_drawer.py +552 -0
  281. nvmath/sparse/ust/_emitter.py +553 -0
  282. nvmath/sparse/ust/_jit.py +212 -0
  283. nvmath/sparse/ust/_kernel.py +346 -0
  284. nvmath/sparse/ust/_semiring.py +71 -0
  285. nvmath/sparse/ust/_utils.py +102 -0
  286. nvmath/sparse/ust/interfaces/__init__.py +0 -0
  287. nvmath/sparse/ust/interfaces/torch_interface.py +476 -0
  288. nvmath/sparse/ust/semiring_ops.py +71 -0
  289. nvmath/sparse/ust/tensor.py +1027 -0
  290. nvmath/sparse/ust/tensor_drawer.py +552 -0
  291. nvmath/sparse/ust/tensor_emitter.py +558 -0
  292. nvmath/sparse/ust/tensor_format.py +935 -0
  293. nvmath/sparse/ust/tensor_jit.py +212 -0
  294. nvmath/sparse/ust/tensor_kernel.py +348 -0
  295. nvmath/sparse/ust/tensor_utils.py +380 -0
  296. nvmath/tensor/__init__.py +6 -0
  297. nvmath/tensor/_configuration.py +95 -0
  298. nvmath/tensor/_internal/__init__.py +3 -0
  299. nvmath/tensor/_internal/cutensor_config_ifc.py +247 -0
  300. nvmath/tensor/_internal/cutensor_utils.py +162 -0
  301. nvmath/tensor/_internal/data.py +43 -0
  302. nvmath/tensor/_internal/einsum_parser.py +444 -0
  303. nvmath/tensor/_internal/typemaps.py +96 -0
  304. nvmath/tensor/contract.py +1861 -0
  305. nvmath_python-0.9.0.dist-info/METADATA +127 -0
  306. nvmath_python-0.9.0.dist-info/RECORD +309 -0
  307. nvmath_python-0.9.0.dist-info/WHEEL +5 -0
  308. nvmath_python-0.9.0.dist-info/licenses/LICENSE +177 -0
  309. nvmath_python-0.9.0.dist-info/top_level.txt +2 -0
nvmath/bindings/cycufft.pxd ADDED
@@ -0,0 +1,326 @@
1
+ # Copyright (c) 2024-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 11.0.3 to 13.2.0, generator version 0.3.1.dev1301+g7215ac36e. Do not modify it directly.
6
+ # This layer exposes the C header to Cython as-is.
7
+
8
+ ###############################################################################
9
+ # Types (structs, enums, ...)
10
+ ###############################################################################
11
+
12
+ # enums
13
+ ctypedef enum libFormat_t "libFormat_t":
14
+ LIB_FORMAT_CUFFT "LIB_FORMAT_CUFFT" = 0x0
15
+ LIB_FORMAT_UNDEFINED "LIB_FORMAT_UNDEFINED" = 0x1
16
+
17
+ ctypedef enum cufftResult "cufftResult":
18
+ CUFFT_SUCCESS "CUFFT_SUCCESS" = 0x0
19
+ CUFFT_INVALID_PLAN "CUFFT_INVALID_PLAN" = 0x1
20
+ CUFFT_ALLOC_FAILED "CUFFT_ALLOC_FAILED" = 0x2
21
+ CUFFT_INVALID_TYPE "CUFFT_INVALID_TYPE" = 0x3
22
+ CUFFT_INVALID_VALUE "CUFFT_INVALID_VALUE" = 0x4
23
+ CUFFT_INTERNAL_ERROR "CUFFT_INTERNAL_ERROR" = 0x5
24
+ CUFFT_EXEC_FAILED "CUFFT_EXEC_FAILED" = 0x6
25
+ CUFFT_SETUP_FAILED "CUFFT_SETUP_FAILED" = 0x7
26
+ CUFFT_INVALID_SIZE "CUFFT_INVALID_SIZE" = 0x8
27
+ CUFFT_UNALIGNED_DATA "CUFFT_UNALIGNED_DATA" = 0x9
28
+ CUFFT_INVALID_DEVICE "CUFFT_INVALID_DEVICE" = 0xB
29
+ CUFFT_NO_WORKSPACE "CUFFT_NO_WORKSPACE" = 0xD
30
+ CUFFT_NOT_IMPLEMENTED "CUFFT_NOT_IMPLEMENTED" = 0xE
31
+ CUFFT_NOT_SUPPORTED "CUFFT_NOT_SUPPORTED" = 0x10
32
+ CUFFT_MISSING_DEPENDENCY "CUFFT_MISSING_DEPENDENCY" = 0x11
33
+ CUFFT_NVRTC_FAILURE "CUFFT_NVRTC_FAILURE" = 0x12
34
+ CUFFT_NVJITLINK_FAILURE "CUFFT_NVJITLINK_FAILURE" = 0x13
35
+ CUFFT_NVSHMEM_FAILURE "CUFFT_NVSHMEM_FAILURE" = 0x14
36
+ _CUFFTRESULT_INTERNAL_LOADING_ERROR "_CUFFTRESULT_INTERNAL_LOADING_ERROR" = -42
37
+ CUFFT_INCOMPLETE_PARAMETER_LIST "CUFFT_INCOMPLETE_PARAMETER_LIST" = 0xA
38
+ CUFFT_PARSE_ERROR "CUFFT_PARSE_ERROR" = 0xC
39
+ CUFFT_LICENSE_ERROR "CUFFT_LICENSE_ERROR" = 0x0F
40
+
41
+ ctypedef enum cufftType "cufftType":
42
+ CUFFT_R2C "CUFFT_R2C" = 0x2a
43
+ CUFFT_C2R "CUFFT_C2R" = 0x2c
44
+ CUFFT_C2C "CUFFT_C2C" = 0x29
45
+ CUFFT_D2Z "CUFFT_D2Z" = 0x6a
46
+ CUFFT_Z2D "CUFFT_Z2D" = 0x6c
47
+ CUFFT_Z2Z "CUFFT_Z2Z" = 0x69
48
+
49
+ ctypedef enum cufftCompatibility "cufftCompatibility":
50
+ CUFFT_COMPATIBILITY_FFTW_PADDING "CUFFT_COMPATIBILITY_FFTW_PADDING" = 0x01
51
+
52
+ ctypedef enum cufftXtSubFormat "cufftXtSubFormat":
53
+ CUFFT_XT_FORMAT_INPUT "CUFFT_XT_FORMAT_INPUT" = 0x00
54
+ CUFFT_XT_FORMAT_OUTPUT "CUFFT_XT_FORMAT_OUTPUT" = 0x01
55
+ CUFFT_XT_FORMAT_INPLACE "CUFFT_XT_FORMAT_INPLACE" = 0x02
56
+ CUFFT_XT_FORMAT_INPLACE_SHUFFLED "CUFFT_XT_FORMAT_INPLACE_SHUFFLED" = 0x03
57
+ CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED "CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED" = 0x04
58
+ CUFFT_XT_FORMAT_DISTRIBUTED_INPUT "CUFFT_XT_FORMAT_DISTRIBUTED_INPUT" = 0x05
59
+ CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT "CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT" = 0x06
60
+ CUFFT_FORMAT_UNDEFINED "CUFFT_FORMAT_UNDEFINED" = 0x07
61
+
62
+ ctypedef enum cufftXtCopyType "cufftXtCopyType":
63
+ CUFFT_COPY_HOST_TO_DEVICE "CUFFT_COPY_HOST_TO_DEVICE" = 0x00
64
+ CUFFT_COPY_DEVICE_TO_HOST "CUFFT_COPY_DEVICE_TO_HOST" = 0x01
65
+ CUFFT_COPY_DEVICE_TO_DEVICE "CUFFT_COPY_DEVICE_TO_DEVICE" = 0x02
66
+ CUFFT_COPY_UNDEFINED "CUFFT_COPY_UNDEFINED" = 0x03
67
+
68
+ ctypedef enum cufftXtQueryType "cufftXtQueryType":
69
+ CUFFT_QUERY_1D_FACTORS "CUFFT_QUERY_1D_FACTORS" = 0x00
70
+ CUFFT_QUERY_UNDEFINED "CUFFT_QUERY_UNDEFINED" = 0x01
71
+
72
+ ctypedef enum cufftXtWorkAreaPolicy "cufftXtWorkAreaPolicy":
73
+ CUFFT_WORKAREA_MINIMAL "CUFFT_WORKAREA_MINIMAL" = 0
74
+ CUFFT_WORKAREA_USER "CUFFT_WORKAREA_USER" = 1
75
+ CUFFT_WORKAREA_PERFORMANCE "CUFFT_WORKAREA_PERFORMANCE" = 2
76
+
77
+ ctypedef enum cufftXtCallbackType "cufftXtCallbackType":
78
+ CUFFT_CB_LD_COMPLEX "CUFFT_CB_LD_COMPLEX" = 0x0
79
+ CUFFT_CB_LD_COMPLEX_DOUBLE "CUFFT_CB_LD_COMPLEX_DOUBLE" = 0x1
80
+ CUFFT_CB_LD_REAL "CUFFT_CB_LD_REAL" = 0x2
81
+ CUFFT_CB_LD_REAL_DOUBLE "CUFFT_CB_LD_REAL_DOUBLE" = 0x3
82
+ CUFFT_CB_ST_COMPLEX "CUFFT_CB_ST_COMPLEX" = 0x4
83
+ CUFFT_CB_ST_COMPLEX_DOUBLE "CUFFT_CB_ST_COMPLEX_DOUBLE" = 0x5
84
+ CUFFT_CB_ST_REAL "CUFFT_CB_ST_REAL" = 0x6
85
+ CUFFT_CB_ST_REAL_DOUBLE "CUFFT_CB_ST_REAL_DOUBLE" = 0x7
86
+ CUFFT_CB_UNDEFINED "CUFFT_CB_UNDEFINED" = 0x8
87
+
88
+ ctypedef enum cufftProperty "cufftProperty":
89
+ NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT "NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT" = 0x1
90
+ NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS "NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS" = 0x2
91
+
92
+
93
+ # types
94
+ cdef extern from *:
95
+ """
96
+ #include <driver_types.h>
97
+ #include <library_types.h>
98
+ #include <cuComplex.h>
99
+
100
+ #define MAX_CUDA_DESCRIPTOR_GPUS 64
101
+ #define CUFFT_FORWARD -1
102
+ #define CUFFT_INVERSE 1
103
+ """
104
+ ctypedef void* cudaStream_t 'cudaStream_t'
105
+ ctypedef int cudaDataType_t 'cudaDataType_t'
106
+ ctypedef int cudaDataType 'cudaDataType'
107
+ ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
108
+ ctypedef int libraryPropertyType 'libraryPropertyType'
109
+
110
+ ctypedef struct cuComplex:
111
+ pass
112
+ ctypedef struct cuDoubleComplex:
113
+ pass
114
+
115
+ cdef const int MAX_CUDA_DESCRIPTOR_GPUS
116
+ cdef const int cuFFTFORWARD
117
+ cdef const int cuFFTINVERSE
118
+
119
+
120
+ ctypedef float cufftReal 'cufftReal'
121
+ ctypedef double cufftDoubleReal 'cufftDoubleReal'
122
+ ctypedef cuComplex cufftComplex 'cufftComplex'
123
+ ctypedef cuDoubleComplex cufftDoubleComplex 'cufftDoubleComplex'
124
+ ctypedef int cufftHandle 'cufftHandle'
125
+ ctypedef struct cufftXt1dFactors 'cufftXt1dFactors':
126
+ long long int size
127
+ long long int stringCount
128
+ long long int stringLength
129
+ long long int substringLength
130
+ long long int factor1
131
+ long long int factor2
132
+ long long int stringMask
133
+ long long int substringMask
134
+ long long int factor1Mask
135
+ long long int factor2Mask
136
+ int stringShift
137
+ int substringShift
138
+ int factor1Shift
139
+ int factor2Shift
140
+
141
+ ctypedef cufftComplex (*cufftCallbackLoadC 'cufftCallbackLoadC')(
142
+ void* dataIn,
143
+ size_t offset,
144
+ void* callerInfo,
145
+ void* sharedPointer
146
+ )
147
+ ctypedef cufftDoubleComplex (*cufftCallbackLoadZ 'cufftCallbackLoadZ')(
148
+ void* dataIn,
149
+ size_t offset,
150
+ void* callerInfo,
151
+ void* sharedPointer
152
+ )
153
+ ctypedef cufftReal (*cufftCallbackLoadR 'cufftCallbackLoadR')(
154
+ void* dataIn,
155
+ size_t offset,
156
+ void* callerInfo,
157
+ void* sharedPointer
158
+ )
159
+ ctypedef cufftDoubleReal (*cufftCallbackLoadD 'cufftCallbackLoadD')(
160
+ void* dataIn,
161
+ size_t offset,
162
+ void* callerInfo,
163
+ void* sharedPointer
164
+ )
165
+ ctypedef cufftComplex (*cufftJITCallbackLoadC 'cufftJITCallbackLoadC')(
166
+ void* dataIn,
167
+ unsigned long long offset,
168
+ void* callerInfo,
169
+ void* sharedPointer
170
+ )
171
+ ctypedef cufftDoubleComplex (*cufftJITCallbackLoadZ 'cufftJITCallbackLoadZ')(
172
+ void* dataIn,
173
+ unsigned long long offset,
174
+ void* callerInfo,
175
+ void* sharedPointer
176
+ )
177
+ ctypedef cufftReal (*cufftJITCallbackLoadR 'cufftJITCallbackLoadR')(
178
+ void* dataIn,
179
+ unsigned long long offset,
180
+ void* callerInfo,
181
+ void* sharedPointer
182
+ )
183
+ ctypedef cufftDoubleReal (*cufftJITCallbackLoadD 'cufftJITCallbackLoadD')(
184
+ void* dataIn,
185
+ unsigned long long offset,
186
+ void* callerInfo,
187
+ void* sharedPointer
188
+ )
189
+ ctypedef void (*cufftCallbackStoreR 'cufftCallbackStoreR')(
190
+ void* dataOut,
191
+ size_t offset,
192
+ cufftReal element,
193
+ void* callerInfo,
194
+ void* sharedPointer
195
+ )
196
+ ctypedef void (*cufftJITCallbackStoreR 'cufftJITCallbackStoreR')(
197
+ void* dataOut,
198
+ unsigned long long offset,
199
+ cufftReal element,
200
+ void* callerInfo,
201
+ void* sharedPointer
202
+ )
203
+ ctypedef void (*cufftCallbackStoreD 'cufftCallbackStoreD')(
204
+ void* dataOut,
205
+ size_t offset,
206
+ cufftDoubleReal element,
207
+ void* callerInfo,
208
+ void* sharedPointer
209
+ )
210
+ ctypedef void (*cufftJITCallbackStoreD 'cufftJITCallbackStoreD')(
211
+ void* dataOut,
212
+ unsigned long long offset,
213
+ cufftDoubleReal element,
214
+ void* callerInfo,
215
+ void* sharedPointer
216
+ )
217
+ ctypedef void (*cufftCallbackStoreC 'cufftCallbackStoreC')(
218
+ void* dataOut,
219
+ size_t offset,
220
+ cufftComplex element,
221
+ void* callerInfo,
222
+ void* sharedPointer
223
+ )
224
+ ctypedef void (*cufftJITCallbackStoreC 'cufftJITCallbackStoreC')(
225
+ void* dataOut,
226
+ unsigned long long offset,
227
+ cufftComplex element,
228
+ void* callerInfo,
229
+ void* sharedPointer
230
+ )
231
+ ctypedef void (*cufftCallbackStoreZ 'cufftCallbackStoreZ')(
232
+ void* dataOut,
233
+ size_t offset,
234
+ cufftDoubleComplex element,
235
+ void* callerInfo,
236
+ void* sharedPointer
237
+ )
238
+ ctypedef void (*cufftJITCallbackStoreZ 'cufftJITCallbackStoreZ')(
239
+ void* dataOut,
240
+ unsigned long long offset,
241
+ cufftDoubleComplex element,
242
+ void* callerInfo,
243
+ void* sharedPointer
244
+ )
245
+
246
+ ctypedef struct cudaXtDesc_t 'cudaXtDesc_t':
247
+ int version "version"
248
+ int nGPUs "nGPUs"
249
+ int GPUs "GPUs" [MAX_CUDA_DESCRIPTOR_GPUS]
250
+ void* data "data" [MAX_CUDA_DESCRIPTOR_GPUS]
251
+ size_t size "size" [MAX_CUDA_DESCRIPTOR_GPUS]
252
+ void* cudaXtState "cudaXtState"
253
+
254
+ ctypedef cudaXtDesc_t cudaXtDesc 'cudaXtDesc'
255
+
256
+ ctypedef struct cudaLibXtDesc_t 'cudaLibXtDesc_t':
257
+ int version "version"
258
+ cudaXtDesc* descriptor "descriptor"
259
+ libFormat_t library "library"
260
+ int subFormat "subFormat"
261
+ void* libDescriptor "libDescriptor"
262
+
263
+ ctypedef cudaLibXtDesc_t cudaLibXtDesc 'cudaLibXtDesc'
264
+
265
+
266
+ ###############################################################################
267
+ # Functions
268
+ ###############################################################################
269
+
270
+ cdef cufftResult cufftPlan1d(cufftHandle* plan, int nx, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
271
+ cdef cufftResult cufftPlan2d(cufftHandle* plan, int nx, int ny, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
272
+ cdef cufftResult cufftPlan3d(cufftHandle* plan, int nx, int ny, int nz, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
273
+ cdef cufftResult cufftPlanMany(cufftHandle* plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
274
+ cdef cufftResult cufftMakePlan1d(cufftHandle plan, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
275
+ cdef cufftResult cufftMakePlan2d(cufftHandle plan, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
276
+ cdef cufftResult cufftMakePlan3d(cufftHandle plan, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
277
+ cdef cufftResult cufftMakePlanMany(cufftHandle plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
278
+ cdef cufftResult cufftMakePlanMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
279
+ cdef cufftResult cufftGetSizeMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
280
+ cdef cufftResult cufftEstimate1d(int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
281
+ cdef cufftResult cufftEstimate2d(int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
282
+ cdef cufftResult cufftEstimate3d(int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
283
+ cdef cufftResult cufftEstimateMany(int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
284
+ cdef cufftResult cufftCreate(cufftHandle* handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
285
+ cdef cufftResult cufftGetSize1d(cufftHandle handle, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
286
+ cdef cufftResult cufftGetSize2d(cufftHandle handle, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
287
+ cdef cufftResult cufftGetSize3d(cufftHandle handle, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
288
+ cdef cufftResult cufftGetSizeMany(cufftHandle handle, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
289
+ cdef cufftResult cufftGetSize(cufftHandle handle, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
290
+ cdef cufftResult cufftSetWorkArea(cufftHandle plan, void* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
291
+ cdef cufftResult cufftSetAutoAllocation(cufftHandle plan, int autoAllocate) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
292
+ cdef cufftResult cufftExecC2C(cufftHandle plan, cufftComplex* idata, cufftComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
293
+ cdef cufftResult cufftExecR2C(cufftHandle plan, cufftReal* idata, cufftComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
294
+ cdef cufftResult cufftExecC2R(cufftHandle plan, cufftComplex* idata, cufftReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
295
+ cdef cufftResult cufftExecZ2Z(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
296
+ cdef cufftResult cufftExecD2Z(cufftHandle plan, cufftDoubleReal* idata, cufftDoubleComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
297
+ cdef cufftResult cufftExecZ2D(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
298
+ cdef cufftResult cufftSetStream(cufftHandle plan, cudaStream_t stream) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
299
+ cdef cufftResult cufftDestroy(cufftHandle plan) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
300
+ cdef cufftResult cufftGetVersion(int* version) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
301
+ cdef cufftResult cufftGetProperty(libraryPropertyType type, int* value) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
302
+ cdef cufftResult cufftXtSetGPUs(cufftHandle handle, int nGPUs, int* whichGPUs) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
303
+ cdef cufftResult cufftXtMalloc(cufftHandle plan, cudaLibXtDesc** descriptor, cufftXtSubFormat format) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
304
+ cdef cufftResult cufftXtMemcpy(cufftHandle plan, void* dstPointer, void* srcPointer, cufftXtCopyType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
305
+ cdef cufftResult cufftXtFree(cudaLibXtDesc* descriptor) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
306
+ cdef cufftResult cufftXtSetWorkArea(cufftHandle plan, void** workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
307
+ cdef cufftResult cufftXtExecDescriptorC2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
308
+ cdef cufftResult cufftXtExecDescriptorR2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
309
+ cdef cufftResult cufftXtExecDescriptorC2R(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
310
+ cdef cufftResult cufftXtExecDescriptorZ2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
311
+ cdef cufftResult cufftXtExecDescriptorD2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
312
+ cdef cufftResult cufftXtExecDescriptorZ2D(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
313
+ cdef cufftResult cufftXtQueryPlan(cufftHandle plan, void* queryStruct, cufftXtQueryType queryType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
314
+ cdef cufftResult cufftXtClearCallback(cufftHandle plan, cufftXtCallbackType cbType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
315
+ cdef cufftResult cufftXtSetCallbackSharedSize(cufftHandle plan, cufftXtCallbackType cbType, size_t sharedSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
316
+ cdef cufftResult cufftXtMakePlanMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
317
+ cdef cufftResult cufftXtGetSizeMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
318
+ cdef cufftResult cufftXtExec(cufftHandle plan, void* input, void* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
319
+ cdef cufftResult cufftXtExecDescriptor(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
320
+ cdef cufftResult cufftXtSetWorkAreaPolicy(cufftHandle plan, cufftXtWorkAreaPolicy policy, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
321
+ cdef cufftResult cufftXtSetJITCallback(cufftHandle plan, const char* lto_callback_symbol_name, const void* lto_callback_fatbin, size_t lto_callback_fatbin_size, cufftXtCallbackType type, void** caller_info) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
322
+ cdef cufftResult cufftXtSetSubformatDefault(cufftHandle plan, cufftXtSubFormat subformat_forward, cufftXtSubFormat subformat_inverse) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
323
+ cdef cufftResult cufftSetPlanPropertyInt64(cufftHandle plan, cufftProperty property, const long long int inputValueInt) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
324
+ cdef cufftResult cufftGetPlanPropertyInt64(cufftHandle plan, cufftProperty property, long long int* returnPtrValue) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
325
+ cdef cufftResult cufftResetPlanProperty(cufftHandle plan, cufftProperty property) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
326
+ cdef cufftResult __cufftXtSetJITCallback_12_7(cufftHandle plan, const char* lto_callback_symbol_name, const void* lto_callback_fatbin, size_t lto_callback_fatbin_size, cufftXtCallbackType type, void** caller_info) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
nvmath/bindings/cycufftMp.pxd ADDED
@@ -0,0 +1,334 @@
1
+ # Copyright (c) 2025-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 11.2.6 to 11.4.0, generator version 0.3.1.dev1303+g031f1197f. Do not modify it directly.
6
+ # This layer exposes the C header to Cython as-is.
7
+
8
+ ###############################################################################
9
+ # Types (structs, enums, ...)
10
+ ###############################################################################
11
+
12
+ # enums
13
+ ctypedef enum libFormat_t "libFormat_t":
14
+ LIB_FORMAT_CUFFT "LIB_FORMAT_CUFFT" = 0x0
15
+ LIB_FORMAT_UNDEFINED "LIB_FORMAT_UNDEFINED" = 0x1
16
+
17
+ ctypedef enum cufftResult "cufftResult":
18
+ CUFFT_SUCCESS "CUFFT_SUCCESS" = 0x0
19
+ CUFFT_INVALID_PLAN "CUFFT_INVALID_PLAN" = 0x1
20
+ CUFFT_ALLOC_FAILED "CUFFT_ALLOC_FAILED" = 0x2
21
+ CUFFT_INVALID_TYPE "CUFFT_INVALID_TYPE" = 0x3
22
+ CUFFT_INVALID_VALUE "CUFFT_INVALID_VALUE" = 0x4
23
+ CUFFT_INTERNAL_ERROR "CUFFT_INTERNAL_ERROR" = 0x5
24
+ CUFFT_EXEC_FAILED "CUFFT_EXEC_FAILED" = 0x6
25
+ CUFFT_SETUP_FAILED "CUFFT_SETUP_FAILED" = 0x7
26
+ CUFFT_INVALID_SIZE "CUFFT_INVALID_SIZE" = 0x8
27
+ CUFFT_UNALIGNED_DATA "CUFFT_UNALIGNED_DATA" = 0x9
28
+ CUFFT_INCOMPLETE_PARAMETER_LIST "CUFFT_INCOMPLETE_PARAMETER_LIST" = 0xA
29
+ CUFFT_INVALID_DEVICE "CUFFT_INVALID_DEVICE" = 0xB
30
+ CUFFT_PARSE_ERROR "CUFFT_PARSE_ERROR" = 0xC
31
+ CUFFT_NO_WORKSPACE "CUFFT_NO_WORKSPACE" = 0xD
32
+ CUFFT_NOT_IMPLEMENTED "CUFFT_NOT_IMPLEMENTED" = 0xE
33
+ CUFFT_LICENSE_ERROR "CUFFT_LICENSE_ERROR" = 0x0F
34
+ CUFFT_NOT_SUPPORTED "CUFFT_NOT_SUPPORTED" = 0x10
35
+ _CUFFTRESULT_INTERNAL_LOADING_ERROR "_CUFFTRESULT_INTERNAL_LOADING_ERROR" = -42
36
+
37
+ ctypedef enum cufftType "cufftType":
38
+ CUFFT_R2C "CUFFT_R2C" = 0x2a
39
+ CUFFT_C2R "CUFFT_C2R" = 0x2c
40
+ CUFFT_C2C "CUFFT_C2C" = 0x29
41
+ CUFFT_D2Z "CUFFT_D2Z" = 0x6a
42
+ CUFFT_Z2D "CUFFT_Z2D" = 0x6c
43
+ CUFFT_Z2Z "CUFFT_Z2Z" = 0x69
44
+
45
+ ctypedef enum cufftCompatibility "cufftCompatibility":
46
+ CUFFT_COMPATIBILITY_FFTW_PADDING "CUFFT_COMPATIBILITY_FFTW_PADDING" = 0x01
47
+
48
+ ctypedef enum cufftProperty "cufftProperty":
49
+ NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT "NVFFT_PLAN_PROPERTY_INT64_PATIENT_JIT" = 0x1
50
+ NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS "NVFFT_PLAN_PROPERTY_INT64_MAX_NUM_HOST_THREADS" = 0x2
51
+
52
+ ctypedef enum cufftXtSubFormat "cufftXtSubFormat":
53
+ CUFFT_XT_FORMAT_INPUT "CUFFT_XT_FORMAT_INPUT" = 0x00
54
+ CUFFT_XT_FORMAT_OUTPUT "CUFFT_XT_FORMAT_OUTPUT" = 0x01
55
+ CUFFT_XT_FORMAT_INPLACE "CUFFT_XT_FORMAT_INPLACE" = 0x02
56
+ CUFFT_XT_FORMAT_INPLACE_SHUFFLED "CUFFT_XT_FORMAT_INPLACE_SHUFFLED" = 0x03
57
+ CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED "CUFFT_XT_FORMAT_1D_INPUT_SHUFFLED" = 0x04
58
+ CUFFT_XT_FORMAT_DISTRIBUTED_INPUT "CUFFT_XT_FORMAT_DISTRIBUTED_INPUT" = 0x05
59
+ CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT "CUFFT_XT_FORMAT_DISTRIBUTED_OUTPUT" = 0x06
60
+ CUFFT_FORMAT_UNDEFINED "CUFFT_FORMAT_UNDEFINED" = 0x07
61
+
62
+ ctypedef enum cufftXtCopyType "cufftXtCopyType":
63
+ CUFFT_COPY_HOST_TO_DEVICE "CUFFT_COPY_HOST_TO_DEVICE" = 0x00
64
+ CUFFT_COPY_DEVICE_TO_HOST "CUFFT_COPY_DEVICE_TO_HOST" = 0x01
65
+ CUFFT_COPY_DEVICE_TO_DEVICE "CUFFT_COPY_DEVICE_TO_DEVICE" = 0x02
66
+ CUFFT_COPY_UNDEFINED "CUFFT_COPY_UNDEFINED" = 0x03
67
+
68
+ ctypedef enum cufftXtQueryType "cufftXtQueryType":
69
+ CUFFT_QUERY_1D_FACTORS "CUFFT_QUERY_1D_FACTORS" = 0x00
70
+ CUFFT_QUERY_UNDEFINED "CUFFT_QUERY_UNDEFINED" = 0x01
71
+
72
+ ctypedef enum cufftXtWorkAreaPolicy "cufftXtWorkAreaPolicy":
73
+ CUFFT_WORKAREA_MINIMAL "CUFFT_WORKAREA_MINIMAL" = 0
74
+ CUFFT_WORKAREA_USER "CUFFT_WORKAREA_USER" = 1
75
+ CUFFT_WORKAREA_PERFORMANCE "CUFFT_WORKAREA_PERFORMANCE" = 2
76
+
77
+ ctypedef enum cufftXtCallbackType "cufftXtCallbackType":
78
+ CUFFT_CB_LD_COMPLEX "CUFFT_CB_LD_COMPLEX" = 0x0
79
+ CUFFT_CB_LD_COMPLEX_DOUBLE "CUFFT_CB_LD_COMPLEX_DOUBLE" = 0x1
80
+ CUFFT_CB_LD_REAL "CUFFT_CB_LD_REAL" = 0x2
81
+ CUFFT_CB_LD_REAL_DOUBLE "CUFFT_CB_LD_REAL_DOUBLE" = 0x3
82
+ CUFFT_CB_ST_COMPLEX "CUFFT_CB_ST_COMPLEX" = 0x4
83
+ CUFFT_CB_ST_COMPLEX_DOUBLE "CUFFT_CB_ST_COMPLEX_DOUBLE" = 0x5
84
+ CUFFT_CB_ST_REAL "CUFFT_CB_ST_REAL" = 0x6
85
+ CUFFT_CB_ST_REAL_DOUBLE "CUFFT_CB_ST_REAL_DOUBLE" = 0x7
86
+ CUFFT_CB_UNDEFINED "CUFFT_CB_UNDEFINED" = 0x8
87
+
88
+ ctypedef enum cufftMpCommType "cufftMpCommType":
89
+ CUFFT_COMM_MPI "CUFFT_COMM_MPI" = 0x00
90
+ CUFFT_COMM_NONE "CUFFT_COMM_NONE" = 0x01
91
+
92
+
93
+ # types
94
+ cdef extern from *:
95
+ """
96
+ #include <driver_types.h>
97
+ #include <library_types.h>
98
+ #include <cuComplex.h>
99
+
100
+ #define MAX_CUDA_DESCRIPTOR_GPUS 64
101
+ #define CUFFT_FORWARD -1
102
+ #define CUFFT_INVERSE 1
103
+ """
104
+ ctypedef void* cudaStream_t 'cudaStream_t'
105
+ ctypedef int cudaDataType_t 'cudaDataType_t'
106
+ ctypedef int cudaDataType 'cudaDataType'
107
+ ctypedef int libraryPropertyType_t 'libraryPropertyType_t'
108
+ ctypedef int libraryPropertyType 'libraryPropertyType'
109
+
110
+ ctypedef struct cuComplex:
111
+ pass
112
+ ctypedef struct cuDoubleComplex:
113
+ pass
114
+
115
+ cdef const int MAX_CUDA_DESCRIPTOR_GPUS
116
+ cdef const int cuFFTFORWARD
117
+ cdef const int cuFFTINVERSE
118
+
119
+
120
+ ctypedef float cufftReal 'cufftReal'
121
+ ctypedef double cufftDoubleReal 'cufftDoubleReal'
122
+ ctypedef cuComplex cufftComplex 'cufftComplex'
123
+ ctypedef cuDoubleComplex cufftDoubleComplex 'cufftDoubleComplex'
124
+ ctypedef int cufftHandle 'cufftHandle'
125
+ ctypedef void* cufftReshapeHandle 'cufftReshapeHandle'
126
+ ctypedef struct cufftXt1dFactors 'cufftXt1dFactors':
127
+ long long int size
128
+ long long int stringCount
129
+ long long int stringLength
130
+ long long int substringLength
131
+ long long int factor1
132
+ long long int factor2
133
+ long long int stringMask
134
+ long long int substringMask
135
+ long long int factor1Mask
136
+ long long int factor2Mask
137
+ int stringShift
138
+ int substringShift
139
+ int factor1Shift
140
+ int factor2Shift
141
+
142
+ ctypedef cufftComplex (*cufftCallbackLoadC 'cufftCallbackLoadC')(
143
+ void* dataIn,
144
+ size_t offset,
145
+ void* callerInfo,
146
+ void* sharedPointer
147
+ )
148
+ ctypedef cufftDoubleComplex (*cufftCallbackLoadZ 'cufftCallbackLoadZ')(
149
+ void* dataIn,
150
+ size_t offset,
151
+ void* callerInfo,
152
+ void* sharedPointer
153
+ )
154
+ ctypedef cufftReal (*cufftCallbackLoadR 'cufftCallbackLoadR')(
155
+ void* dataIn,
156
+ size_t offset,
157
+ void* callerInfo,
158
+ void* sharedPointer
159
+ )
160
+ ctypedef cufftDoubleReal (*cufftCallbackLoadD 'cufftCallbackLoadD')(
161
+ void* dataIn,
162
+ size_t offset,
163
+ void* callerInfo,
164
+ void* sharedPointer
165
+ )
166
+ ctypedef cufftComplex (*cufftJITCallbackLoadC 'cufftJITCallbackLoadC')(
167
+ void* dataIn,
168
+ unsigned long long offset,
169
+ void* callerInfo,
170
+ void* sharedPointer
171
+ )
172
+ ctypedef cufftDoubleComplex (*cufftJITCallbackLoadZ 'cufftJITCallbackLoadZ')(
173
+ void* dataIn,
174
+ unsigned long long offset,
175
+ void* callerInfo,
176
+ void* sharedPointer
177
+ )
178
+ ctypedef cufftReal (*cufftJITCallbackLoadR 'cufftJITCallbackLoadR')(
179
+ void* dataIn,
180
+ unsigned long long offset,
181
+ void* callerInfo,
182
+ void* sharedPointer
183
+ )
184
+ ctypedef cufftDoubleReal (*cufftJITCallbackLoadD 'cufftJITCallbackLoadD')(
185
+ void* dataIn,
186
+ unsigned long long offset,
187
+ void* callerInfo,
188
+ void* sharedPointer
189
+ )
190
+ ctypedef void (*cufftCallbackStoreR 'cufftCallbackStoreR')(
191
+ void* dataOut,
192
+ size_t offset,
193
+ cufftReal element,
194
+ void* callerInfo,
195
+ void* sharedPointer
196
+ )
197
+ ctypedef void (*cufftJITCallbackStoreR 'cufftJITCallbackStoreR')(
198
+ void* dataOut,
199
+ unsigned long long offset,
200
+ cufftReal element,
201
+ void* callerInfo,
202
+ void* sharedPointer
203
+ )
204
+ ctypedef void (*cufftCallbackStoreD 'cufftCallbackStoreD')(
205
+ void* dataOut,
206
+ size_t offset,
207
+ cufftDoubleReal element,
208
+ void* callerInfo,
209
+ void* sharedPointer
210
+ )
211
+ ctypedef void (*cufftJITCallbackStoreD 'cufftJITCallbackStoreD')(
212
+ void* dataOut,
213
+ unsigned long long offset,
214
+ cufftDoubleReal element,
215
+ void* callerInfo,
216
+ void* sharedPointer
217
+ )
218
+ ctypedef void (*cufftCallbackStoreC 'cufftCallbackStoreC')(
219
+ void* dataOut,
220
+ size_t offset,
221
+ cufftComplex element,
222
+ void* callerInfo,
223
+ void* sharedPointer
224
+ )
225
+ ctypedef void (*cufftJITCallbackStoreC 'cufftJITCallbackStoreC')(
226
+ void* dataOut,
227
+ unsigned long long offset,
228
+ cufftComplex element,
229
+ void* callerInfo,
230
+ void* sharedPointer
231
+ )
232
+ ctypedef void (*cufftCallbackStoreZ 'cufftCallbackStoreZ')(
233
+ void* dataOut,
234
+ size_t offset,
235
+ cufftDoubleComplex element,
236
+ void* callerInfo,
237
+ void* sharedPointer
238
+ )
239
+ ctypedef void (*cufftJITCallbackStoreZ 'cufftJITCallbackStoreZ')(
240
+ void* dataOut,
241
+ unsigned long long offset,
242
+ cufftDoubleComplex element,
243
+ void* callerInfo,
244
+ void* sharedPointer
245
+ )
246
+
247
+ ctypedef struct cudaXtDesc_t 'cudaXtDesc_t':
248
+ int version "version"
249
+ int nGPUs "nGPUs"
250
+ int GPUs "GPUs" [MAX_CUDA_DESCRIPTOR_GPUS]
251
+ void* data "data" [MAX_CUDA_DESCRIPTOR_GPUS]
252
+ size_t size "size" [MAX_CUDA_DESCRIPTOR_GPUS]
253
+ void* cudaXtState "cudaXtState"
254
+
255
+ ctypedef cudaXtDesc_t cudaXtDesc 'cudaXtDesc'
256
+
257
+ ctypedef struct cudaLibXtDesc_t 'cudaLibXtDesc_t':
258
+ int version "version"
259
+ cudaXtDesc* descriptor "descriptor"
260
+ libFormat_t library "library"
261
+ int subFormat "subFormat"
262
+ void* libDescriptor "libDescriptor"
263
+
264
+ ctypedef cudaLibXtDesc_t cudaLibXtDesc 'cudaLibXtDesc'
265
+
266
+
267
+ ###############################################################################
268
+ # Functions
269
+ ###############################################################################
270
+
271
+ cdef cufftResult cufftPlan1d(cufftHandle* plan, int nx, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
272
+ cdef cufftResult cufftPlan2d(cufftHandle* plan, int nx, int ny, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
273
+ cdef cufftResult cufftPlan3d(cufftHandle* plan, int nx, int ny, int nz, cufftType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
274
+ cdef cufftResult cufftPlanMany(cufftHandle* plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
275
+ cdef cufftResult cufftMakePlan1d(cufftHandle plan, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
276
+ cdef cufftResult cufftMakePlan2d(cufftHandle plan, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
277
+ cdef cufftResult cufftMakePlan3d(cufftHandle plan, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
278
+ cdef cufftResult cufftMakePlanMany(cufftHandle plan, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
279
+ cdef cufftResult cufftMakePlanMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
280
+ cdef cufftResult cufftGetSizeMany64(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, long long int* onembed, long long int ostride, long long int odist, cufftType type, long long int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
281
+ cdef cufftResult cufftEstimate1d(int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
282
+ cdef cufftResult cufftEstimate2d(int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
283
+ cdef cufftResult cufftEstimate3d(int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
284
+ cdef cufftResult cufftEstimateMany(int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
285
+ cdef cufftResult cufftCreate(cufftHandle* handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
286
+ cdef cufftResult cufftGetSize1d(cufftHandle handle, int nx, cufftType type, int batch, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
287
+ cdef cufftResult cufftGetSize2d(cufftHandle handle, int nx, int ny, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
288
+ cdef cufftResult cufftGetSize3d(cufftHandle handle, int nx, int ny, int nz, cufftType type, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
289
+ cdef cufftResult cufftGetSizeMany(cufftHandle handle, int rank, int* n, int* inembed, int istride, int idist, int* onembed, int ostride, int odist, cufftType type, int batch, size_t* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
290
+ cdef cufftResult cufftGetSize(cufftHandle handle, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
291
+ cdef cufftResult cufftSetWorkArea(cufftHandle plan, void* workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
292
+ cdef cufftResult cufftSetAutoAllocation(cufftHandle plan, int autoAllocate) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
293
+ cdef cufftResult cufftExecC2C(cufftHandle plan, cufftComplex* idata, cufftComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
294
+ cdef cufftResult cufftExecR2C(cufftHandle plan, cufftReal* idata, cufftComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
295
+ cdef cufftResult cufftExecC2R(cufftHandle plan, cufftComplex* idata, cufftReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
296
+ cdef cufftResult cufftExecZ2Z(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleComplex* odata, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
297
+ cdef cufftResult cufftExecD2Z(cufftHandle plan, cufftDoubleReal* idata, cufftDoubleComplex* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
298
+ cdef cufftResult cufftExecZ2D(cufftHandle plan, cufftDoubleComplex* idata, cufftDoubleReal* odata) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
299
+ cdef cufftResult cufftSetStream(cufftHandle plan, cudaStream_t stream) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
300
+ cdef cufftResult cufftDestroy(cufftHandle plan) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
301
+ cdef cufftResult cufftGetVersion(int* version) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
302
+ cdef cufftResult cufftGetProperty(libraryPropertyType type, int* value) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
303
+ cdef cufftResult cufftSetPlanPropertyInt64(cufftHandle plan, cufftProperty property, const long long int inputValueInt) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
304
+ cdef cufftResult cufftGetPlanPropertyInt64(cufftHandle plan, cufftProperty property, long long int* returnPtrValue) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
305
+ cdef cufftResult cufftResetPlanProperty(cufftHandle plan, cufftProperty property) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
306
+ cdef cufftResult cufftXtSetGPUs(cufftHandle handle, int nGPUs, int* whichGPUs) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
307
+ cdef cufftResult cufftXtMalloc(cufftHandle plan, cudaLibXtDesc** descriptor, cufftXtSubFormat format) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
308
+ cdef cufftResult cufftXtMemcpy(cufftHandle plan, void* dstPointer, void* srcPointer, cufftXtCopyType type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
309
+ cdef cufftResult cufftXtFree(cudaLibXtDesc* descriptor) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
310
+ cdef cufftResult cufftXtSetWorkArea(cufftHandle plan, void** workArea) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
311
+ cdef cufftResult cufftXtExecDescriptorC2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
312
+ cdef cufftResult cufftXtExecDescriptorR2C(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
313
+ cdef cufftResult cufftXtExecDescriptorC2R(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
314
+ cdef cufftResult cufftXtExecDescriptorZ2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
315
+ cdef cufftResult cufftXtExecDescriptorD2Z(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
316
+ cdef cufftResult cufftXtExecDescriptorZ2D(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
317
+ cdef cufftResult cufftXtQueryPlan(cufftHandle plan, void* queryStruct, cufftXtQueryType queryType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
318
+ cdef cufftResult cufftXtClearCallback(cufftHandle plan, cufftXtCallbackType cbType) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
319
+ cdef cufftResult cufftXtSetCallbackSharedSize(cufftHandle plan, cufftXtCallbackType cbType, size_t sharedSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
320
+ cdef cufftResult cufftXtMakePlanMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
321
+ cdef cufftResult cufftXtGetSizeMany(cufftHandle plan, int rank, long long int* n, long long int* inembed, long long int istride, long long int idist, cudaDataType inputtype, long long int* onembed, long long int ostride, long long int odist, cudaDataType outputtype, long long int batch, size_t* workSize, cudaDataType executiontype) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
322
+ cdef cufftResult cufftXtExec(cufftHandle plan, void* input, void* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
323
+ cdef cufftResult cufftXtExecDescriptor(cufftHandle plan, cudaLibXtDesc* input, cudaLibXtDesc* output, int direction) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
324
+ cdef cufftResult cufftXtSetWorkAreaPolicy(cufftHandle plan, cufftXtWorkAreaPolicy policy, size_t* workSize) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
325
+ cdef cufftResult cufftMpAttachComm(cufftHandle plan, cufftMpCommType comm_type, void* comm_handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
326
+ cdef cufftResult cufftXtSetDistribution(cufftHandle plan, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_input, const long long int* strides_output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
327
+ cdef cufftResult cufftXtSetSubformatDefault(cufftHandle plan, cufftXtSubFormat subformat_forward, cufftXtSubFormat subformat_inverse) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
328
+ cdef cufftResult cufftMpCreateReshape(cufftReshapeHandle* handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
329
+ cdef cufftResult cufftMpAttachReshapeComm(cufftReshapeHandle handle, cufftMpCommType comm_type, void* comm_handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
330
+ cdef cufftResult cufftMpGetReshapeSize(cufftReshapeHandle handle, size_t* workspace_size) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
331
+ cdef cufftResult __cufftMpMakeReshape_11_2(cufftReshapeHandle handle, size_t element_size, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_input, const long long int* strides_output) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
332
+ cdef cufftResult cufftMpExecReshapeAsync(cufftReshapeHandle handle, void* data_out, const void* data_in, void* workspace, cudaStream_t stream) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
333
+ cdef cufftResult cufftMpDestroyReshape(cufftReshapeHandle handle) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil
334
+ cdef cufftResult __cufftMpMakeReshape_11_4(cufftReshapeHandle handle, size_t element_size, int rank, const long long int* lower_input, const long long int* upper_input, const long long int* strides_input, const long long int* lower_output, const long long int* upper_output, const long long int* strides_output, void* comm_handle, cufftMpCommType comm_type) except?_CUFFTRESULT_INTERNAL_LOADING_ERROR nogil