4: #include <cusparse_v2.h>
5: #include <petsc/private/cudavecimpl.h>
7: typedef struct {
8: /* The following are used by GPU capabilities to store matrix storage formats on the device */
9: MatCUSPARSEStorageFormat diagGPUMatFormat;
10: MatCUSPARSEStorageFormat offdiagGPUMatFormat;
11: cudaStream_t stream;
12: cusparseHandle_t handle;
13: PetscSplitCSRDataStructure *deviceMat;
14: PetscInt coo_nd,coo_no;
15: THRUSTINTARRAY *coo_p;
16: THRUSTARRAY *coo_pw;
17: } Mat_MPIAIJCUSPARSE;
19: PETSC_INTERN PetscErrorCode MatCUSPARSESetStream(Mat, const cudaStream_t stream);
20: PETSC_INTERN PetscErrorCode MatCUSPARSESetHandle(Mat, const cusparseHandle_t handle);
21: PETSC_INTERN PetscErrorCode MatCUSPARSEClearHandle(Mat);
23: #endif