1c8564c30SJames Wright #ifndef MAT_CEED_IMPL_H 2c8564c30SJames Wright #define MAT_CEED_IMPL_H 3c8564c30SJames Wright 4c8564c30SJames Wright #include <ceed.h> 5c8564c30SJames Wright #include <petscdm.h> 6c8564c30SJames Wright #include <petscmat.h> 7c8564c30SJames Wright 8c8564c30SJames Wright #if defined(__clang_analyzer__) 9c8564c30SJames Wright #define MATCEED_EXTERN extern 10c8564c30SJames Wright #elif defined(__cplusplus) 11c8564c30SJames Wright #define MATCEED_EXTERN extern "C" 12c8564c30SJames Wright #else 13c8564c30SJames Wright #define MATCEED_EXTERN extern 14c8564c30SJames Wright #endif 15c8564c30SJames Wright 16c8564c30SJames Wright #if defined(__clang_analyzer__) 17c8564c30SJames Wright #define MATCEED_INTERN 18c8564c30SJames Wright #else 19c8564c30SJames Wright #define MATCEED_INTERN MATCEED_EXTERN __attribute__((visibility("hidden"))) 20c8564c30SJames Wright #endif 21c8564c30SJames Wright 22c8564c30SJames Wright /** 23c8564c30SJames Wright @brief Calls a libCEED function and then checks the resulting error code. 24c8564c30SJames Wright If the error code is non-zero, then a PETSc error is set with the libCEED error message. 25c8564c30SJames Wright **/ 26*54831c5fSJames Wright #define PetscCallCeed(ceed_, ...) \ 27c8564c30SJames Wright do { \ 28c8564c30SJames Wright int ierr_q_ = __VA_ARGS__; \ 29c8564c30SJames Wright if (ierr_q_ != CEED_ERROR_SUCCESS) { \ 30c8564c30SJames Wright const char *error_message; \ 31c8564c30SJames Wright CeedGetErrorMessage(ceed_, &error_message); \ 32c8564c30SJames Wright SETERRQ(PETSC_COMM_SELF, PETSC_ERR_LIB, "%s", error_message); \ 33c8564c30SJames Wright } \ 34c8564c30SJames Wright } while (0) 35c8564c30SJames Wright 36c8564c30SJames Wright // MatCeed context for applying composite CeedOperator on a DM 37c8564c30SJames Wright typedef struct MatCeedContext_private *MatCeedContext; 38c8564c30SJames Wright struct MatCeedContext_private { 39c8564c30SJames Wright Ceed ceed; 40c8564c30SJames Wright char *name, *internal_mat_type; 41c8564c30SJames Wright PetscMemType mem_type; 42c8564c30SJames Wright PetscInt ref_count, num_mats_assembled_full, num_mats_assembled_pbd; 43c8564c30SJames Wright PetscBool is_destroyed, is_ceed_pbd_valid, is_ceed_vpbd_valid; 44c8564c30SJames Wright PetscLogEvent log_event_mult, log_event_mult_transpose; 45c8564c30SJames Wright DM dm_x, dm_y; 46c8564c30SJames Wright Mat *mats_assembled_full, *mats_assembled_pbd, mat_assembled_full_internal, mat_assembled_pbd_internal; 47c8564c30SJames Wright Vec X_loc, Y_loc_transpose; 48c8564c30SJames Wright CeedVector x_loc, y_loc, coo_values_full, coo_values_pbd; 49c8564c30SJames Wright CeedOperator op_mult, op_mult_transpose; 50c8564c30SJames Wright PetscLogDouble flops_mult, flops_mult_transpose; 51c8564c30SJames Wright }; 52c8564c30SJames Wright 53c8564c30SJames Wright // Context data 54c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatCeedContextCreate(DM dm_x, DM dm_y, Vec X_loc, Vec Y_loc_transpose, CeedOperator op_mult, 55c8564c30SJames Wright CeedOperator op_mult_transpose, PetscLogEvent log_event_mult, 56c8564c30SJames Wright PetscLogEvent log_event_mult_transpose, MatCeedContext *ctx); 57c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatCeedContextReference(MatCeedContext ctx); 58c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatCeedContextReferenceCopy(MatCeedContext ctx, MatCeedContext *ctx_copy); 59c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatCeedContextDestroy(MatCeedContext ctx); 60c8564c30SJames Wright 61c8564c30SJames Wright // Mat Ceed 62c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatGetDiagonal_Ceed(Mat A, Vec D); 63c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatMult_Ceed(Mat A, Vec X, Vec Y); 64c8564c30SJames Wright MATCEED_INTERN PetscErrorCode MatMultTranspose_Ceed(Mat A, Vec Y, Vec X); 65c8564c30SJames Wright 66c8564c30SJames Wright extern PetscClassId MATCEED_CLASSID; 67c8564c30SJames Wright extern PetscLogEvent MATCEED_MULT, MATCEED_MULT_TRANSPOSE; 68c8564c30SJames Wright 69c8564c30SJames Wright #endif // MAT_CEED_IMPL_H 70