/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipsparselt/checkouts/develop/library/include/hipsparselt.h Source File

/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipsparselt/checkouts/develop/library/include/hipsparselt.h Source File#

hipSPARSELt: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipsparselt/checkouts/develop/library/include/hipsparselt.h Source File
hipsparselt.h
Go to the documentation of this file.
1 /*******************************************************************************
2  *
3  * MIT License
4  *
5  * Copyright (c) 2022-2024 Advanced Micro Devices, Inc.
6  *
7  * Permission is hereby granted, free of charge, to any person obtaining a copy
8  * of this software and associated documentation files (the "Software"), to deal
9  * in the Software without restriction, including without limitation the rights
10  * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
11  * copies of the Software, and to permit persons to whom the Software is
12  * furnished to do so, subject to the following conditions:
13  *
14  * The above copyright notice and this permission notice shall be included in
15  * all copies or substantial portions of the Software.
16  *
17  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
18  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
19  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
20  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
21  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
22  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
23  * SOFTWARE.
24  *
25  *******************************************************************************/
26 
40 //
41 
67 #pragma once
68 #ifndef _HIPSPARSELT_H_
69 #define _HIPSPARSELT_H_
70 
71 #include <hipsparse/hipsparse.h>
72 #include <hipsparselt/hipsparselt-export.h>
73 #include <hipsparselt/hipsparselt-version.h>
74 
75 #include <hip/hip_complex.h>
76 #include <hip/hip_runtime_api.h>
77 
78 #if defined(__HIP_PLATFORM_AMD__)
79 #include <hip/hip_bfloat16.h>
80 #include <hip/hip_fp16.h>
81 #include <hip/library_types.h>
82 #else
83 #include <cuda_bf16.h>
84 #include <cuda_fp16.h>
85 #endif
86 
87 /* Opaque structures holding information */
88 // clang-format off
89 
90 #if defined(__HIP_PLATFORM_AMD__)
100 typedef struct hipsparseLtHandle_t {uint8_t data[11024];} hipsparseLtHandle_t;
101 
112 
122 
130 
140 typedef struct hipsparseLtMatmulPlan_t {uint8_t data[11024];} hipsparseLtMatmulPlan_t;
141 #elif defined(__HIP_PLATFORM_NVIDIA__)
142 typedef __nv_bfloat16 hip_bfloat16;
143 typedef struct {uint8_t data[11024];} hipsparseLtHandle_t;
144 typedef struct {uint8_t data[11024];} hipsparseLtMatDescriptor_t;
145 typedef struct {uint8_t data[11024];} hipsparseLtMatmulDescriptor_t;
146 typedef struct {uint8_t data[11024];} hipsparseLtMatmulAlgSelection_t;
147 typedef struct {uint8_t data[11024];} hipsparseLtMatmulPlan_t;
148 #endif
149 
150 
151 /* Types definitions */
161 typedef enum
162 {
170 
179 typedef enum {
183 
191 typedef enum {
195 
201 typedef enum {
210 
218 typedef enum {
241 
248 typedef enum {
251 
259 typedef enum {
267 
274 typedef enum {
278 
285 typedef enum {
289 
290 // clang-format on
291 
292 #ifdef __cplusplus
293 extern "C" {
294 #endif
295 
304 HIPSPARSELT_EXPORT
306 
321 HIPSPARSELT_EXPORT
322 hipsparseStatus_t hipsparseLtGetVersion(const hipsparseLtHandle_t* handle, int* version);
323 
337 HIPSPARSELT_EXPORT
338 hipsparseStatus_t hipsparseLtGetProperty(hipLibraryPropertyType propertyType, int* value);
339 
340 HIPSPARSELT_EXPORT
341 hipsparseStatus_t hipsparseLtGetGitRevision(hipsparseLtHandle_t handle, char* rev);
342 
343 HIPSPARSELT_EXPORT
344 hipsparseStatus_t hipsparseLtGetArchName(char** archName);
345 
361 /* hipSPARSE initialization and management routines */
362 HIPSPARSELT_EXPORT
363 hipsparseStatus_t hipsparseLtInit(hipsparseLtHandle_t* handle);
364 
378 HIPSPARSELT_EXPORT
379 hipsparseStatus_t hipsparseLtDestroy(const hipsparseLtHandle_t* handle);
380 
381 /* matrix descriptor */
409 HIPSPARSELT_EXPORT
410 hipsparseStatus_t hipsparseLtDenseDescriptorInit(const hipsparseLtHandle_t* handle,
411  hipsparseLtMatDescriptor_t* matDescr,
412  int64_t rows,
413  int64_t cols,
414  int64_t ld,
415  uint32_t alignment,
416  hipDataType valueType,
417  hipsparseOrder_t order);
418 
449 HIPSPARSELT_EXPORT
451  hipsparseLtMatDescriptor_t* matDescr,
452  int64_t rows,
453  int64_t cols,
454  int64_t ld,
455  uint32_t alignment,
456  hipDataType valueType,
457  hipsparseOrder_t order,
458  hipsparseLtSparsity_t sparsity);
459 
473 HIPSPARSELT_EXPORT
475 
497 HIPSPARSELT_EXPORT
498 hipsparseStatus_t hipsparseLtMatDescSetAttribute(const hipsparseLtHandle_t* handle,
499  hipsparseLtMatDescriptor_t* matDescr,
500  hipsparseLtMatDescAttribute_t matAttribute,
501  const void* data,
502  size_t dataSize);
503 
525 HIPSPARSELT_EXPORT
526 hipsparseStatus_t hipsparseLtMatDescGetAttribute(const hipsparseLtHandle_t* handle,
527  const hipsparseLtMatDescriptor_t* matDescr,
528  hipsparseLtMatDescAttribute_t matAttribute,
529  void* data,
530  size_t dataSize);
531 
532 /* matmul descriptor */
562 HIPSPARSELT_EXPORT
563 hipsparseStatus_t hipsparseLtMatmulDescriptorInit(const hipsparseLtHandle_t* handle,
564  hipsparseLtMatmulDescriptor_t* matmulDescr,
565  hipsparseOperation_t opA,
566  hipsparseOperation_t opB,
567  const hipsparseLtMatDescriptor_t* matA,
568  const hipsparseLtMatDescriptor_t* matB,
569  const hipsparseLtMatDescriptor_t* matC,
570  const hipsparseLtMatDescriptor_t* matD,
571  hipsparseLtComputetype_t computeType);
572 
595 HIPSPARSELT_EXPORT
596 hipsparseStatus_t
598  hipsparseLtMatmulDescriptor_t* matmulDescr,
599  hipsparseLtMatmulDescAttribute_t matmulAttribute,
600  const void* data,
601  size_t dataSize);
602 
625 HIPSPARSELT_EXPORT
626 hipsparseStatus_t
628  const hipsparseLtMatmulDescriptor_t* matmulDescr,
629  hipsparseLtMatmulDescAttribute_t matmulAttribute,
630  void* data,
631  size_t dataSize);
632 
633 /* algorithm selection */
651 HIPSPARSELT_EXPORT
652 hipsparseStatus_t
654  hipsparseLtMatmulAlgSelection_t* algSelection,
655  const hipsparseLtMatmulDescriptor_t* matmulDescr,
657 
680 HIPSPARSELT_EXPORT
682  hipsparseLtMatmulAlgSelection_t* algSelection,
684  const void* data,
685  size_t dataSize);
686 
710 HIPSPARSELT_EXPORT
711 hipsparseStatus_t
713  const hipsparseLtMatmulAlgSelection_t* algSelection,
715  void* data,
716  size_t dataSize);
717 
718 /* matmul plan */
735 HIPSPARSELT_EXPORT
736 hipsparseStatus_t hipsparseLtMatmulGetWorkspace(const hipsparseLtHandle_t* handle,
737  const hipsparseLtMatmulPlan_t* plan,
738  size_t* workspaceSize);
739 
758 HIPSPARSELT_EXPORT
759 hipsparseStatus_t hipsparseLtMatmulPlanInit(const hipsparseLtHandle_t* handle,
761  const hipsparseLtMatmulDescriptor_t* matmulDescr,
762  const hipsparseLtMatmulAlgSelection_t* algSelection);
763 
777 HIPSPARSELT_EXPORT
779 
780 /* matmul execution */
826 HIPSPARSELT_EXPORT
827 hipsparseStatus_t hipsparseLtMatmul(const hipsparseLtHandle_t* handle,
828  const hipsparseLtMatmulPlan_t* plan,
829  const void* alpha,
830  const void* d_A,
831  const void* d_B,
832  const void* beta,
833  const void* d_C,
834  void* d_D,
835  void* workspace,
836  hipStream_t* streams,
837  int32_t numStreams);
838 
893 HIPSPARSELT_EXPORT
894 hipsparseStatus_t hipsparseLtMatmulSearch(const hipsparseLtHandle_t* handle,
896  const void* alpha,
897  const void* d_A,
898  const void* d_B,
899  const void* beta,
900  const void* d_C,
901  void* d_D,
902  void* workspace,
903  hipStream_t* streams,
904  int32_t numStreams);
905 
906 /* helper */
907 // prune
937 HIPSPARSELT_EXPORT
938 hipsparseStatus_t hipsparseLtSpMMAPrune(const hipsparseLtHandle_t* handle,
939  const hipsparseLtMatmulDescriptor_t* matmulDescr,
940  const void* d_in,
941  void* d_out,
942  hipsparseLtPruneAlg_t pruneAlg,
943  hipStream_t stream);
944 
967 HIPSPARSELT_EXPORT
968 hipsparseStatus_t hipsparseLtSpMMAPruneCheck(const hipsparseLtHandle_t* handle,
969  const hipsparseLtMatmulDescriptor_t* matmulDescr,
970  const void* d_in,
971  int* d_valid,
972  hipStream_t stream);
973 
1008 HIPSPARSELT_EXPORT
1009 hipsparseStatus_t hipsparseLtSpMMAPrune2(const hipsparseLtHandle_t* handle,
1010  const hipsparseLtMatDescriptor_t* sparseMatDescr,
1011  int isSparseA,
1012  hipsparseOperation_t op,
1013  const void* d_in,
1014  void* d_out,
1015  hipsparseLtPruneAlg_t pruneAlg,
1016  hipStream_t stream);
1017 
1044 HIPSPARSELT_EXPORT
1045 hipsparseStatus_t hipsparseLtSpMMAPruneCheck2(const hipsparseLtHandle_t* handle,
1046  const hipsparseLtMatDescriptor_t* sparseMatDescr,
1047  int isSparseA,
1048  hipsparseOperation_t op,
1049  const void* d_in,
1050  int* d_valid,
1051  hipStream_t stream);
1052 
1053 // compression
1073 HIPSPARSELT_EXPORT
1074 hipsparseStatus_t hipsparseLtSpMMACompressedSize(const hipsparseLtHandle_t* handle,
1075  const hipsparseLtMatmulPlan_t* plan,
1076  size_t* compressedSize,
1077  size_t* compressBufferSize);
1078 
1104 HIPSPARSELT_EXPORT
1105 hipsparseStatus_t hipsparseLtSpMMACompress(const hipsparseLtHandle_t* handle,
1106  const hipsparseLtMatmulPlan_t* plan,
1107  const void* d_dense,
1108  void* d_compressed,
1109  void* d_compressBuffer,
1110  hipStream_t stream);
1111 
1132 HIPSPARSELT_EXPORT
1134  const hipsparseLtMatDescriptor_t* sparseMatDescr,
1135  size_t* compressedSize,
1136  size_t* compressBufferSize);
1137 
1167 HIPSPARSELT_EXPORT
1168 hipsparseStatus_t hipsparseLtSpMMACompress2(const hipsparseLtHandle_t* handle,
1169  const hipsparseLtMatDescriptor_t* sparseMatDescr,
1170  int isSparseA,
1171  hipsparseOperation_t op,
1172  const void* d_dense,
1173  void* d_compressed,
1174  void* d_compressBuffer,
1175  hipStream_t stream);
1176 
1177 #ifdef __cplusplus
1178 }
1179 #endif
1180 
1181 #endif // _HIPSPARSELT_H_
void hipsparseLtInitialize()
Initialize hipSPARSELt for the current HIP device.
hipsparseStatus_t hipsparseLtSpMMAPrune2(const hipsparseLtHandle_t *handle, const hipsparseLtMatDescriptor_t *sparseMatDescr, int isSparseA, hipsparseOperation_t op, const void *d_in, void *d_out, hipsparseLtPruneAlg_t pruneAlg, hipStream_t stream)
Purnes a dense matrix.
hipsparseStatus_t hipsparseLtSpMMACompressedSize2(const hipsparseLtHandle_t *handle, const hipsparseLtMatDescriptor_t *sparseMatDescr, size_t *compressedSize, size_t *compressBufferSize)
provide the size of the compressed matrix.
hipsparseStatus_t hipsparseLtSpMMACompress2(const hipsparseLtHandle_t *handle, const hipsparseLtMatDescriptor_t *sparseMatDescr, int isSparseA, hipsparseOperation_t op, const void *d_dense, void *d_compressed, void *d_compressBuffer, hipStream_t stream)
compresses a dense matrix to structured matrix.
hipsparseStatus_t hipsparseLtSpMMAPruneCheck2(const hipsparseLtHandle_t *handle, const hipsparseLtMatDescriptor_t *sparseMatDescr, int isSparseA, hipsparseOperation_t op, const void *d_in, int *d_valid, hipStream_t stream)
checks the correctness of the pruning structure for a given matrix.
hipsparseStatus_t hipsparseLtSpMMAPrune(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulDescriptor_t *matmulDescr, const void *d_in, void *d_out, hipsparseLtPruneAlg_t pruneAlg, hipStream_t stream)
Purnes a dense matrix.
hipsparseStatus_t hipsparseLtSpMMACompressedSize(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulPlan_t *plan, size_t *compressedSize, size_t *compressBufferSize)
provide the size of the compressed matrix.
hipsparseStatus_t hipsparseLtSpMMACompress(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulPlan_t *plan, const void *d_dense, void *d_compressed, void *d_compressBuffer, hipStream_t stream)
compresses a dense matrix to structured matrix.
hipsparseStatus_t hipsparseLtSpMMAPruneCheck(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulDescriptor_t *matmulDescr, const void *d_in, int *d_valid, hipStream_t stream)
checks the correctness of the pruning structure for a given matrix.
hipsparseStatus_t hipsparseLtInit(hipsparseLtHandle_t *handle)
Create a hipsparselt handle.
hipsparseStatus_t hipsparseLtGetVersion(const hipsparseLtHandle_t *handle, int *version)
Retrive the version number of the hipSPARSELt library.
hipsparseStatus_t hipsparseLtDestroy(const hipsparseLtHandle_t *handle)
Destroy a hipsparselt handle.
hipsparseStatus_t hipsparseLtGetProperty(hipLibraryPropertyType propertyType, int *value)
Retrive the value of the requested property.
hipsparseStatus_t hipsparseLtMatmulAlgSelectionInit(const hipsparseLtHandle_t *handle, hipsparseLtMatmulAlgSelection_t *algSelection, const hipsparseLtMatmulDescriptor_t *matmulDescr, hipsparseLtMatmulAlg_t alg)
Initializes the algorithm selection descriptor.
hipsparseStatus_t hipsparseLtMatmulAlgSetAttribute(const hipsparseLtHandle_t *handle, hipsparseLtMatmulAlgSelection_t *algSelection, hipsparseLtMatmulAlgAttribute_t attribute, const void *data, size_t dataSize)
Specify the algorithm attribute of a algorithm selection descriptor.
hipsparseStatus_t hipsparseLtMatmulAlgGetAttribute(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulAlgSelection_t *algSelection, hipsparseLtMatmulAlgAttribute_t attribute, void *data, size_t dataSize)
Get the specific algorithm attribute from algorithm selection descriptor.
hipsparseStatus_t hipsparseLtMatmulDescGetAttribute(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulDescriptor_t *matmulDescr, hipsparseLtMatmulDescAttribute_t matmulAttribute, void *data, size_t dataSize)
Get the matrix type of a matrix descriptor.
hipsparseStatus_t hipsparseLtMatmulDescriptorInit(const hipsparseLtHandle_t *handle, hipsparseLtMatmulDescriptor_t *matmulDescr, hipsparseOperation_t opA, hipsparseOperation_t opB, const hipsparseLtMatDescriptor_t *matA, const hipsparseLtMatDescriptor_t *matB, const hipsparseLtMatDescriptor_t *matC, const hipsparseLtMatDescriptor_t *matD, hipsparseLtComputetype_t computeType)
Initializes the matrix multiplication descriptor.
hipsparseStatus_t hipsparseLtMatmulDescSetAttribute(const hipsparseLtHandle_t *handle, hipsparseLtMatmulDescriptor_t *matmulDescr, hipsparseLtMatmulDescAttribute_t matmulAttribute, const void *data, size_t dataSize)
Specify the matrix attribute of a matrix descriptor.
hipsparseStatus_t hipsparseLtMatmulSearch(const hipsparseLtHandle_t *handle, hipsparseLtMatmulPlan_t *plan, const void *alpha, const void *d_A, const void *d_B, const void *beta, const void *d_C, void *d_D, void *workspace, hipStream_t *streams, int32_t numStreams)
Sparse matrix dense matrix multiplication.
hipsparseStatus_t hipsparseLtMatmul(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulPlan_t *plan, const void *alpha, const void *d_A, const void *d_B, const void *beta, const void *d_C, void *d_D, void *workspace, hipStream_t *streams, int32_t numStreams)
Sparse matrix dense matrix multiplication.
hipsparseStatus_t hipsparseLtMatmulGetWorkspace(const hipsparseLtHandle_t *handle, const hipsparseLtMatmulPlan_t *plan, size_t *workspaceSize)
Determines the required workspace size.
hipsparseStatus_t hipsparseLtMatmulPlanDestroy(const hipsparseLtMatmulPlan_t *plan)
Destroy a matrix multiplication plan descriptor.
hipsparseStatus_t hipsparseLtMatmulPlanInit(const hipsparseLtHandle_t *handle, hipsparseLtMatmulPlan_t *plan, const hipsparseLtMatmulDescriptor_t *matmulDescr, const hipsparseLtMatmulAlgSelection_t *algSelection)
Initializes the matrix multiplication plan descriptor.
hipsparseStatus_t hipsparseLtStructuredDescriptorInit(const hipsparseLtHandle_t *handle, hipsparseLtMatDescriptor_t *matDescr, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, hipDataType valueType, hipsparseOrder_t order, hipsparseLtSparsity_t sparsity)
Create a descriptor for structured matrix.
hipsparseStatus_t hipsparseLtMatDescGetAttribute(const hipsparseLtHandle_t *handle, const hipsparseLtMatDescriptor_t *matDescr, hipsparseLtMatDescAttribute_t matAttribute, void *data, size_t dataSize)
Get the matrix type of a matrix descriptor.
hipsparseStatus_t hipsparseLtMatDescSetAttribute(const hipsparseLtHandle_t *handle, hipsparseLtMatDescriptor_t *matDescr, hipsparseLtMatDescAttribute_t matAttribute, const void *data, size_t dataSize)
Specify the matrix attribute of a matrix descriptor.
hipsparseStatus_t hipsparseLtDenseDescriptorInit(const hipsparseLtHandle_t *handle, hipsparseLtMatDescriptor_t *matDescr, int64_t rows, int64_t cols, int64_t ld, uint32_t alignment, hipDataType valueType, hipsparseOrder_t order)
Create a descriptor for dense matrix.
hipsparseStatus_t hipsparseLtMatDescriptorDestroy(const hipsparseLtMatDescriptor_t *matDescr)
Destroy a matrix descriptor.
hipsparseLtSparsity_t
Specify the sparsity of the structured matrix.
Definition: hipsparselt.h:179
hipsparseLtMatmulAlg_t
Specify the algorithm for matrix-matrix multiplication.
Definition: hipsparselt.h:248
hipsparseLtMatmulAlgAttribute_t
Specify the matrix multiplication algorithm attributes.
Definition: hipsparselt.h:259
hipsparseLtComputetype_t
Specify the compute precision modes of the matrix.
Definition: hipsparselt.h:201
hipsparseLtSplitKMode_t
Specify the split k mode value.
Definition: hipsparselt.h:285
hipsparseLtMatmulDescAttribute_t
Specify the additional attributes of a matrix multiplication descriptor.
Definition: hipsparselt.h:218
hipsparseLtDatatype_t
List of hipsparselt data types.
Definition: hipsparselt.h:162
hipsparseLtPruneAlg_t
Specify the pruning algorithm to apply to the structured matrix before the compression.
Definition: hipsparselt.h:274
hipsparseLtMatDescAttribute_t
Specify the additional attributes of a matrix descriptor.
Definition: hipsparselt.h:191
@ HIPSPARSELT_SPARSITY_50_PERCENT
Definition: hipsparselt.h:180
@ HIPSPARSELT_MATMUL_ALG_DEFAULT
Definition: hipsparselt.h:249
@ HIPSPARSELT_MATMUL_SPLIT_K
Definition: hipsparselt.h:263
@ HIPSPARSELT_MATMUL_SPLIT_K_BUFFERS
Definition: hipsparselt.h:265
@ HIPSPARSELT_MATMUL_SEARCH_ITERATIONS
Definition: hipsparselt.h:262
@ HIPSPARSELT_MATMUL_SPLIT_K_MODE
Definition: hipsparselt.h:264
@ HIPSPARSELT_MATMUL_ALG_CONFIG_ID
Definition: hipsparselt.h:260
@ HIPSPARSELT_MATMUL_ALG_CONFIG_MAX_ID
Definition: hipsparselt.h:261
@ HIPSPARSELT_COMPUTE_32F
Definition: hipsparselt.h:204
@ HIPSPARSELT_COMPUTE_TF32_FAST
Definition: hipsparselt.h:207
@ HIPSPARSELT_COMPUTE_TF32
Definition: hipsparselt.h:205
@ HIPSPARSELT_COMPUTE_32I
Definition: hipsparselt.h:203
@ HIPSPARSELT_COMPUTE_16F
Definition: hipsparselt.h:202
@ HIPSPARSELT_SPLIT_K_MODE_TWO_KERNELS
Definition: hipsparselt.h:287
@ HIPSPARSELT_SPLIT_K_MODE_ONE_KERNEL
Definition: hipsparselt.h:286
@ HIPSPARSELT_MATMUL_ACTIVATION_RELU_UPPERBOUND
Definition: hipsparselt.h:220
@ HIPSPARSELT_MATMUL_BIAS_STRIDE
Definition: hipsparselt.h:226
@ HIPSPARSELT_MATMUL_BIAS_POINTER
Definition: hipsparselt.h:227
@ HIPSPARSELT_MATMUL_ACTIVATION_LEAKYRELU
Definition: hipsparselt.h:229
@ HIPSPARSELT_MATMUL_BIAS_TYPE
Definition: hipsparselt.h:235
@ HIPSPARSELT_MATMUL_ACTIVATION_LEAKYRELU_ALPHA
Definition: hipsparselt.h:230
@ HIPSPARSELT_MATMUL_ACTIVATION_TANH_BETA
Definition: hipsparselt.h:234
@ HIPSPARSELT_MATMUL_ACTIVATION_GELU_SCALING
Definition: hipsparselt.h:223
@ HIPSPARSELT_MATMUL_ALPHA_VECTOR_SCALING
Definition: hipsparselt.h:224
@ HIPSPARSELT_MATMUL_ACTIVATION_TANH
Definition: hipsparselt.h:232
@ HIPSPARSELT_MATMUL_ACTIVATION_SIGMOID
Definition: hipsparselt.h:231
@ HIPSPARSELT_MATMUL_ACTIVATION_TANH_ALPHA
Definition: hipsparselt.h:233
@ HIPSPARSELT_MATMUL_BETA_VECTOR_SCALING
Definition: hipsparselt.h:225
@ HIPSPARSELT_MATMUL_ACTIVATION_RELU
Definition: hipsparselt.h:219
@ HIPSPARSELT_MATMUL_ACTIVATION_GELU
Definition: hipsparselt.h:222
@ HIPSPARSELT_MATMUL_ACTIVATION_ABS
Definition: hipsparselt.h:228
@ HIPSPARSELT_MATMUL_ACTIVATION_RELU_THRESHOLD
Definition: hipsparselt.h:221
@ HIPSPARSELT_MATMUL_SPARSE_MAT_POINTER
Definition: hipsparselt.h:239
@ HIPSPARSELT_R_8F
Definition: hipsparselt.h:167
@ HIPSPARSELT_R_32F
Definition: hipsparselt.h:164
@ HIPSPARSELT_R_16F
Definition: hipsparselt.h:163
@ HIPSPARSELT_R_16BF
Definition: hipsparselt.h:166
@ HIPSPARSELT_R_8I
Definition: hipsparselt.h:165
@ HIPSPARSELT_R_8BF
Definition: hipsparselt.h:168
@ HIPSPARSELT_PRUNE_SPMMA_TILE
Definition: hipsparselt.h:275
@ HIPSPARSELT_PRUNE_SPMMA_STRIP
Definition: hipsparselt.h:276
@ HIPSPARSELT_MAT_BATCH_STRIDE
Definition: hipsparselt.h:193
@ HIPSPARSELT_MAT_NUM_BATCHES
Definition: hipsparselt.h:192
hipsparseStatus_t hipsparseLtGetGitRevision(hipsparseLtHandle_t handle, char *rev)
hipsparseStatus_t hipsparseLtGetArchName(char **archName)
Handle to the hipSPARSELt library context queue.
Definition: hipsparselt.h:100
uint8_t data[11024]
Definition: hipsparselt.h:100
Descriptor of the matrix.
Definition: hipsparselt.h:111
uint8_t data[11024]
Definition: hipsparselt.h:111
Descriptor of the matrix multiplication algorithm.
Definition: hipsparselt.h:129
uint8_t data[11024]
Definition: hipsparselt.h:129
Descriptor of the matrix multiplication operation.
Definition: hipsparselt.h:121
uint8_t data[11024]
Definition: hipsparselt.h:121
Descriptor of the matrix multiplication execution plan.
Definition: hipsparselt.h:140
uint8_t data[11024]
Definition: hipsparselt.h:140