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,1461 @@
1
+ # Copyright (c) 2026, NVIDIA CORPORATION & AFFILIATES. ALL RIGHTS RESERVED.
2
+ #
3
+ # SPDX-License-Identifier: Apache-2.0
4
+
5
+ import _cython_3_2_5
6
+ import enum
7
+ import numpy.dtypes
8
+ from _typeshed import Incomplete
9
+ from typing import Any, Callable, ClassVar
10
+
11
+ __pyx_capi__: dict
12
+ __test__: dict
13
+ check_status: _cython_3_2_5.cython_function_or_method
14
+ create: _cython_3_2_5.cython_function_or_method
15
+ destroy: _cython_3_2_5.cython_function_or_method
16
+ disable_cpu_instructions_set_mask: _cython_3_2_5.cython_function_or_method
17
+ emulation_desc_create: _cython_3_2_5.cython_function_or_method
18
+ emulation_desc_destroy: _cython_3_2_5.cython_function_or_method
19
+ emulation_desc_get_attribute: _cython_3_2_5.cython_function_or_method
20
+ emulation_desc_set_attribute: _cython_3_2_5.cython_function_or_method
21
+ get_cudart_version: _cython_3_2_5.cython_function_or_method
22
+ get_emulation_desc_attribute_dtype: _cython_3_2_5.cython_function_or_method
23
+ get_matmul_algo_cap_attribute_dtype: _cython_3_2_5.cython_function_or_method
24
+ get_matmul_algo_config_attribute_dtype: _cython_3_2_5.cython_function_or_method
25
+ get_matmul_desc_attribute_dtype: _cython_3_2_5.cython_function_or_method
26
+ get_matmul_preference_attribute_dtype: _cython_3_2_5.cython_function_or_method
27
+ get_matrix_layout_attribute_dtype: _cython_3_2_5.cython_function_or_method
28
+ get_matrix_transform_desc_attribute_dtype: _cython_3_2_5.cython_function_or_method
29
+ get_property: _cython_3_2_5.cython_function_or_method
30
+ get_status_name: _cython_3_2_5.cython_function_or_method
31
+ get_status_string: _cython_3_2_5.cython_function_or_method
32
+ get_version: _cython_3_2_5.cython_function_or_method
33
+ grouped_matrix_layout_create: _cython_3_2_5.cython_function_or_method
34
+ heuristics_cache_get_capacity: _cython_3_2_5.cython_function_or_method
35
+ heuristics_cache_set_capacity: _cython_3_2_5.cython_function_or_method
36
+ logger_force_disable: _cython_3_2_5.cython_function_or_method
37
+ logger_open_file: _cython_3_2_5.cython_function_or_method
38
+ logger_set_level: _cython_3_2_5.cython_function_or_method
39
+ logger_set_mask: _cython_3_2_5.cython_function_or_method
40
+ matmul: _cython_3_2_5.cython_function_or_method
41
+ matmul_algo_cap_get_attribute: _cython_3_2_5.cython_function_or_method
42
+ matmul_algo_check: _cython_3_2_5.cython_function_or_method
43
+ matmul_algo_config_get_attribute: _cython_3_2_5.cython_function_or_method
44
+ matmul_algo_config_set_attribute: _cython_3_2_5.cython_function_or_method
45
+ matmul_algo_dtype: numpy.dtypes.VoidDType
46
+ matmul_algo_get_heuristic: _cython_3_2_5.cython_function_or_method
47
+ matmul_algo_get_ids: _cython_3_2_5.cython_function_or_method
48
+ matmul_algo_init: _cython_3_2_5.cython_function_or_method
49
+ matmul_desc_create: _cython_3_2_5.cython_function_or_method
50
+ matmul_desc_destroy: _cython_3_2_5.cython_function_or_method
51
+ matmul_desc_get_attribute: _cython_3_2_5.cython_function_or_method
52
+ matmul_desc_set_attribute: _cython_3_2_5.cython_function_or_method
53
+ matmul_heuristic_result_dtype: numpy.dtypes.VoidDType
54
+ matmul_preference_create: _cython_3_2_5.cython_function_or_method
55
+ matmul_preference_destroy: _cython_3_2_5.cython_function_or_method
56
+ matmul_preference_get_attribute: _cython_3_2_5.cython_function_or_method
57
+ matmul_preference_set_attribute: _cython_3_2_5.cython_function_or_method
58
+ matrix_layout_create: _cython_3_2_5.cython_function_or_method
59
+ matrix_layout_destroy: _cython_3_2_5.cython_function_or_method
60
+ matrix_layout_get_attribute: _cython_3_2_5.cython_function_or_method
61
+ matrix_layout_set_attribute: _cython_3_2_5.cython_function_or_method
62
+ matrix_transform: _cython_3_2_5.cython_function_or_method
63
+ matrix_transform_desc_create: _cython_3_2_5.cython_function_or_method
64
+ matrix_transform_desc_destroy: _cython_3_2_5.cython_function_or_method
65
+ matrix_transform_desc_get_attribute: _cython_3_2_5.cython_function_or_method
66
+ matrix_transform_desc_set_attribute: _cython_3_2_5.cython_function_or_method
67
+
68
+ class BatchMode(enum.IntEnum):
69
+ """
70
+ See `cublasLtBatchMode_t`.
71
+ """
72
+ __new__: ClassVar[Callable] = ...
73
+ GROUPED: ClassVar[BatchMode] = ...
74
+ POINTER_ARRAY: ClassVar[BatchMode] = ...
75
+ STRIDED: ClassVar[BatchMode] = ...
76
+ _generate_next_value_: ClassVar[Callable] = ...
77
+ _hashable_values_: ClassVar[list] = ...
78
+ _member_map_: ClassVar[dict] = ...
79
+ _member_names_: ClassVar[list] = ...
80
+ _member_type_: ClassVar[type[int]] = ...
81
+ _unhashable_values_: ClassVar[list] = ...
82
+ _unhashable_values_map_: ClassVar[dict] = ...
83
+ _use_args_: ClassVar[bool] = ...
84
+ _value2member_map_: ClassVar[dict] = ...
85
+ def __format__(self, *args, **kwargs) -> str:
86
+ """Convert to a string according to format_spec."""
87
+
88
+ class ClusterShape(enum.IntEnum):
89
+ """
90
+ See `cublasLtClusterShape_t`.
91
+ """
92
+ __new__: ClassVar[Callable] = ...
93
+ SHAPE_10x1x1: ClassVar[ClusterShape] = ...
94
+ SHAPE_11x1x1: ClassVar[ClusterShape] = ...
95
+ SHAPE_12x1x1: ClassVar[ClusterShape] = ...
96
+ SHAPE_13x1x1: ClassVar[ClusterShape] = ...
97
+ SHAPE_14x1x1: ClassVar[ClusterShape] = ...
98
+ SHAPE_15x1x1: ClassVar[ClusterShape] = ...
99
+ SHAPE_16x1x1: ClassVar[ClusterShape] = ...
100
+ SHAPE_1x10x1: ClassVar[ClusterShape] = ...
101
+ SHAPE_1x11x1: ClassVar[ClusterShape] = ...
102
+ SHAPE_1x12x1: ClassVar[ClusterShape] = ...
103
+ SHAPE_1x13x1: ClassVar[ClusterShape] = ...
104
+ SHAPE_1x14x1: ClassVar[ClusterShape] = ...
105
+ SHAPE_1x15x1: ClassVar[ClusterShape] = ...
106
+ SHAPE_1x16x1: ClassVar[ClusterShape] = ...
107
+ SHAPE_1x1x1: ClassVar[ClusterShape] = ...
108
+ SHAPE_1x2x1: ClassVar[ClusterShape] = ...
109
+ SHAPE_1x3x1: ClassVar[ClusterShape] = ...
110
+ SHAPE_1x4x1: ClassVar[ClusterShape] = ...
111
+ SHAPE_1x5x1: ClassVar[ClusterShape] = ...
112
+ SHAPE_1x6x1: ClassVar[ClusterShape] = ...
113
+ SHAPE_1x7x1: ClassVar[ClusterShape] = ...
114
+ SHAPE_1x8x1: ClassVar[ClusterShape] = ...
115
+ SHAPE_1x9x1: ClassVar[ClusterShape] = ...
116
+ SHAPE_2x1x1: ClassVar[ClusterShape] = ...
117
+ SHAPE_2x2x1: ClassVar[ClusterShape] = ...
118
+ SHAPE_2x3x1: ClassVar[ClusterShape] = ...
119
+ SHAPE_2x4x1: ClassVar[ClusterShape] = ...
120
+ SHAPE_2x5x1: ClassVar[ClusterShape] = ...
121
+ SHAPE_2x6x1: ClassVar[ClusterShape] = ...
122
+ SHAPE_2x7x1: ClassVar[ClusterShape] = ...
123
+ SHAPE_2x8x1: ClassVar[ClusterShape] = ...
124
+ SHAPE_3x1x1: ClassVar[ClusterShape] = ...
125
+ SHAPE_3x2x1: ClassVar[ClusterShape] = ...
126
+ SHAPE_3x3x1: ClassVar[ClusterShape] = ...
127
+ SHAPE_3x4x1: ClassVar[ClusterShape] = ...
128
+ SHAPE_3x5x1: ClassVar[ClusterShape] = ...
129
+ SHAPE_4x1x1: ClassVar[ClusterShape] = ...
130
+ SHAPE_4x2x1: ClassVar[ClusterShape] = ...
131
+ SHAPE_4x3x1: ClassVar[ClusterShape] = ...
132
+ SHAPE_4x4x1: ClassVar[ClusterShape] = ...
133
+ SHAPE_5x1x1: ClassVar[ClusterShape] = ...
134
+ SHAPE_5x2x1: ClassVar[ClusterShape] = ...
135
+ SHAPE_5x3x1: ClassVar[ClusterShape] = ...
136
+ SHAPE_6x1x1: ClassVar[ClusterShape] = ...
137
+ SHAPE_6x2x1: ClassVar[ClusterShape] = ...
138
+ SHAPE_7x1x1: ClassVar[ClusterShape] = ...
139
+ SHAPE_7x2x1: ClassVar[ClusterShape] = ...
140
+ SHAPE_8x1x1: ClassVar[ClusterShape] = ...
141
+ SHAPE_8x2x1: ClassVar[ClusterShape] = ...
142
+ SHAPE_9x1x1: ClassVar[ClusterShape] = ...
143
+ SHAPE_AUTO: ClassVar[ClusterShape] = ...
144
+ _generate_next_value_: ClassVar[Callable] = ...
145
+ _hashable_values_: ClassVar[list] = ...
146
+ _member_map_: ClassVar[dict] = ...
147
+ _member_names_: ClassVar[list] = ...
148
+ _member_type_: ClassVar[type[int]] = ...
149
+ _unhashable_values_: ClassVar[list] = ...
150
+ _unhashable_values_map_: ClassVar[dict] = ...
151
+ _use_args_: ClassVar[bool] = ...
152
+ _value2member_map_: ClassVar[dict] = ...
153
+ def __format__(self, *args, **kwargs) -> str:
154
+ """Convert to a string according to format_spec."""
155
+
156
+ class EmulationDescAttribute(enum.IntEnum):
157
+ """
158
+ See `cublasLtEmulationDescAttributes_t`.
159
+ """
160
+ __new__: ClassVar[Callable] = ...
161
+ FIXEDPOINT_MANTISSA_BIT_COUNT_POINTER: ClassVar[EmulationDescAttribute] = ...
162
+ FIXEDPOINT_MANTISSA_BIT_OFFSET: ClassVar[EmulationDescAttribute] = ...
163
+ FIXEDPOINT_MANTISSA_CONTROL: ClassVar[EmulationDescAttribute] = ...
164
+ FIXEDPOINT_MAX_MANTISSA_BIT_COUNT: ClassVar[EmulationDescAttribute] = ...
165
+ SPECIAL_VALUES_SUPPORT: ClassVar[EmulationDescAttribute] = ...
166
+ STRATEGY: ClassVar[EmulationDescAttribute] = ...
167
+ _generate_next_value_: ClassVar[Callable] = ...
168
+ _hashable_values_: ClassVar[list] = ...
169
+ _member_map_: ClassVar[dict] = ...
170
+ _member_names_: ClassVar[list] = ...
171
+ _member_type_: ClassVar[type[int]] = ...
172
+ _unhashable_values_: ClassVar[list] = ...
173
+ _unhashable_values_map_: ClassVar[dict] = ...
174
+ _use_args_: ClassVar[bool] = ...
175
+ _value2member_map_: ClassVar[dict] = ...
176
+ def __format__(self, *args, **kwargs) -> str:
177
+ """Convert to a string according to format_spec."""
178
+
179
+ class Epilogue(enum.IntEnum):
180
+ """
181
+ See `cublasLtEpilogue_t`.
182
+ """
183
+ __new__: ClassVar[Callable] = ...
184
+ BGRADA: ClassVar[Epilogue] = ...
185
+ BGRADB: ClassVar[Epilogue] = ...
186
+ BIAS: ClassVar[Epilogue] = ...
187
+ DEFAULT: ClassVar[Epilogue] = ...
188
+ DGELU: ClassVar[Epilogue] = ...
189
+ DGELU_BGRAD: ClassVar[Epilogue] = ...
190
+ DRELU: ClassVar[Epilogue] = ...
191
+ DRELU_BGRAD: ClassVar[Epilogue] = ...
192
+ GELU: ClassVar[Epilogue] = ...
193
+ GELU_AUX: ClassVar[Epilogue] = ...
194
+ GELU_AUX_BIAS: ClassVar[Epilogue] = ...
195
+ GELU_BIAS: ClassVar[Epilogue] = ...
196
+ RELU: ClassVar[Epilogue] = ...
197
+ RELU_AUX: ClassVar[Epilogue] = ...
198
+ RELU_AUX_BIAS: ClassVar[Epilogue] = ...
199
+ RELU_BIAS: ClassVar[Epilogue] = ...
200
+ _generate_next_value_: ClassVar[Callable] = ...
201
+ _hashable_values_: ClassVar[list] = ...
202
+ _member_map_: ClassVar[dict] = ...
203
+ _member_names_: ClassVar[list] = ...
204
+ _member_type_: ClassVar[type[int]] = ...
205
+ _unhashable_values_: ClassVar[list] = ...
206
+ _unhashable_values_map_: ClassVar[dict] = ...
207
+ _use_args_: ClassVar[bool] = ...
208
+ _value2member_map_: ClassVar[dict] = ...
209
+ def __format__(self, *args, **kwargs) -> str:
210
+ """Convert to a string according to format_spec."""
211
+
212
+ class IntegerWidth(enum.IntEnum):
213
+ """
214
+ See `cublasLtIntegerWidth_t`.
215
+ """
216
+ __new__: ClassVar[Callable] = ...
217
+ WIDTH_32: ClassVar[IntegerWidth] = ...
218
+ WIDTH_64: ClassVar[IntegerWidth] = ...
219
+ _generate_next_value_: ClassVar[Callable] = ...
220
+ _hashable_values_: ClassVar[list] = ...
221
+ _member_map_: ClassVar[dict] = ...
222
+ _member_names_: ClassVar[list] = ...
223
+ _member_type_: ClassVar[type[int]] = ...
224
+ _unhashable_values_: ClassVar[list] = ...
225
+ _unhashable_values_map_: ClassVar[dict] = ...
226
+ _use_args_: ClassVar[bool] = ...
227
+ _value2member_map_: ClassVar[dict] = ...
228
+ def __format__(self, *args, **kwargs) -> str:
229
+ """Convert to a string according to format_spec."""
230
+
231
+ class MatmulAlgo:
232
+ """MatmulAlgo(size=1)
233
+
234
+ Empty-initialize an array of `cublasLtMatmulAlgo_t`.
235
+ The resulting object is of length `size` and of dtype `matmul_algo_dtype`.
236
+ If default-constructed, the instance represents a single struct.
237
+
238
+ Args:
239
+ size (int): number of structs, default=1.
240
+
241
+ .. seealso:: `cublasLtMatmulAlgo_t`"""
242
+ __pyx_vtable__: ClassVar[PyCapsule] = ...
243
+ data_: Incomplete
244
+ ptr: Incomplete
245
+ def __init__(self, size=...) -> Any:
246
+ """Initialize self. See help(type(self)) for accurate signature."""
247
+ @staticmethod
248
+ def from_buffer(buffer) -> Any:
249
+ """MatmulAlgo.from_buffer(buffer)
250
+
251
+ Create an MatmulAlgo instance with the memory from the given buffer."""
252
+ @staticmethod
253
+ def from_data(data) -> Any:
254
+ """MatmulAlgo.from_data(data)
255
+
256
+ Create an MatmulAlgo instance wrapping the given NumPy array.
257
+
258
+ Args:
259
+ data (_numpy.ndarray): a 1D array of dtype `matmul_algo_dtype` holding the data."""
260
+ @staticmethod
261
+ def from_ptr(intptr_tptr, size_tsize=..., boolreadonly=...) -> Any:
262
+ """MatmulAlgo.from_ptr(intptr_t ptr, size_t size=1, bool readonly=False)
263
+
264
+ Create an MatmulAlgo instance wrapping the given pointer.
265
+
266
+ Args:
267
+ ptr (intptr_t): pointer address as Python :class:`int` to the data.
268
+ size (int): number of structs, default=1.
269
+ readonly (bool): whether the data is read-only (to the user). default is `False`."""
270
+ def __buffer__(self, *args, **kwargs):
271
+ """Return a buffer object that exposes the underlying memory of the object."""
272
+ def __delitem__(self, other) -> None:
273
+ """Delete self[key]."""
274
+ def __eq__(self, other: object) -> bool:
275
+ """Return self==value."""
276
+ def __ge__(self, other: object) -> bool:
277
+ """Return self>=value."""
278
+ def __getitem__(self, index):
279
+ """Return self[key]."""
280
+ def __gt__(self, other: object) -> bool:
281
+ """Return self>value."""
282
+ def __int__(self) -> int:
283
+ """int(self)"""
284
+ def __le__(self, other: object) -> bool:
285
+ """Return self<=value."""
286
+ def __len__(self) -> int:
287
+ """Return len(self)."""
288
+ def __lt__(self, other: object) -> bool:
289
+ """Return self<value."""
290
+ def __ne__(self, other: object) -> bool:
291
+ """Return self!=value."""
292
+ def __reduce__(self):
293
+ """MatmulAlgo.__reduce_cython__(self)"""
294
+ def __reduce_cython__(self) -> Any:
295
+ """MatmulAlgo.__reduce_cython__(self)"""
296
+ def __release_buffer__(self, *args, **kwargs):
297
+ """Release the buffer object that exposes the underlying memory of the object."""
298
+ def __setitem__(self, index, object) -> None:
299
+ """Set self[key] to value."""
300
+ def __setstate_cython__(self, __pyx_state) -> Any:
301
+ """MatmulAlgo.__setstate_cython__(self, __pyx_state)"""
302
+
303
+ class MatmulAlgoCapAttribute(enum.IntEnum):
304
+ """
305
+ See `cublasLtMatmulAlgoCapAttributes_t`.
306
+ """
307
+ __new__: ClassVar[Callable] = ...
308
+ ATOMIC_SYNC: ClassVar[MatmulAlgoCapAttribute] = ...
309
+ CTA_SWIZZLING_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
310
+ CUSTOM_MEMORY_ORDER: ClassVar[MatmulAlgoCapAttribute] = ...
311
+ CUSTOM_OPTION_MAX: ClassVar[MatmulAlgoCapAttribute] = ...
312
+ EPILOGUE_MASK: ClassVar[MatmulAlgoCapAttribute] = ...
313
+ FLOATING_POINT_EMULATION_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
314
+ GAUSSIAN_IMPL: ClassVar[MatmulAlgoCapAttribute] = ...
315
+ LD_NEGATIVE: ClassVar[MatmulAlgoCapAttribute] = ...
316
+ MATHMODE_IMPL: ClassVar[MatmulAlgoCapAttribute] = ...
317
+ MIN_ALIGNMENT_A_BYTES: ClassVar[MatmulAlgoCapAttribute] = ...
318
+ MIN_ALIGNMENT_B_BYTES: ClassVar[MatmulAlgoCapAttribute] = ...
319
+ MIN_ALIGNMENT_C_BYTES: ClassVar[MatmulAlgoCapAttribute] = ...
320
+ MIN_ALIGNMENT_D_BYTES: ClassVar[MatmulAlgoCapAttribute] = ...
321
+ NUMERICAL_IMPL_FLAGS: ClassVar[MatmulAlgoCapAttribute] = ...
322
+ OUT_OF_PLACE_RESULT_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
323
+ POINTER_ARRAY_BATCH_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
324
+ POINTER_ARRAY_GROUPED_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
325
+ POINTER_MODE_MASK: ClassVar[MatmulAlgoCapAttribute] = ...
326
+ REDUCTION_SCHEME_MASK: ClassVar[MatmulAlgoCapAttribute] = ...
327
+ SPLITK_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
328
+ STAGES_IDS: ClassVar[MatmulAlgoCapAttribute] = ...
329
+ STRIDED_BATCH_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
330
+ TILE_IDS: ClassVar[MatmulAlgoCapAttribute] = ...
331
+ UPLO_SUPPORT: ClassVar[MatmulAlgoCapAttribute] = ...
332
+ _generate_next_value_: ClassVar[Callable] = ...
333
+ _hashable_values_: ClassVar[list] = ...
334
+ _member_map_: ClassVar[dict] = ...
335
+ _member_names_: ClassVar[list] = ...
336
+ _member_type_: ClassVar[type[int]] = ...
337
+ _unhashable_values_: ClassVar[list] = ...
338
+ _unhashable_values_map_: ClassVar[dict] = ...
339
+ _use_args_: ClassVar[bool] = ...
340
+ _value2member_map_: ClassVar[dict] = ...
341
+ def __format__(self, *args, **kwargs) -> str:
342
+ """Convert to a string according to format_spec."""
343
+
344
+ class MatmulAlgoConfigAttribute(enum.IntEnum):
345
+ """
346
+ See `cublasLtMatmulAlgoConfigAttributes_t`.
347
+ """
348
+ __new__: ClassVar[Callable] = ...
349
+ CLUSTER_SHAPE_ID: ClassVar[MatmulAlgoConfigAttribute] = ...
350
+ CTA_SWIZZLING: ClassVar[MatmulAlgoConfigAttribute] = ...
351
+ CUSTOM_OPTION: ClassVar[MatmulAlgoConfigAttribute] = ...
352
+ ID: ClassVar[MatmulAlgoConfigAttribute] = ...
353
+ INNER_SHAPE_ID: ClassVar[MatmulAlgoConfigAttribute] = ...
354
+ REDUCTION_SCHEME: ClassVar[MatmulAlgoConfigAttribute] = ...
355
+ SPLITK_NUM: ClassVar[MatmulAlgoConfigAttribute] = ...
356
+ STAGES_ID: ClassVar[MatmulAlgoConfigAttribute] = ...
357
+ TILE_ID: ClassVar[MatmulAlgoConfigAttribute] = ...
358
+ _generate_next_value_: ClassVar[Callable] = ...
359
+ _hashable_values_: ClassVar[list] = ...
360
+ _member_map_: ClassVar[dict] = ...
361
+ _member_names_: ClassVar[list] = ...
362
+ _member_type_: ClassVar[type[int]] = ...
363
+ _unhashable_values_: ClassVar[list] = ...
364
+ _unhashable_values_map_: ClassVar[dict] = ...
365
+ _use_args_: ClassVar[bool] = ...
366
+ _value2member_map_: ClassVar[dict] = ...
367
+ def __format__(self, *args, **kwargs) -> str:
368
+ """Convert to a string according to format_spec."""
369
+
370
+ class MatmulDescAttribute(enum.IntEnum):
371
+ """
372
+ See `cublasLtMatmulDescAttributes_t`.
373
+ """
374
+ __new__: ClassVar[Callable] = ...
375
+ ALPHA_BATCH_STRIDE: ClassVar[MatmulDescAttribute] = ...
376
+ ALPHA_VECTOR_BATCH_STRIDE: ClassVar[MatmulDescAttribute] = ...
377
+ AMAX_D_POINTER: ClassVar[MatmulDescAttribute] = ...
378
+ ATOMIC_SYNC_IN_COUNTERS_POINTER: ClassVar[MatmulDescAttribute] = ...
379
+ ATOMIC_SYNC_NUM_CHUNKS_D_COLS: ClassVar[MatmulDescAttribute] = ...
380
+ ATOMIC_SYNC_NUM_CHUNKS_D_ROWS: ClassVar[MatmulDescAttribute] = ...
381
+ ATOMIC_SYNC_OUT_COUNTERS_POINTER: ClassVar[MatmulDescAttribute] = ...
382
+ A_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
383
+ A_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
384
+ BETA_BATCH_STRIDE: ClassVar[MatmulDescAttribute] = ...
385
+ BIAS_BATCH_STRIDE: ClassVar[MatmulDescAttribute] = ...
386
+ BIAS_DATA_TYPE: ClassVar[MatmulDescAttribute] = ...
387
+ BIAS_POINTER: ClassVar[MatmulDescAttribute] = ...
388
+ B_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
389
+ B_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
390
+ COMPUTE_TYPE: ClassVar[MatmulDescAttribute] = ...
391
+ C_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
392
+ C_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
393
+ D_OUT_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
394
+ D_OUT_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
395
+ D_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
396
+ D_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
397
+ EMULATION_DESCRIPTOR: ClassVar[MatmulDescAttribute] = ...
398
+ EPILOGUE: ClassVar[MatmulDescAttribute] = ...
399
+ EPILOGUE_AUX_AMAX_POINTER: ClassVar[MatmulDescAttribute] = ...
400
+ EPILOGUE_AUX_BATCH_STRIDE: ClassVar[MatmulDescAttribute] = ...
401
+ EPILOGUE_AUX_DATA_TYPE: ClassVar[MatmulDescAttribute] = ...
402
+ EPILOGUE_AUX_LD: ClassVar[MatmulDescAttribute] = ...
403
+ EPILOGUE_AUX_POINTER: ClassVar[MatmulDescAttribute] = ...
404
+ EPILOGUE_AUX_SCALE_MODE: ClassVar[MatmulDescAttribute] = ...
405
+ EPILOGUE_AUX_SCALE_POINTER: ClassVar[MatmulDescAttribute] = ...
406
+ FAST_ACCUM: ClassVar[MatmulDescAttribute] = ...
407
+ FILL_MODE: ClassVar[MatmulDescAttribute] = ...
408
+ POINTER_MODE: ClassVar[MatmulDescAttribute] = ...
409
+ SCALE_TYPE: ClassVar[MatmulDescAttribute] = ...
410
+ SM_COUNT_TARGET: ClassVar[MatmulDescAttribute] = ...
411
+ TRANSA: ClassVar[MatmulDescAttribute] = ...
412
+ TRANSB: ClassVar[MatmulDescAttribute] = ...
413
+ TRANSC: ClassVar[MatmulDescAttribute] = ...
414
+ _generate_next_value_: ClassVar[Callable] = ...
415
+ _hashable_values_: ClassVar[list] = ...
416
+ _member_map_: ClassVar[dict] = ...
417
+ _member_names_: ClassVar[list] = ...
418
+ _member_type_: ClassVar[type[int]] = ...
419
+ _unhashable_values_: ClassVar[list] = ...
420
+ _unhashable_values_map_: ClassVar[dict] = ...
421
+ _use_args_: ClassVar[bool] = ...
422
+ _value2member_map_: ClassVar[dict] = ...
423
+ def __format__(self, *args, **kwargs) -> str:
424
+ """Convert to a string according to format_spec."""
425
+
426
+ class MatmulHeuristicResult:
427
+ """MatmulHeuristicResult(size=1)
428
+
429
+ Empty-initialize an array of `cublasLtMatmulHeuristicResult_t`.
430
+ The resulting object is of length `size` and of dtype `matmul_heuristic_result_dtype`.
431
+ If default-constructed, the instance represents a single struct.
432
+
433
+ Args:
434
+ size (int): number of structs, default=1.
435
+
436
+ .. seealso:: `cublasLtMatmulHeuristicResult_t`"""
437
+ __pyx_vtable__: ClassVar[PyCapsule] = ...
438
+ algo: matmul_algo_dtype
439
+ ptr: Incomplete
440
+ state: Incomplete
441
+ waves_count: Incomplete
442
+ workspace_size: Incomplete
443
+ def __init__(self, size=...) -> Any:
444
+ """Initialize self. See help(type(self)) for accurate signature."""
445
+ @staticmethod
446
+ def from_buffer(buffer) -> Any:
447
+ """MatmulHeuristicResult.from_buffer(buffer)
448
+
449
+ Create an MatmulHeuristicResult instance with the memory from the given buffer."""
450
+ @staticmethod
451
+ def from_data(data) -> Any:
452
+ """MatmulHeuristicResult.from_data(data)
453
+
454
+ Create an MatmulHeuristicResult instance wrapping the given NumPy array.
455
+
456
+ Args:
457
+ data (_numpy.ndarray): a 1D array of dtype `matmul_heuristic_result_dtype` holding the data."""
458
+ @staticmethod
459
+ def from_ptr(intptr_tptr, size_tsize=..., boolreadonly=...) -> Any:
460
+ """MatmulHeuristicResult.from_ptr(intptr_t ptr, size_t size=1, bool readonly=False)
461
+
462
+ Create an MatmulHeuristicResult instance wrapping the given pointer.
463
+
464
+ Args:
465
+ ptr (intptr_t): pointer address as Python :class:`int` to the data.
466
+ size (int): number of structs, default=1.
467
+ readonly (bool): whether the data is read-only (to the user). default is `False`."""
468
+ def __buffer__(self, *args, **kwargs):
469
+ """Return a buffer object that exposes the underlying memory of the object."""
470
+ def __delitem__(self, other) -> None:
471
+ """Delete self[key]."""
472
+ def __eq__(self, other: object) -> bool:
473
+ """Return self==value."""
474
+ def __ge__(self, other: object) -> bool:
475
+ """Return self>=value."""
476
+ def __getitem__(self, index):
477
+ """Return self[key]."""
478
+ def __gt__(self, other: object) -> bool:
479
+ """Return self>value."""
480
+ def __int__(self) -> int:
481
+ """int(self)"""
482
+ def __le__(self, other: object) -> bool:
483
+ """Return self<=value."""
484
+ def __len__(self) -> int:
485
+ """Return len(self)."""
486
+ def __lt__(self, other: object) -> bool:
487
+ """Return self<value."""
488
+ def __ne__(self, other: object) -> bool:
489
+ """Return self!=value."""
490
+ def __reduce__(self):
491
+ """MatmulHeuristicResult.__reduce_cython__(self)"""
492
+ def __reduce_cython__(self) -> Any:
493
+ """MatmulHeuristicResult.__reduce_cython__(self)"""
494
+ def __release_buffer__(self, *args, **kwargs):
495
+ """Release the buffer object that exposes the underlying memory of the object."""
496
+ def __setitem__(self, index, object) -> None:
497
+ """Set self[key] to value."""
498
+ def __setstate_cython__(self, __pyx_state) -> Any:
499
+ """MatmulHeuristicResult.__setstate_cython__(self, __pyx_state)"""
500
+
501
+ class MatmulInnerShape(enum.IntEnum):
502
+ """
503
+ See `cublasLtMatmulInnerShape_t`.
504
+ """
505
+ __new__: ClassVar[Callable] = ...
506
+ MMA16816: ClassVar[MatmulInnerShape] = ...
507
+ MMA1684: ClassVar[MatmulInnerShape] = ...
508
+ MMA1688: ClassVar[MatmulInnerShape] = ...
509
+ MMA884: ClassVar[MatmulInnerShape] = ...
510
+ UNDEFINED: ClassVar[MatmulInnerShape] = ...
511
+ _generate_next_value_: ClassVar[Callable] = ...
512
+ _hashable_values_: ClassVar[list] = ...
513
+ _member_map_: ClassVar[dict] = ...
514
+ _member_names_: ClassVar[list] = ...
515
+ _member_type_: ClassVar[type[int]] = ...
516
+ _unhashable_values_: ClassVar[list] = ...
517
+ _unhashable_values_map_: ClassVar[dict] = ...
518
+ _use_args_: ClassVar[bool] = ...
519
+ _value2member_map_: ClassVar[dict] = ...
520
+ def __format__(self, *args, **kwargs) -> str:
521
+ """Convert to a string according to format_spec."""
522
+
523
+ class MatmulMatrixScale(enum.IntEnum):
524
+ """
525
+ See `cublasLtMatmulMatrixScale_t`.
526
+ """
527
+ __new__: ClassVar[Callable] = ...
528
+ BLK128x128_32F: ClassVar[MatmulMatrixScale] = ...
529
+ OUTER_VEC_32F: ClassVar[MatmulMatrixScale] = ...
530
+ PER_BATCH_SCALAR_32F: ClassVar[MatmulMatrixScale] = ...
531
+ SCALAR_32F: ClassVar[MatmulMatrixScale] = ...
532
+ VEC128_32F: ClassVar[MatmulMatrixScale] = ...
533
+ VEC16_UE4M3: ClassVar[MatmulMatrixScale] = ...
534
+ VEC32_UE8M0: ClassVar[MatmulMatrixScale] = ...
535
+ _generate_next_value_: ClassVar[Callable] = ...
536
+ _hashable_values_: ClassVar[list] = ...
537
+ _member_map_: ClassVar[dict] = ...
538
+ _member_names_: ClassVar[list] = ...
539
+ _member_type_: ClassVar[type[int]] = ...
540
+ _unhashable_values_: ClassVar[list] = ...
541
+ _unhashable_values_map_: ClassVar[dict] = ...
542
+ _use_args_: ClassVar[bool] = ...
543
+ _value2member_map_: ClassVar[dict] = ...
544
+ def __format__(self, *args, **kwargs) -> str:
545
+ """Convert to a string according to format_spec."""
546
+
547
+ class MatmulPreferenceAttribute(enum.IntEnum):
548
+ """
549
+ See `cublasLtMatmulPreferenceAttributes_t`.
550
+ """
551
+ __new__: ClassVar[Callable] = ...
552
+ EPILOGUE_MASK: ClassVar[MatmulPreferenceAttribute] = ...
553
+ GAUSSIAN_MODE_MASK: ClassVar[MatmulPreferenceAttribute] = ...
554
+ GROUPED_AVERAGE_REDUCTION_DIM: ClassVar[MatmulPreferenceAttribute] = ...
555
+ GROUPED_DESC_D_AVERAGE_COLS: ClassVar[MatmulPreferenceAttribute] = ...
556
+ GROUPED_DESC_D_AVERAGE_ROWS: ClassVar[MatmulPreferenceAttribute] = ...
557
+ IMPL_MASK: ClassVar[MatmulPreferenceAttribute] = ...
558
+ MATH_MODE_MASK: ClassVar[MatmulPreferenceAttribute] = ...
559
+ MAX_WAVES_COUNT: ClassVar[MatmulPreferenceAttribute] = ...
560
+ MAX_WORKSPACE_BYTES: ClassVar[MatmulPreferenceAttribute] = ...
561
+ MIN_ALIGNMENT_A_BYTES: ClassVar[MatmulPreferenceAttribute] = ...
562
+ MIN_ALIGNMENT_B_BYTES: ClassVar[MatmulPreferenceAttribute] = ...
563
+ MIN_ALIGNMENT_C_BYTES: ClassVar[MatmulPreferenceAttribute] = ...
564
+ MIN_ALIGNMENT_D_BYTES: ClassVar[MatmulPreferenceAttribute] = ...
565
+ POINTER_MODE_MASK: ClassVar[MatmulPreferenceAttribute] = ...
566
+ REDUCTION_SCHEME_MASK: ClassVar[MatmulPreferenceAttribute] = ...
567
+ SEARCH_MODE: ClassVar[MatmulPreferenceAttribute] = ...
568
+ SM_COUNT_TARGET: ClassVar[MatmulPreferenceAttribute] = ...
569
+ _generate_next_value_: ClassVar[Callable] = ...
570
+ _hashable_values_: ClassVar[list] = ...
571
+ _member_map_: ClassVar[dict] = ...
572
+ _member_names_: ClassVar[list] = ...
573
+ _member_type_: ClassVar[type[int]] = ...
574
+ _unhashable_values_: ClassVar[list] = ...
575
+ _unhashable_values_map_: ClassVar[dict] = ...
576
+ _use_args_: ClassVar[bool] = ...
577
+ _value2member_map_: ClassVar[dict] = ...
578
+ def __format__(self, *args, **kwargs) -> str:
579
+ """Convert to a string according to format_spec."""
580
+
581
+ class MatmulSearch(enum.IntEnum):
582
+ """
583
+ See `cublasLtMatmulSearch_t`.
584
+ """
585
+ __new__: ClassVar[Callable] = ...
586
+ BEST_FIT: ClassVar[MatmulSearch] = ...
587
+ LIMITED_BY_ALGO_ID: ClassVar[MatmulSearch] = ...
588
+ RESERVED_02: ClassVar[MatmulSearch] = ...
589
+ RESERVED_03: ClassVar[MatmulSearch] = ...
590
+ RESERVED_04: ClassVar[MatmulSearch] = ...
591
+ RESERVED_05: ClassVar[MatmulSearch] = ...
592
+ RESERVED_06: ClassVar[MatmulSearch] = ...
593
+ RESERVED_07: ClassVar[MatmulSearch] = ...
594
+ RESERVED_08: ClassVar[MatmulSearch] = ...
595
+ RESERVED_09: ClassVar[MatmulSearch] = ...
596
+ _generate_next_value_: ClassVar[Callable] = ...
597
+ _hashable_values_: ClassVar[list] = ...
598
+ _member_map_: ClassVar[dict] = ...
599
+ _member_names_: ClassVar[list] = ...
600
+ _member_type_: ClassVar[type[int]] = ...
601
+ _unhashable_values_: ClassVar[list] = ...
602
+ _unhashable_values_map_: ClassVar[dict] = ...
603
+ _use_args_: ClassVar[bool] = ...
604
+ _value2member_map_: ClassVar[dict] = ...
605
+ def __format__(self, *args, **kwargs) -> str:
606
+ """Convert to a string according to format_spec."""
607
+
608
+ class MatmulStages(enum.IntEnum):
609
+ """
610
+ See `cublasLtMatmulStages_t`.
611
+ """
612
+ __new__: ClassVar[Callable] = ...
613
+ STAGES_128x1: ClassVar[MatmulStages] = ...
614
+ STAGES_128x2: ClassVar[MatmulStages] = ...
615
+ STAGES_128x3: ClassVar[MatmulStages] = ...
616
+ STAGES_128x4: ClassVar[MatmulStages] = ...
617
+ STAGES_128x5: ClassVar[MatmulStages] = ...
618
+ STAGES_128x6: ClassVar[MatmulStages] = ...
619
+ STAGES_128xAUTO: ClassVar[MatmulStages] = ...
620
+ STAGES_16x1: ClassVar[MatmulStages] = ...
621
+ STAGES_16x10: ClassVar[MatmulStages] = ...
622
+ STAGES_16x2: ClassVar[MatmulStages] = ...
623
+ STAGES_16x3: ClassVar[MatmulStages] = ...
624
+ STAGES_16x4: ClassVar[MatmulStages] = ...
625
+ STAGES_16x5: ClassVar[MatmulStages] = ...
626
+ STAGES_16x6: ClassVar[MatmulStages] = ...
627
+ STAGES_16x80: ClassVar[MatmulStages] = ...
628
+ STAGES_16xAUTO: ClassVar[MatmulStages] = ...
629
+ STAGES_256xAUTO: ClassVar[MatmulStages] = ...
630
+ STAGES_32x1: ClassVar[MatmulStages] = ...
631
+ STAGES_32x10: ClassVar[MatmulStages] = ...
632
+ STAGES_32x2: ClassVar[MatmulStages] = ...
633
+ STAGES_32x3: ClassVar[MatmulStages] = ...
634
+ STAGES_32x4: ClassVar[MatmulStages] = ...
635
+ STAGES_32x5: ClassVar[MatmulStages] = ...
636
+ STAGES_32x6: ClassVar[MatmulStages] = ...
637
+ STAGES_32xAUTO: ClassVar[MatmulStages] = ...
638
+ STAGES_64x1: ClassVar[MatmulStages] = ...
639
+ STAGES_64x2: ClassVar[MatmulStages] = ...
640
+ STAGES_64x3: ClassVar[MatmulStages] = ...
641
+ STAGES_64x4: ClassVar[MatmulStages] = ...
642
+ STAGES_64x5: ClassVar[MatmulStages] = ...
643
+ STAGES_64x6: ClassVar[MatmulStages] = ...
644
+ STAGES_64x80: ClassVar[MatmulStages] = ...
645
+ STAGES_64xAUTO: ClassVar[MatmulStages] = ...
646
+ STAGES_768xAUTO: ClassVar[MatmulStages] = ...
647
+ STAGES_8x3: ClassVar[MatmulStages] = ...
648
+ STAGES_8x4: ClassVar[MatmulStages] = ...
649
+ STAGES_8x5: ClassVar[MatmulStages] = ...
650
+ STAGES_8xAUTO: ClassVar[MatmulStages] = ...
651
+ STAGES_UNDEFINED: ClassVar[MatmulStages] = ...
652
+ _generate_next_value_: ClassVar[Callable] = ...
653
+ _hashable_values_: ClassVar[list] = ...
654
+ _member_map_: ClassVar[dict] = ...
655
+ _member_names_: ClassVar[list] = ...
656
+ _member_type_: ClassVar[type[int]] = ...
657
+ _unhashable_values_: ClassVar[list] = ...
658
+ _unhashable_values_map_: ClassVar[dict] = ...
659
+ _use_args_: ClassVar[bool] = ...
660
+ _value2member_map_: ClassVar[dict] = ...
661
+ def __format__(self, *args, **kwargs) -> str:
662
+ """Convert to a string according to format_spec."""
663
+
664
+ class MatmulTile(enum.IntEnum):
665
+ """
666
+ See `cublasLtMatmulTile_t`.
667
+ """
668
+ __new__: ClassVar[Callable] = ...
669
+ TILE_104x128: ClassVar[MatmulTile] = ...
670
+ TILE_104x192: ClassVar[MatmulTile] = ...
671
+ TILE_104x256: ClassVar[MatmulTile] = ...
672
+ TILE_104x320: ClassVar[MatmulTile] = ...
673
+ TILE_104x384: ClassVar[MatmulTile] = ...
674
+ TILE_104x448: ClassVar[MatmulTile] = ...
675
+ TILE_104x64: ClassVar[MatmulTile] = ...
676
+ TILE_112x128: ClassVar[MatmulTile] = ...
677
+ TILE_112x192: ClassVar[MatmulTile] = ...
678
+ TILE_112x256: ClassVar[MatmulTile] = ...
679
+ TILE_112x320: ClassVar[MatmulTile] = ...
680
+ TILE_112x384: ClassVar[MatmulTile] = ...
681
+ TILE_112x64: ClassVar[MatmulTile] = ...
682
+ TILE_120x128: ClassVar[MatmulTile] = ...
683
+ TILE_120x192: ClassVar[MatmulTile] = ...
684
+ TILE_120x256: ClassVar[MatmulTile] = ...
685
+ TILE_120x320: ClassVar[MatmulTile] = ...
686
+ TILE_120x384: ClassVar[MatmulTile] = ...
687
+ TILE_120x64: ClassVar[MatmulTile] = ...
688
+ TILE_128x104: ClassVar[MatmulTile] = ...
689
+ TILE_128x112: ClassVar[MatmulTile] = ...
690
+ TILE_128x120: ClassVar[MatmulTile] = ...
691
+ TILE_128x128: ClassVar[MatmulTile] = ...
692
+ TILE_128x136: ClassVar[MatmulTile] = ...
693
+ TILE_128x144: ClassVar[MatmulTile] = ...
694
+ TILE_128x152: ClassVar[MatmulTile] = ...
695
+ TILE_128x16: ClassVar[MatmulTile] = ...
696
+ TILE_128x160: ClassVar[MatmulTile] = ...
697
+ TILE_128x168: ClassVar[MatmulTile] = ...
698
+ TILE_128x176: ClassVar[MatmulTile] = ...
699
+ TILE_128x184: ClassVar[MatmulTile] = ...
700
+ TILE_128x192: ClassVar[MatmulTile] = ...
701
+ TILE_128x200: ClassVar[MatmulTile] = ...
702
+ TILE_128x208: ClassVar[MatmulTile] = ...
703
+ TILE_128x216: ClassVar[MatmulTile] = ...
704
+ TILE_128x224: ClassVar[MatmulTile] = ...
705
+ TILE_128x232: ClassVar[MatmulTile] = ...
706
+ TILE_128x24: ClassVar[MatmulTile] = ...
707
+ TILE_128x240: ClassVar[MatmulTile] = ...
708
+ TILE_128x248: ClassVar[MatmulTile] = ...
709
+ TILE_128x256: ClassVar[MatmulTile] = ...
710
+ TILE_128x264: ClassVar[MatmulTile] = ...
711
+ TILE_128x272: ClassVar[MatmulTile] = ...
712
+ TILE_128x280: ClassVar[MatmulTile] = ...
713
+ TILE_128x288: ClassVar[MatmulTile] = ...
714
+ TILE_128x296: ClassVar[MatmulTile] = ...
715
+ TILE_128x304: ClassVar[MatmulTile] = ...
716
+ TILE_128x312: ClassVar[MatmulTile] = ...
717
+ TILE_128x32: ClassVar[MatmulTile] = ...
718
+ TILE_128x320: ClassVar[MatmulTile] = ...
719
+ TILE_128x328: ClassVar[MatmulTile] = ...
720
+ TILE_128x336: ClassVar[MatmulTile] = ...
721
+ TILE_128x344: ClassVar[MatmulTile] = ...
722
+ TILE_128x352: ClassVar[MatmulTile] = ...
723
+ TILE_128x360: ClassVar[MatmulTile] = ...
724
+ TILE_128x368: ClassVar[MatmulTile] = ...
725
+ TILE_128x376: ClassVar[MatmulTile] = ...
726
+ TILE_128x384: ClassVar[MatmulTile] = ...
727
+ TILE_128x392: ClassVar[MatmulTile] = ...
728
+ TILE_128x40: ClassVar[MatmulTile] = ...
729
+ TILE_128x400: ClassVar[MatmulTile] = ...
730
+ TILE_128x408: ClassVar[MatmulTile] = ...
731
+ TILE_128x416: ClassVar[MatmulTile] = ...
732
+ TILE_128x424: ClassVar[MatmulTile] = ...
733
+ TILE_128x432: ClassVar[MatmulTile] = ...
734
+ TILE_128x440: ClassVar[MatmulTile] = ...
735
+ TILE_128x448: ClassVar[MatmulTile] = ...
736
+ TILE_128x456: ClassVar[MatmulTile] = ...
737
+ TILE_128x464: ClassVar[MatmulTile] = ...
738
+ TILE_128x472: ClassVar[MatmulTile] = ...
739
+ TILE_128x48: ClassVar[MatmulTile] = ...
740
+ TILE_128x480: ClassVar[MatmulTile] = ...
741
+ TILE_128x488: ClassVar[MatmulTile] = ...
742
+ TILE_128x496: ClassVar[MatmulTile] = ...
743
+ TILE_128x504: ClassVar[MatmulTile] = ...
744
+ TILE_128x512: ClassVar[MatmulTile] = ...
745
+ TILE_128x56: ClassVar[MatmulTile] = ...
746
+ TILE_128x64: ClassVar[MatmulTile] = ...
747
+ TILE_128x72: ClassVar[MatmulTile] = ...
748
+ TILE_128x8: ClassVar[MatmulTile] = ...
749
+ TILE_128x80: ClassVar[MatmulTile] = ...
750
+ TILE_128x88: ClassVar[MatmulTile] = ...
751
+ TILE_128x96: ClassVar[MatmulTile] = ...
752
+ TILE_136x128: ClassVar[MatmulTile] = ...
753
+ TILE_136x192: ClassVar[MatmulTile] = ...
754
+ TILE_136x256: ClassVar[MatmulTile] = ...
755
+ TILE_136x320: ClassVar[MatmulTile] = ...
756
+ TILE_136x64: ClassVar[MatmulTile] = ...
757
+ TILE_144x128: ClassVar[MatmulTile] = ...
758
+ TILE_144x192: ClassVar[MatmulTile] = ...
759
+ TILE_144x256: ClassVar[MatmulTile] = ...
760
+ TILE_144x320: ClassVar[MatmulTile] = ...
761
+ TILE_144x64: ClassVar[MatmulTile] = ...
762
+ TILE_152x128: ClassVar[MatmulTile] = ...
763
+ TILE_152x192: ClassVar[MatmulTile] = ...
764
+ TILE_152x256: ClassVar[MatmulTile] = ...
765
+ TILE_152x320: ClassVar[MatmulTile] = ...
766
+ TILE_152x64: ClassVar[MatmulTile] = ...
767
+ TILE_160x128: ClassVar[MatmulTile] = ...
768
+ TILE_160x192: ClassVar[MatmulTile] = ...
769
+ TILE_160x256: ClassVar[MatmulTile] = ...
770
+ TILE_160x64: ClassVar[MatmulTile] = ...
771
+ TILE_168x128: ClassVar[MatmulTile] = ...
772
+ TILE_168x192: ClassVar[MatmulTile] = ...
773
+ TILE_168x256: ClassVar[MatmulTile] = ...
774
+ TILE_168x64: ClassVar[MatmulTile] = ...
775
+ TILE_16x128: ClassVar[MatmulTile] = ...
776
+ TILE_16x16: ClassVar[MatmulTile] = ...
777
+ TILE_16x192: ClassVar[MatmulTile] = ...
778
+ TILE_16x256: ClassVar[MatmulTile] = ...
779
+ TILE_16x32: ClassVar[MatmulTile] = ...
780
+ TILE_16x320: ClassVar[MatmulTile] = ...
781
+ TILE_16x384: ClassVar[MatmulTile] = ...
782
+ TILE_16x448: ClassVar[MatmulTile] = ...
783
+ TILE_16x512: ClassVar[MatmulTile] = ...
784
+ TILE_16x576: ClassVar[MatmulTile] = ...
785
+ TILE_16x64: ClassVar[MatmulTile] = ...
786
+ TILE_16x640: ClassVar[MatmulTile] = ...
787
+ TILE_16x704: ClassVar[MatmulTile] = ...
788
+ TILE_16x768: ClassVar[MatmulTile] = ...
789
+ TILE_16x8: ClassVar[MatmulTile] = ...
790
+ TILE_176x128: ClassVar[MatmulTile] = ...
791
+ TILE_176x192: ClassVar[MatmulTile] = ...
792
+ TILE_176x256: ClassVar[MatmulTile] = ...
793
+ TILE_176x64: ClassVar[MatmulTile] = ...
794
+ TILE_184x128: ClassVar[MatmulTile] = ...
795
+ TILE_184x192: ClassVar[MatmulTile] = ...
796
+ TILE_184x256: ClassVar[MatmulTile] = ...
797
+ TILE_184x64: ClassVar[MatmulTile] = ...
798
+ TILE_192x104: ClassVar[MatmulTile] = ...
799
+ TILE_192x112: ClassVar[MatmulTile] = ...
800
+ TILE_192x120: ClassVar[MatmulTile] = ...
801
+ TILE_192x128: ClassVar[MatmulTile] = ...
802
+ TILE_192x136: ClassVar[MatmulTile] = ...
803
+ TILE_192x144: ClassVar[MatmulTile] = ...
804
+ TILE_192x152: ClassVar[MatmulTile] = ...
805
+ TILE_192x16: ClassVar[MatmulTile] = ...
806
+ TILE_192x160: ClassVar[MatmulTile] = ...
807
+ TILE_192x168: ClassVar[MatmulTile] = ...
808
+ TILE_192x176: ClassVar[MatmulTile] = ...
809
+ TILE_192x184: ClassVar[MatmulTile] = ...
810
+ TILE_192x192: ClassVar[MatmulTile] = ...
811
+ TILE_192x200: ClassVar[MatmulTile] = ...
812
+ TILE_192x208: ClassVar[MatmulTile] = ...
813
+ TILE_192x216: ClassVar[MatmulTile] = ...
814
+ TILE_192x224: ClassVar[MatmulTile] = ...
815
+ TILE_192x232: ClassVar[MatmulTile] = ...
816
+ TILE_192x24: ClassVar[MatmulTile] = ...
817
+ TILE_192x240: ClassVar[MatmulTile] = ...
818
+ TILE_192x248: ClassVar[MatmulTile] = ...
819
+ TILE_192x256: ClassVar[MatmulTile] = ...
820
+ TILE_192x264: ClassVar[MatmulTile] = ...
821
+ TILE_192x272: ClassVar[MatmulTile] = ...
822
+ TILE_192x280: ClassVar[MatmulTile] = ...
823
+ TILE_192x288: ClassVar[MatmulTile] = ...
824
+ TILE_192x296: ClassVar[MatmulTile] = ...
825
+ TILE_192x304: ClassVar[MatmulTile] = ...
826
+ TILE_192x312: ClassVar[MatmulTile] = ...
827
+ TILE_192x32: ClassVar[MatmulTile] = ...
828
+ TILE_192x320: ClassVar[MatmulTile] = ...
829
+ TILE_192x328: ClassVar[MatmulTile] = ...
830
+ TILE_192x336: ClassVar[MatmulTile] = ...
831
+ TILE_192x40: ClassVar[MatmulTile] = ...
832
+ TILE_192x48: ClassVar[MatmulTile] = ...
833
+ TILE_192x56: ClassVar[MatmulTile] = ...
834
+ TILE_192x64: ClassVar[MatmulTile] = ...
835
+ TILE_192x72: ClassVar[MatmulTile] = ...
836
+ TILE_192x8: ClassVar[MatmulTile] = ...
837
+ TILE_192x80: ClassVar[MatmulTile] = ...
838
+ TILE_192x88: ClassVar[MatmulTile] = ...
839
+ TILE_192x96: ClassVar[MatmulTile] = ...
840
+ TILE_200x128: ClassVar[MatmulTile] = ...
841
+ TILE_200x192: ClassVar[MatmulTile] = ...
842
+ TILE_200x64: ClassVar[MatmulTile] = ...
843
+ TILE_208x128: ClassVar[MatmulTile] = ...
844
+ TILE_208x192: ClassVar[MatmulTile] = ...
845
+ TILE_208x64: ClassVar[MatmulTile] = ...
846
+ TILE_216x128: ClassVar[MatmulTile] = ...
847
+ TILE_216x192: ClassVar[MatmulTile] = ...
848
+ TILE_216x64: ClassVar[MatmulTile] = ...
849
+ TILE_224x128: ClassVar[MatmulTile] = ...
850
+ TILE_224x192: ClassVar[MatmulTile] = ...
851
+ TILE_224x64: ClassVar[MatmulTile] = ...
852
+ TILE_232x128: ClassVar[MatmulTile] = ...
853
+ TILE_232x192: ClassVar[MatmulTile] = ...
854
+ TILE_232x64: ClassVar[MatmulTile] = ...
855
+ TILE_240x128: ClassVar[MatmulTile] = ...
856
+ TILE_240x192: ClassVar[MatmulTile] = ...
857
+ TILE_240x64: ClassVar[MatmulTile] = ...
858
+ TILE_248x128: ClassVar[MatmulTile] = ...
859
+ TILE_248x192: ClassVar[MatmulTile] = ...
860
+ TILE_248x64: ClassVar[MatmulTile] = ...
861
+ TILE_24x128: ClassVar[MatmulTile] = ...
862
+ TILE_24x192: ClassVar[MatmulTile] = ...
863
+ TILE_24x256: ClassVar[MatmulTile] = ...
864
+ TILE_24x320: ClassVar[MatmulTile] = ...
865
+ TILE_24x384: ClassVar[MatmulTile] = ...
866
+ TILE_24x448: ClassVar[MatmulTile] = ...
867
+ TILE_24x512: ClassVar[MatmulTile] = ...
868
+ TILE_24x576: ClassVar[MatmulTile] = ...
869
+ TILE_24x64: ClassVar[MatmulTile] = ...
870
+ TILE_24x640: ClassVar[MatmulTile] = ...
871
+ TILE_24x704: ClassVar[MatmulTile] = ...
872
+ TILE_24x768: ClassVar[MatmulTile] = ...
873
+ TILE_256x1024: ClassVar[MatmulTile] = ...
874
+ TILE_256x104: ClassVar[MatmulTile] = ...
875
+ TILE_256x112: ClassVar[MatmulTile] = ...
876
+ TILE_256x120: ClassVar[MatmulTile] = ...
877
+ TILE_256x128: ClassVar[MatmulTile] = ...
878
+ TILE_256x136: ClassVar[MatmulTile] = ...
879
+ TILE_256x144: ClassVar[MatmulTile] = ...
880
+ TILE_256x152: ClassVar[MatmulTile] = ...
881
+ TILE_256x16: ClassVar[MatmulTile] = ...
882
+ TILE_256x160: ClassVar[MatmulTile] = ...
883
+ TILE_256x168: ClassVar[MatmulTile] = ...
884
+ TILE_256x176: ClassVar[MatmulTile] = ...
885
+ TILE_256x184: ClassVar[MatmulTile] = ...
886
+ TILE_256x192: ClassVar[MatmulTile] = ...
887
+ TILE_256x200: ClassVar[MatmulTile] = ...
888
+ TILE_256x208: ClassVar[MatmulTile] = ...
889
+ TILE_256x216: ClassVar[MatmulTile] = ...
890
+ TILE_256x224: ClassVar[MatmulTile] = ...
891
+ TILE_256x232: ClassVar[MatmulTile] = ...
892
+ TILE_256x24: ClassVar[MatmulTile] = ...
893
+ TILE_256x240: ClassVar[MatmulTile] = ...
894
+ TILE_256x248: ClassVar[MatmulTile] = ...
895
+ TILE_256x256: ClassVar[MatmulTile] = ...
896
+ TILE_256x32: ClassVar[MatmulTile] = ...
897
+ TILE_256x40: ClassVar[MatmulTile] = ...
898
+ TILE_256x48: ClassVar[MatmulTile] = ...
899
+ TILE_256x512: ClassVar[MatmulTile] = ...
900
+ TILE_256x56: ClassVar[MatmulTile] = ...
901
+ TILE_256x64: ClassVar[MatmulTile] = ...
902
+ TILE_256x72: ClassVar[MatmulTile] = ...
903
+ TILE_256x8: ClassVar[MatmulTile] = ...
904
+ TILE_256x80: ClassVar[MatmulTile] = ...
905
+ TILE_256x88: ClassVar[MatmulTile] = ...
906
+ TILE_256x96: ClassVar[MatmulTile] = ...
907
+ TILE_264x128: ClassVar[MatmulTile] = ...
908
+ TILE_264x64: ClassVar[MatmulTile] = ...
909
+ TILE_272x128: ClassVar[MatmulTile] = ...
910
+ TILE_272x64: ClassVar[MatmulTile] = ...
911
+ TILE_280x128: ClassVar[MatmulTile] = ...
912
+ TILE_280x64: ClassVar[MatmulTile] = ...
913
+ TILE_288x128: ClassVar[MatmulTile] = ...
914
+ TILE_288x64: ClassVar[MatmulTile] = ...
915
+ TILE_296x128: ClassVar[MatmulTile] = ...
916
+ TILE_296x64: ClassVar[MatmulTile] = ...
917
+ TILE_304x128: ClassVar[MatmulTile] = ...
918
+ TILE_304x64: ClassVar[MatmulTile] = ...
919
+ TILE_312x128: ClassVar[MatmulTile] = ...
920
+ TILE_312x64: ClassVar[MatmulTile] = ...
921
+ TILE_320x104: ClassVar[MatmulTile] = ...
922
+ TILE_320x112: ClassVar[MatmulTile] = ...
923
+ TILE_320x120: ClassVar[MatmulTile] = ...
924
+ TILE_320x128: ClassVar[MatmulTile] = ...
925
+ TILE_320x136: ClassVar[MatmulTile] = ...
926
+ TILE_320x144: ClassVar[MatmulTile] = ...
927
+ TILE_320x152: ClassVar[MatmulTile] = ...
928
+ TILE_320x16: ClassVar[MatmulTile] = ...
929
+ TILE_320x160: ClassVar[MatmulTile] = ...
930
+ TILE_320x168: ClassVar[MatmulTile] = ...
931
+ TILE_320x176: ClassVar[MatmulTile] = ...
932
+ TILE_320x184: ClassVar[MatmulTile] = ...
933
+ TILE_320x192: ClassVar[MatmulTile] = ...
934
+ TILE_320x200: ClassVar[MatmulTile] = ...
935
+ TILE_320x24: ClassVar[MatmulTile] = ...
936
+ TILE_320x32: ClassVar[MatmulTile] = ...
937
+ TILE_320x40: ClassVar[MatmulTile] = ...
938
+ TILE_320x48: ClassVar[MatmulTile] = ...
939
+ TILE_320x56: ClassVar[MatmulTile] = ...
940
+ TILE_320x64: ClassVar[MatmulTile] = ...
941
+ TILE_320x72: ClassVar[MatmulTile] = ...
942
+ TILE_320x8: ClassVar[MatmulTile] = ...
943
+ TILE_320x80: ClassVar[MatmulTile] = ...
944
+ TILE_320x88: ClassVar[MatmulTile] = ...
945
+ TILE_320x96: ClassVar[MatmulTile] = ...
946
+ TILE_328x128: ClassVar[MatmulTile] = ...
947
+ TILE_328x64: ClassVar[MatmulTile] = ...
948
+ TILE_32x128: ClassVar[MatmulTile] = ...
949
+ TILE_32x16: ClassVar[MatmulTile] = ...
950
+ TILE_32x192: ClassVar[MatmulTile] = ...
951
+ TILE_32x256: ClassVar[MatmulTile] = ...
952
+ TILE_32x32: ClassVar[MatmulTile] = ...
953
+ TILE_32x320: ClassVar[MatmulTile] = ...
954
+ TILE_32x384: ClassVar[MatmulTile] = ...
955
+ TILE_32x448: ClassVar[MatmulTile] = ...
956
+ TILE_32x512: ClassVar[MatmulTile] = ...
957
+ TILE_32x576: ClassVar[MatmulTile] = ...
958
+ TILE_32x64: ClassVar[MatmulTile] = ...
959
+ TILE_32x640: ClassVar[MatmulTile] = ...
960
+ TILE_32x704: ClassVar[MatmulTile] = ...
961
+ TILE_32x768: ClassVar[MatmulTile] = ...
962
+ TILE_32x8: ClassVar[MatmulTile] = ...
963
+ TILE_336x128: ClassVar[MatmulTile] = ...
964
+ TILE_336x64: ClassVar[MatmulTile] = ...
965
+ TILE_344x128: ClassVar[MatmulTile] = ...
966
+ TILE_344x64: ClassVar[MatmulTile] = ...
967
+ TILE_352x128: ClassVar[MatmulTile] = ...
968
+ TILE_352x64: ClassVar[MatmulTile] = ...
969
+ TILE_360x128: ClassVar[MatmulTile] = ...
970
+ TILE_360x64: ClassVar[MatmulTile] = ...
971
+ TILE_368x128: ClassVar[MatmulTile] = ...
972
+ TILE_368x64: ClassVar[MatmulTile] = ...
973
+ TILE_376x128: ClassVar[MatmulTile] = ...
974
+ TILE_376x64: ClassVar[MatmulTile] = ...
975
+ TILE_384x104: ClassVar[MatmulTile] = ...
976
+ TILE_384x112: ClassVar[MatmulTile] = ...
977
+ TILE_384x120: ClassVar[MatmulTile] = ...
978
+ TILE_384x128: ClassVar[MatmulTile] = ...
979
+ TILE_384x136: ClassVar[MatmulTile] = ...
980
+ TILE_384x144: ClassVar[MatmulTile] = ...
981
+ TILE_384x152: ClassVar[MatmulTile] = ...
982
+ TILE_384x16: ClassVar[MatmulTile] = ...
983
+ TILE_384x160: ClassVar[MatmulTile] = ...
984
+ TILE_384x168: ClassVar[MatmulTile] = ...
985
+ TILE_384x24: ClassVar[MatmulTile] = ...
986
+ TILE_384x32: ClassVar[MatmulTile] = ...
987
+ TILE_384x40: ClassVar[MatmulTile] = ...
988
+ TILE_384x48: ClassVar[MatmulTile] = ...
989
+ TILE_384x56: ClassVar[MatmulTile] = ...
990
+ TILE_384x64: ClassVar[MatmulTile] = ...
991
+ TILE_384x72: ClassVar[MatmulTile] = ...
992
+ TILE_384x8: ClassVar[MatmulTile] = ...
993
+ TILE_384x80: ClassVar[MatmulTile] = ...
994
+ TILE_384x88: ClassVar[MatmulTile] = ...
995
+ TILE_384x96: ClassVar[MatmulTile] = ...
996
+ TILE_392x64: ClassVar[MatmulTile] = ...
997
+ TILE_400x64: ClassVar[MatmulTile] = ...
998
+ TILE_408x64: ClassVar[MatmulTile] = ...
999
+ TILE_40x128: ClassVar[MatmulTile] = ...
1000
+ TILE_40x192: ClassVar[MatmulTile] = ...
1001
+ TILE_40x256: ClassVar[MatmulTile] = ...
1002
+ TILE_40x320: ClassVar[MatmulTile] = ...
1003
+ TILE_40x384: ClassVar[MatmulTile] = ...
1004
+ TILE_40x448: ClassVar[MatmulTile] = ...
1005
+ TILE_40x512: ClassVar[MatmulTile] = ...
1006
+ TILE_40x576: ClassVar[MatmulTile] = ...
1007
+ TILE_40x64: ClassVar[MatmulTile] = ...
1008
+ TILE_40x640: ClassVar[MatmulTile] = ...
1009
+ TILE_40x704: ClassVar[MatmulTile] = ...
1010
+ TILE_40x768: ClassVar[MatmulTile] = ...
1011
+ TILE_416x64: ClassVar[MatmulTile] = ...
1012
+ TILE_424x64: ClassVar[MatmulTile] = ...
1013
+ TILE_432x64: ClassVar[MatmulTile] = ...
1014
+ TILE_440x64: ClassVar[MatmulTile] = ...
1015
+ TILE_448x104: ClassVar[MatmulTile] = ...
1016
+ TILE_448x112: ClassVar[MatmulTile] = ...
1017
+ TILE_448x120: ClassVar[MatmulTile] = ...
1018
+ TILE_448x128: ClassVar[MatmulTile] = ...
1019
+ TILE_448x136: ClassVar[MatmulTile] = ...
1020
+ TILE_448x144: ClassVar[MatmulTile] = ...
1021
+ TILE_448x16: ClassVar[MatmulTile] = ...
1022
+ TILE_448x24: ClassVar[MatmulTile] = ...
1023
+ TILE_448x32: ClassVar[MatmulTile] = ...
1024
+ TILE_448x40: ClassVar[MatmulTile] = ...
1025
+ TILE_448x48: ClassVar[MatmulTile] = ...
1026
+ TILE_448x56: ClassVar[MatmulTile] = ...
1027
+ TILE_448x64: ClassVar[MatmulTile] = ...
1028
+ TILE_448x72: ClassVar[MatmulTile] = ...
1029
+ TILE_448x8: ClassVar[MatmulTile] = ...
1030
+ TILE_448x80: ClassVar[MatmulTile] = ...
1031
+ TILE_448x88: ClassVar[MatmulTile] = ...
1032
+ TILE_448x96: ClassVar[MatmulTile] = ...
1033
+ TILE_456x64: ClassVar[MatmulTile] = ...
1034
+ TILE_464x64: ClassVar[MatmulTile] = ...
1035
+ TILE_472x64: ClassVar[MatmulTile] = ...
1036
+ TILE_480x64: ClassVar[MatmulTile] = ...
1037
+ TILE_488x64: ClassVar[MatmulTile] = ...
1038
+ TILE_48x128: ClassVar[MatmulTile] = ...
1039
+ TILE_48x192: ClassVar[MatmulTile] = ...
1040
+ TILE_48x256: ClassVar[MatmulTile] = ...
1041
+ TILE_48x320: ClassVar[MatmulTile] = ...
1042
+ TILE_48x384: ClassVar[MatmulTile] = ...
1043
+ TILE_48x448: ClassVar[MatmulTile] = ...
1044
+ TILE_48x512: ClassVar[MatmulTile] = ...
1045
+ TILE_48x576: ClassVar[MatmulTile] = ...
1046
+ TILE_48x64: ClassVar[MatmulTile] = ...
1047
+ TILE_48x640: ClassVar[MatmulTile] = ...
1048
+ TILE_48x704: ClassVar[MatmulTile] = ...
1049
+ TILE_48x768: ClassVar[MatmulTile] = ...
1050
+ TILE_496x64: ClassVar[MatmulTile] = ...
1051
+ TILE_504x64: ClassVar[MatmulTile] = ...
1052
+ TILE_512x1024: ClassVar[MatmulTile] = ...
1053
+ TILE_512x104: ClassVar[MatmulTile] = ...
1054
+ TILE_512x112: ClassVar[MatmulTile] = ...
1055
+ TILE_512x120: ClassVar[MatmulTile] = ...
1056
+ TILE_512x128: ClassVar[MatmulTile] = ...
1057
+ TILE_512x16: ClassVar[MatmulTile] = ...
1058
+ TILE_512x24: ClassVar[MatmulTile] = ...
1059
+ TILE_512x32: ClassVar[MatmulTile] = ...
1060
+ TILE_512x40: ClassVar[MatmulTile] = ...
1061
+ TILE_512x48: ClassVar[MatmulTile] = ...
1062
+ TILE_512x512: ClassVar[MatmulTile] = ...
1063
+ TILE_512x56: ClassVar[MatmulTile] = ...
1064
+ TILE_512x64: ClassVar[MatmulTile] = ...
1065
+ TILE_512x72: ClassVar[MatmulTile] = ...
1066
+ TILE_512x8: ClassVar[MatmulTile] = ...
1067
+ TILE_512x80: ClassVar[MatmulTile] = ...
1068
+ TILE_512x88: ClassVar[MatmulTile] = ...
1069
+ TILE_512x96: ClassVar[MatmulTile] = ...
1070
+ TILE_520x64: ClassVar[MatmulTile] = ...
1071
+ TILE_528x64: ClassVar[MatmulTile] = ...
1072
+ TILE_536x64: ClassVar[MatmulTile] = ...
1073
+ TILE_544x64: ClassVar[MatmulTile] = ...
1074
+ TILE_552x64: ClassVar[MatmulTile] = ...
1075
+ TILE_560x64: ClassVar[MatmulTile] = ...
1076
+ TILE_568x64: ClassVar[MatmulTile] = ...
1077
+ TILE_56x128: ClassVar[MatmulTile] = ...
1078
+ TILE_56x192: ClassVar[MatmulTile] = ...
1079
+ TILE_56x256: ClassVar[MatmulTile] = ...
1080
+ TILE_56x320: ClassVar[MatmulTile] = ...
1081
+ TILE_56x384: ClassVar[MatmulTile] = ...
1082
+ TILE_56x448: ClassVar[MatmulTile] = ...
1083
+ TILE_56x512: ClassVar[MatmulTile] = ...
1084
+ TILE_56x576: ClassVar[MatmulTile] = ...
1085
+ TILE_56x64: ClassVar[MatmulTile] = ...
1086
+ TILE_56x640: ClassVar[MatmulTile] = ...
1087
+ TILE_56x704: ClassVar[MatmulTile] = ...
1088
+ TILE_56x768: ClassVar[MatmulTile] = ...
1089
+ TILE_576x104: ClassVar[MatmulTile] = ...
1090
+ TILE_576x112: ClassVar[MatmulTile] = ...
1091
+ TILE_576x16: ClassVar[MatmulTile] = ...
1092
+ TILE_576x24: ClassVar[MatmulTile] = ...
1093
+ TILE_576x32: ClassVar[MatmulTile] = ...
1094
+ TILE_576x40: ClassVar[MatmulTile] = ...
1095
+ TILE_576x48: ClassVar[MatmulTile] = ...
1096
+ TILE_576x56: ClassVar[MatmulTile] = ...
1097
+ TILE_576x64: ClassVar[MatmulTile] = ...
1098
+ TILE_576x72: ClassVar[MatmulTile] = ...
1099
+ TILE_576x8: ClassVar[MatmulTile] = ...
1100
+ TILE_576x80: ClassVar[MatmulTile] = ...
1101
+ TILE_576x88: ClassVar[MatmulTile] = ...
1102
+ TILE_576x96: ClassVar[MatmulTile] = ...
1103
+ TILE_584x64: ClassVar[MatmulTile] = ...
1104
+ TILE_592x64: ClassVar[MatmulTile] = ...
1105
+ TILE_600x64: ClassVar[MatmulTile] = ...
1106
+ TILE_608x64: ClassVar[MatmulTile] = ...
1107
+ TILE_616x64: ClassVar[MatmulTile] = ...
1108
+ TILE_624x64: ClassVar[MatmulTile] = ...
1109
+ TILE_632x64: ClassVar[MatmulTile] = ...
1110
+ TILE_640x16: ClassVar[MatmulTile] = ...
1111
+ TILE_640x24: ClassVar[MatmulTile] = ...
1112
+ TILE_640x32: ClassVar[MatmulTile] = ...
1113
+ TILE_640x40: ClassVar[MatmulTile] = ...
1114
+ TILE_640x48: ClassVar[MatmulTile] = ...
1115
+ TILE_640x56: ClassVar[MatmulTile] = ...
1116
+ TILE_640x64: ClassVar[MatmulTile] = ...
1117
+ TILE_640x72: ClassVar[MatmulTile] = ...
1118
+ TILE_640x8: ClassVar[MatmulTile] = ...
1119
+ TILE_640x80: ClassVar[MatmulTile] = ...
1120
+ TILE_640x88: ClassVar[MatmulTile] = ...
1121
+ TILE_640x96: ClassVar[MatmulTile] = ...
1122
+ TILE_648x64: ClassVar[MatmulTile] = ...
1123
+ TILE_64x104: ClassVar[MatmulTile] = ...
1124
+ TILE_64x112: ClassVar[MatmulTile] = ...
1125
+ TILE_64x120: ClassVar[MatmulTile] = ...
1126
+ TILE_64x128: ClassVar[MatmulTile] = ...
1127
+ TILE_64x136: ClassVar[MatmulTile] = ...
1128
+ TILE_64x144: ClassVar[MatmulTile] = ...
1129
+ TILE_64x152: ClassVar[MatmulTile] = ...
1130
+ TILE_64x16: ClassVar[MatmulTile] = ...
1131
+ TILE_64x160: ClassVar[MatmulTile] = ...
1132
+ TILE_64x168: ClassVar[MatmulTile] = ...
1133
+ TILE_64x176: ClassVar[MatmulTile] = ...
1134
+ TILE_64x184: ClassVar[MatmulTile] = ...
1135
+ TILE_64x192: ClassVar[MatmulTile] = ...
1136
+ TILE_64x200: ClassVar[MatmulTile] = ...
1137
+ TILE_64x208: ClassVar[MatmulTile] = ...
1138
+ TILE_64x216: ClassVar[MatmulTile] = ...
1139
+ TILE_64x224: ClassVar[MatmulTile] = ...
1140
+ TILE_64x232: ClassVar[MatmulTile] = ...
1141
+ TILE_64x24: ClassVar[MatmulTile] = ...
1142
+ TILE_64x240: ClassVar[MatmulTile] = ...
1143
+ TILE_64x248: ClassVar[MatmulTile] = ...
1144
+ TILE_64x256: ClassVar[MatmulTile] = ...
1145
+ TILE_64x264: ClassVar[MatmulTile] = ...
1146
+ TILE_64x272: ClassVar[MatmulTile] = ...
1147
+ TILE_64x280: ClassVar[MatmulTile] = ...
1148
+ TILE_64x288: ClassVar[MatmulTile] = ...
1149
+ TILE_64x296: ClassVar[MatmulTile] = ...
1150
+ TILE_64x304: ClassVar[MatmulTile] = ...
1151
+ TILE_64x312: ClassVar[MatmulTile] = ...
1152
+ TILE_64x32: ClassVar[MatmulTile] = ...
1153
+ TILE_64x320: ClassVar[MatmulTile] = ...
1154
+ TILE_64x328: ClassVar[MatmulTile] = ...
1155
+ TILE_64x336: ClassVar[MatmulTile] = ...
1156
+ TILE_64x344: ClassVar[MatmulTile] = ...
1157
+ TILE_64x352: ClassVar[MatmulTile] = ...
1158
+ TILE_64x360: ClassVar[MatmulTile] = ...
1159
+ TILE_64x368: ClassVar[MatmulTile] = ...
1160
+ TILE_64x376: ClassVar[MatmulTile] = ...
1161
+ TILE_64x384: ClassVar[MatmulTile] = ...
1162
+ TILE_64x392: ClassVar[MatmulTile] = ...
1163
+ TILE_64x40: ClassVar[MatmulTile] = ...
1164
+ TILE_64x400: ClassVar[MatmulTile] = ...
1165
+ TILE_64x408: ClassVar[MatmulTile] = ...
1166
+ TILE_64x416: ClassVar[MatmulTile] = ...
1167
+ TILE_64x424: ClassVar[MatmulTile] = ...
1168
+ TILE_64x432: ClassVar[MatmulTile] = ...
1169
+ TILE_64x440: ClassVar[MatmulTile] = ...
1170
+ TILE_64x448: ClassVar[MatmulTile] = ...
1171
+ TILE_64x456: ClassVar[MatmulTile] = ...
1172
+ TILE_64x464: ClassVar[MatmulTile] = ...
1173
+ TILE_64x472: ClassVar[MatmulTile] = ...
1174
+ TILE_64x48: ClassVar[MatmulTile] = ...
1175
+ TILE_64x480: ClassVar[MatmulTile] = ...
1176
+ TILE_64x488: ClassVar[MatmulTile] = ...
1177
+ TILE_64x496: ClassVar[MatmulTile] = ...
1178
+ TILE_64x504: ClassVar[MatmulTile] = ...
1179
+ TILE_64x512: ClassVar[MatmulTile] = ...
1180
+ TILE_64x520: ClassVar[MatmulTile] = ...
1181
+ TILE_64x528: ClassVar[MatmulTile] = ...
1182
+ TILE_64x536: ClassVar[MatmulTile] = ...
1183
+ TILE_64x544: ClassVar[MatmulTile] = ...
1184
+ TILE_64x552: ClassVar[MatmulTile] = ...
1185
+ TILE_64x56: ClassVar[MatmulTile] = ...
1186
+ TILE_64x560: ClassVar[MatmulTile] = ...
1187
+ TILE_64x568: ClassVar[MatmulTile] = ...
1188
+ TILE_64x576: ClassVar[MatmulTile] = ...
1189
+ TILE_64x584: ClassVar[MatmulTile] = ...
1190
+ TILE_64x592: ClassVar[MatmulTile] = ...
1191
+ TILE_64x600: ClassVar[MatmulTile] = ...
1192
+ TILE_64x608: ClassVar[MatmulTile] = ...
1193
+ TILE_64x616: ClassVar[MatmulTile] = ...
1194
+ TILE_64x624: ClassVar[MatmulTile] = ...
1195
+ TILE_64x632: ClassVar[MatmulTile] = ...
1196
+ TILE_64x64: ClassVar[MatmulTile] = ...
1197
+ TILE_64x640: ClassVar[MatmulTile] = ...
1198
+ TILE_64x648: ClassVar[MatmulTile] = ...
1199
+ TILE_64x656: ClassVar[MatmulTile] = ...
1200
+ TILE_64x664: ClassVar[MatmulTile] = ...
1201
+ TILE_64x672: ClassVar[MatmulTile] = ...
1202
+ TILE_64x680: ClassVar[MatmulTile] = ...
1203
+ TILE_64x688: ClassVar[MatmulTile] = ...
1204
+ TILE_64x696: ClassVar[MatmulTile] = ...
1205
+ TILE_64x704: ClassVar[MatmulTile] = ...
1206
+ TILE_64x712: ClassVar[MatmulTile] = ...
1207
+ TILE_64x72: ClassVar[MatmulTile] = ...
1208
+ TILE_64x720: ClassVar[MatmulTile] = ...
1209
+ TILE_64x728: ClassVar[MatmulTile] = ...
1210
+ TILE_64x736: ClassVar[MatmulTile] = ...
1211
+ TILE_64x744: ClassVar[MatmulTile] = ...
1212
+ TILE_64x752: ClassVar[MatmulTile] = ...
1213
+ TILE_64x760: ClassVar[MatmulTile] = ...
1214
+ TILE_64x768: ClassVar[MatmulTile] = ...
1215
+ TILE_64x8: ClassVar[MatmulTile] = ...
1216
+ TILE_64x80: ClassVar[MatmulTile] = ...
1217
+ TILE_64x88: ClassVar[MatmulTile] = ...
1218
+ TILE_64x96: ClassVar[MatmulTile] = ...
1219
+ TILE_656x64: ClassVar[MatmulTile] = ...
1220
+ TILE_664x64: ClassVar[MatmulTile] = ...
1221
+ TILE_672x64: ClassVar[MatmulTile] = ...
1222
+ TILE_680x64: ClassVar[MatmulTile] = ...
1223
+ TILE_688x64: ClassVar[MatmulTile] = ...
1224
+ TILE_696x64: ClassVar[MatmulTile] = ...
1225
+ TILE_704x16: ClassVar[MatmulTile] = ...
1226
+ TILE_704x24: ClassVar[MatmulTile] = ...
1227
+ TILE_704x32: ClassVar[MatmulTile] = ...
1228
+ TILE_704x40: ClassVar[MatmulTile] = ...
1229
+ TILE_704x48: ClassVar[MatmulTile] = ...
1230
+ TILE_704x56: ClassVar[MatmulTile] = ...
1231
+ TILE_704x64: ClassVar[MatmulTile] = ...
1232
+ TILE_704x72: ClassVar[MatmulTile] = ...
1233
+ TILE_704x8: ClassVar[MatmulTile] = ...
1234
+ TILE_704x80: ClassVar[MatmulTile] = ...
1235
+ TILE_704x88: ClassVar[MatmulTile] = ...
1236
+ TILE_712x64: ClassVar[MatmulTile] = ...
1237
+ TILE_720x64: ClassVar[MatmulTile] = ...
1238
+ TILE_728x64: ClassVar[MatmulTile] = ...
1239
+ TILE_72x128: ClassVar[MatmulTile] = ...
1240
+ TILE_72x192: ClassVar[MatmulTile] = ...
1241
+ TILE_72x256: ClassVar[MatmulTile] = ...
1242
+ TILE_72x320: ClassVar[MatmulTile] = ...
1243
+ TILE_72x384: ClassVar[MatmulTile] = ...
1244
+ TILE_72x448: ClassVar[MatmulTile] = ...
1245
+ TILE_72x512: ClassVar[MatmulTile] = ...
1246
+ TILE_72x576: ClassVar[MatmulTile] = ...
1247
+ TILE_72x64: ClassVar[MatmulTile] = ...
1248
+ TILE_72x640: ClassVar[MatmulTile] = ...
1249
+ TILE_736x64: ClassVar[MatmulTile] = ...
1250
+ TILE_744x64: ClassVar[MatmulTile] = ...
1251
+ TILE_752x64: ClassVar[MatmulTile] = ...
1252
+ TILE_760x64: ClassVar[MatmulTile] = ...
1253
+ TILE_768x16: ClassVar[MatmulTile] = ...
1254
+ TILE_768x24: ClassVar[MatmulTile] = ...
1255
+ TILE_768x32: ClassVar[MatmulTile] = ...
1256
+ TILE_768x40: ClassVar[MatmulTile] = ...
1257
+ TILE_768x48: ClassVar[MatmulTile] = ...
1258
+ TILE_768x56: ClassVar[MatmulTile] = ...
1259
+ TILE_768x64: ClassVar[MatmulTile] = ...
1260
+ TILE_768x72: ClassVar[MatmulTile] = ...
1261
+ TILE_768x8: ClassVar[MatmulTile] = ...
1262
+ TILE_768x80: ClassVar[MatmulTile] = ...
1263
+ TILE_80x128: ClassVar[MatmulTile] = ...
1264
+ TILE_80x192: ClassVar[MatmulTile] = ...
1265
+ TILE_80x256: ClassVar[MatmulTile] = ...
1266
+ TILE_80x320: ClassVar[MatmulTile] = ...
1267
+ TILE_80x384: ClassVar[MatmulTile] = ...
1268
+ TILE_80x448: ClassVar[MatmulTile] = ...
1269
+ TILE_80x512: ClassVar[MatmulTile] = ...
1270
+ TILE_80x576: ClassVar[MatmulTile] = ...
1271
+ TILE_80x64: ClassVar[MatmulTile] = ...
1272
+ TILE_88x128: ClassVar[MatmulTile] = ...
1273
+ TILE_88x192: ClassVar[MatmulTile] = ...
1274
+ TILE_88x256: ClassVar[MatmulTile] = ...
1275
+ TILE_88x320: ClassVar[MatmulTile] = ...
1276
+ TILE_88x384: ClassVar[MatmulTile] = ...
1277
+ TILE_88x448: ClassVar[MatmulTile] = ...
1278
+ TILE_88x512: ClassVar[MatmulTile] = ...
1279
+ TILE_88x64: ClassVar[MatmulTile] = ...
1280
+ TILE_8x128: ClassVar[MatmulTile] = ...
1281
+ TILE_8x16: ClassVar[MatmulTile] = ...
1282
+ TILE_8x192: ClassVar[MatmulTile] = ...
1283
+ TILE_8x256: ClassVar[MatmulTile] = ...
1284
+ TILE_8x32: ClassVar[MatmulTile] = ...
1285
+ TILE_8x320: ClassVar[MatmulTile] = ...
1286
+ TILE_8x384: ClassVar[MatmulTile] = ...
1287
+ TILE_8x448: ClassVar[MatmulTile] = ...
1288
+ TILE_8x512: ClassVar[MatmulTile] = ...
1289
+ TILE_8x576: ClassVar[MatmulTile] = ...
1290
+ TILE_8x64: ClassVar[MatmulTile] = ...
1291
+ TILE_8x640: ClassVar[MatmulTile] = ...
1292
+ TILE_8x704: ClassVar[MatmulTile] = ...
1293
+ TILE_8x768: ClassVar[MatmulTile] = ...
1294
+ TILE_8x8: ClassVar[MatmulTile] = ...
1295
+ TILE_96x128: ClassVar[MatmulTile] = ...
1296
+ TILE_96x192: ClassVar[MatmulTile] = ...
1297
+ TILE_96x256: ClassVar[MatmulTile] = ...
1298
+ TILE_96x320: ClassVar[MatmulTile] = ...
1299
+ TILE_96x384: ClassVar[MatmulTile] = ...
1300
+ TILE_96x448: ClassVar[MatmulTile] = ...
1301
+ TILE_96x512: ClassVar[MatmulTile] = ...
1302
+ TILE_96x64: ClassVar[MatmulTile] = ...
1303
+ TILE_UNDEFINED: ClassVar[MatmulTile] = ...
1304
+ _generate_next_value_: ClassVar[Callable] = ...
1305
+ _hashable_values_: ClassVar[list] = ...
1306
+ _member_map_: ClassVar[dict] = ...
1307
+ _member_names_: ClassVar[list] = ...
1308
+ _member_type_: ClassVar[type[int]] = ...
1309
+ _unhashable_values_: ClassVar[list] = ...
1310
+ _unhashable_values_map_: ClassVar[dict] = ...
1311
+ _use_args_: ClassVar[bool] = ...
1312
+ _value2member_map_: ClassVar[dict] = ...
1313
+ def __format__(self, *args, **kwargs) -> str:
1314
+ """Convert to a string according to format_spec."""
1315
+
1316
+ class MatrixLayoutAttribute(enum.IntEnum):
1317
+ """
1318
+ See `cublasLtMatrixLayoutAttribute_t`.
1319
+ """
1320
+ __new__: ClassVar[Callable] = ...
1321
+ BATCH_COUNT: ClassVar[MatrixLayoutAttribute] = ...
1322
+ BATCH_MODE: ClassVar[MatrixLayoutAttribute] = ...
1323
+ COLS: ClassVar[MatrixLayoutAttribute] = ...
1324
+ GROUPED_COLS_ARRAY: ClassVar[MatrixLayoutAttribute] = ...
1325
+ GROUPED_LD_ARRAY: ClassVar[MatrixLayoutAttribute] = ...
1326
+ GROUPED_LD_ARRAY_INTEGER_WIDTH: ClassVar[MatrixLayoutAttribute] = ...
1327
+ GROUPED_ROWS_ARRAY: ClassVar[MatrixLayoutAttribute] = ...
1328
+ GROUPED_ROWS_COLS_ARRAY_INTEGER_WIDTH: ClassVar[MatrixLayoutAttribute] = ...
1329
+ LD: ClassVar[MatrixLayoutAttribute] = ...
1330
+ ORDER: ClassVar[MatrixLayoutAttribute] = ...
1331
+ PLANE_OFFSET: ClassVar[MatrixLayoutAttribute] = ...
1332
+ ROWS: ClassVar[MatrixLayoutAttribute] = ...
1333
+ STRIDED_BATCH_OFFSET: ClassVar[MatrixLayoutAttribute] = ...
1334
+ TYPE: ClassVar[MatrixLayoutAttribute] = ...
1335
+ _generate_next_value_: ClassVar[Callable] = ...
1336
+ _hashable_values_: ClassVar[list] = ...
1337
+ _member_map_: ClassVar[dict] = ...
1338
+ _member_names_: ClassVar[list] = ...
1339
+ _member_type_: ClassVar[type[int]] = ...
1340
+ _unhashable_values_: ClassVar[list] = ...
1341
+ _unhashable_values_map_: ClassVar[dict] = ...
1342
+ _use_args_: ClassVar[bool] = ...
1343
+ _value2member_map_: ClassVar[dict] = ...
1344
+ def __format__(self, *args, **kwargs) -> str:
1345
+ """Convert to a string according to format_spec."""
1346
+
1347
+ class MatrixTransformDescAttribute(enum.IntEnum):
1348
+ """
1349
+ See `cublasLtMatrixTransformDescAttributes_t`.
1350
+ """
1351
+ __new__: ClassVar[Callable] = ...
1352
+ POINTER_MODE: ClassVar[MatrixTransformDescAttribute] = ...
1353
+ SCALE_TYPE: ClassVar[MatrixTransformDescAttribute] = ...
1354
+ TRANSA: ClassVar[MatrixTransformDescAttribute] = ...
1355
+ TRANSB: ClassVar[MatrixTransformDescAttribute] = ...
1356
+ _generate_next_value_: ClassVar[Callable] = ...
1357
+ _hashable_values_: ClassVar[list] = ...
1358
+ _member_map_: ClassVar[dict] = ...
1359
+ _member_names_: ClassVar[list] = ...
1360
+ _member_type_: ClassVar[type[int]] = ...
1361
+ _unhashable_values_: ClassVar[list] = ...
1362
+ _unhashable_values_map_: ClassVar[dict] = ...
1363
+ _use_args_: ClassVar[bool] = ...
1364
+ _value2member_map_: ClassVar[dict] = ...
1365
+ def __format__(self, *args, **kwargs) -> str:
1366
+ """Convert to a string according to format_spec."""
1367
+
1368
+ class Order(enum.IntEnum):
1369
+ """
1370
+ See `cublasLtOrder_t`.
1371
+ """
1372
+ __new__: ClassVar[Callable] = ...
1373
+ COL: ClassVar[Order] = ...
1374
+ COL32: ClassVar[Order] = ...
1375
+ COL32_2R_4R4: ClassVar[Order] = ...
1376
+ COL4_4R2_8C: ClassVar[Order] = ...
1377
+ ROW: ClassVar[Order] = ...
1378
+ _generate_next_value_: ClassVar[Callable] = ...
1379
+ _hashable_values_: ClassVar[list] = ...
1380
+ _member_map_: ClassVar[dict] = ...
1381
+ _member_names_: ClassVar[list] = ...
1382
+ _member_type_: ClassVar[type[int]] = ...
1383
+ _unhashable_values_: ClassVar[list] = ...
1384
+ _unhashable_values_map_: ClassVar[dict] = ...
1385
+ _use_args_: ClassVar[bool] = ...
1386
+ _value2member_map_: ClassVar[dict] = ...
1387
+ def __format__(self, *args, **kwargs) -> str:
1388
+ """Convert to a string according to format_spec."""
1389
+
1390
+ class PointerMode(enum.IntEnum):
1391
+ """
1392
+ See `cublasLtPointerMode_t`.
1393
+ """
1394
+ __new__: ClassVar[Callable] = ...
1395
+ ALPHA_DEVICE_VECTOR_BETA_HOST: ClassVar[PointerMode] = ...
1396
+ ALPHA_DEVICE_VECTOR_BETA_ZERO: ClassVar[PointerMode] = ...
1397
+ DEVICE: ClassVar[PointerMode] = ...
1398
+ DEVICE_VECTOR: ClassVar[PointerMode] = ...
1399
+ HOST: ClassVar[PointerMode] = ...
1400
+ _generate_next_value_: ClassVar[Callable] = ...
1401
+ _hashable_values_: ClassVar[list] = ...
1402
+ _member_map_: ClassVar[dict] = ...
1403
+ _member_names_: ClassVar[list] = ...
1404
+ _member_type_: ClassVar[type[int]] = ...
1405
+ _unhashable_values_: ClassVar[list] = ...
1406
+ _unhashable_values_map_: ClassVar[dict] = ...
1407
+ _use_args_: ClassVar[bool] = ...
1408
+ _value2member_map_: ClassVar[dict] = ...
1409
+ def __format__(self, *args, **kwargs) -> str:
1410
+ """Convert to a string according to format_spec."""
1411
+
1412
+ class PointerModeMask(enum.IntEnum):
1413
+ """
1414
+ See `cublasLtPointerModeMask_t`.
1415
+ """
1416
+ __new__: ClassVar[Callable] = ...
1417
+ ALPHA_DEVICE_VECTOR_BETA_HOST: ClassVar[PointerModeMask] = ...
1418
+ ALPHA_DEVICE_VECTOR_BETA_ZERO: ClassVar[PointerModeMask] = ...
1419
+ DEVICE: ClassVar[PointerModeMask] = ...
1420
+ DEVICE_VECTOR: ClassVar[PointerModeMask] = ...
1421
+ HOST: ClassVar[PointerModeMask] = ...
1422
+ NO_FILTERING: ClassVar[PointerModeMask] = ...
1423
+ _generate_next_value_: ClassVar[Callable] = ...
1424
+ _hashable_values_: ClassVar[list] = ...
1425
+ _member_map_: ClassVar[dict] = ...
1426
+ _member_names_: ClassVar[list] = ...
1427
+ _member_type_: ClassVar[type[int]] = ...
1428
+ _unhashable_values_: ClassVar[list] = ...
1429
+ _unhashable_values_map_: ClassVar[dict] = ...
1430
+ _use_args_: ClassVar[bool] = ...
1431
+ _value2member_map_: ClassVar[dict] = ...
1432
+ def __format__(self, *args, **kwargs) -> str:
1433
+ """Convert to a string according to format_spec."""
1434
+
1435
+ class ReductionScheme(enum.IntEnum):
1436
+ """
1437
+ See `cublasLtReductionScheme_t`.
1438
+ """
1439
+ __new__: ClassVar[Callable] = ...
1440
+ COMPUTE_TYPE: ClassVar[ReductionScheme] = ...
1441
+ INPLACE: ClassVar[ReductionScheme] = ...
1442
+ MASK: ClassVar[ReductionScheme] = ...
1443
+ NONE: ClassVar[ReductionScheme] = ...
1444
+ OUTPUT_TYPE: ClassVar[ReductionScheme] = ...
1445
+ _generate_next_value_: ClassVar[Callable] = ...
1446
+ _hashable_values_: ClassVar[list] = ...
1447
+ _member_map_: ClassVar[dict] = ...
1448
+ _member_names_: ClassVar[list] = ...
1449
+ _member_type_: ClassVar[type[int]] = ...
1450
+ _unhashable_values_: ClassVar[list] = ...
1451
+ _unhashable_values_map_: ClassVar[dict] = ...
1452
+ _use_args_: ClassVar[bool] = ...
1453
+ _value2member_map_: ClassVar[dict] = ...
1454
+ def __format__(self, *args, **kwargs) -> str:
1455
+ """Convert to a string according to format_spec."""
1456
+
1457
+ class cuBLASLtError(Exception):
1458
+ def __init__(self, status) -> Any:
1459
+ """cuBLASLtError.__init__(self, status)"""
1460
+ def __reduce__(self) -> Any:
1461
+ """cuBLASLtError.__reduce__(self)"""