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