Actual source code: mpicusparsematimpl.h

petsc-3.13.6 2020-09-29
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: } Mat_MPIAIJCUSPARSE;

 15: PETSC_INTERN PetscErrorCode MatCUSPARSESetStream(Mat, const cudaStream_t stream);
 16: PETSC_INTERN PetscErrorCode MatCUSPARSESetHandle(Mat, const cusparseHandle_t handle);
 17: PETSC_INTERN PetscErrorCode MatCUSPARSEClearHandle(Mat);

 19: #endif