Actual source code: mpicusparsematimpl.h
petsc-3.15.0 2021-03-30
Report Typos and Errors

      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