1 /* 2 Defines profile/logging in PETSc. 3 */ 4 #pragma once 5 6 #include <petscsys.h> 7 #include <petsctime.h> 8 #include <petscbt.h> 9 #include <petsclogtypes.h> 10 11 /* MANSEC = Sys */ 12 /* SUBMANSEC = Log */ 13 14 /* General logging of information; different from event logging */ 15 PETSC_EXTERN PetscErrorCode PetscInfo_Private(const char[], PetscObject, const char[], ...) PETSC_ATTRIBUTE_FORMAT(3, 4); 16 #if defined(PETSC_USE_INFO) 17 #define PetscInfo(A, ...) PetscInfo_Private(PETSC_FUNCTION_NAME, ((PetscObject)A), __VA_ARGS__) 18 #else 19 #define PetscInfo(A, ...) PETSC_SUCCESS 20 #endif 21 22 #define PetscInfo1(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 23 #define PetscInfo2(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 24 #define PetscInfo3(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 25 #define PetscInfo4(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 26 #define PetscInfo5(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 27 #define PetscInfo6(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 28 #define PetscInfo7(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 29 #define PetscInfo8(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 30 #define PetscInfo9(...) PETSC_DEPRECATED_MACRO(3, 17, 0, "PetscInfo()", ) PetscInfo(__VA_ARGS__) 31 32 /*E 33 PetscInfoCommFlag - Describes the method by which to filter information displayed by `PetscInfo()` by communicator size 34 35 Values: 36 + `PETSC_INFO_COMM_ALL` - Default uninitialized value. `PetscInfo()` will not filter based on 37 communicator size (i.e. will print for all communicators) 38 . `PETSC_INFO_COMM_NO_SELF` - `PetscInfo()` will NOT print for communicators with size = 1 (i.e. *_COMM_SELF) 39 - `PETSC_INFO_COMM_ONLY_SELF` - `PetscInfo()` will ONLY print for communicators with size = 1 40 41 Level: intermediate 42 43 Note: 44 Used as an input for `PetscInfoSetFilterCommSelf()` 45 46 .seealso: `PetscInfo()`, `PetscInfoSetFromOptions()`, `PetscInfoSetFilterCommSelf()` 47 E*/ 48 typedef enum { 49 PETSC_INFO_COMM_ALL = -1, 50 PETSC_INFO_COMM_NO_SELF = 0, 51 PETSC_INFO_COMM_ONLY_SELF = 1 52 } PetscInfoCommFlag; 53 54 PETSC_EXTERN const char *const PetscInfoCommFlags[]; 55 PETSC_EXTERN PetscErrorCode PetscInfoDeactivateClass(PetscClassId); 56 PETSC_EXTERN PetscErrorCode PetscInfoActivateClass(PetscClassId); 57 PETSC_EXTERN PetscErrorCode PetscInfoEnabled(PetscClassId, PetscBool *); 58 PETSC_EXTERN PetscErrorCode PetscInfoAllow(PetscBool); 59 PETSC_EXTERN PetscErrorCode PetscInfoSetFile(const char[], const char[]); 60 PETSC_EXTERN PetscErrorCode PetscInfoGetFile(char **, FILE **); 61 PETSC_EXTERN PetscErrorCode PetscInfoSetClasses(PetscBool, PetscInt, const char *const *); 62 PETSC_EXTERN PetscErrorCode PetscInfoGetClass(const char *, PetscBool *); 63 PETSC_EXTERN PetscErrorCode PetscInfoGetInfo(PetscBool *, PetscBool *, PetscBool *, PetscBool *, PetscInfoCommFlag *); 64 PETSC_EXTERN PetscErrorCode PetscInfoProcessClass(const char[], PetscInt, const PetscClassId[]); 65 PETSC_EXTERN PetscErrorCode PetscInfoSetFilterCommSelf(PetscInfoCommFlag); 66 PETSC_EXTERN PetscErrorCode PetscInfoSetFromOptions(PetscOptions); 67 PETSC_EXTERN PetscErrorCode PetscInfoDestroy(void); 68 PETSC_EXTERN PetscBool PetscLogPrintInfo; /* if true, indicates PetscInfo() is turned on */ 69 70 PETSC_EXTERN PetscErrorCode PetscIntStackCreate(PetscIntStack *); 71 PETSC_EXTERN PetscErrorCode PetscIntStackDestroy(PetscIntStack); 72 PETSC_EXTERN PetscErrorCode PetscIntStackPush(PetscIntStack, int); 73 PETSC_EXTERN PetscErrorCode PetscIntStackPop(PetscIntStack, int *); 74 PETSC_EXTERN PetscErrorCode PetscIntStackTop(PetscIntStack, int *); 75 PETSC_EXTERN PetscErrorCode PetscIntStackEmpty(PetscIntStack, PetscBool *); 76 77 PETSC_EXTERN PetscErrorCode PetscLogStateCreate(PetscLogState *); 78 PETSC_EXTERN PetscErrorCode PetscLogStateDestroy(PetscLogState *); 79 PETSC_EXTERN PetscErrorCode PetscLogStateGetRegistry(PetscLogState, PetscLogRegistry *); 80 81 PETSC_EXTERN PetscErrorCode PetscLogStateClassRegister(PetscLogState, const char[], PetscClassId, PetscLogStage *); 82 PETSC_EXTERN PetscErrorCode PetscLogStateClassSetActive(PetscLogState, PetscLogStage, PetscClassId, PetscBool); 83 PETSC_EXTERN PetscErrorCode PetscLogStateClassSetActiveAll(PetscLogState, PetscClassId, PetscBool); 84 85 PETSC_EXTERN PetscErrorCode PetscLogStateStageRegister(PetscLogState, const char[], PetscLogStage *); 86 PETSC_EXTERN PetscErrorCode PetscLogStateStagePush(PetscLogState, PetscLogStage); 87 PETSC_EXTERN PetscErrorCode PetscLogStateStagePop(PetscLogState); 88 PETSC_EXTERN PetscErrorCode PetscLogStateStageSetActive(PetscLogState, PetscLogStage, PetscBool); 89 PETSC_EXTERN PetscErrorCode PetscLogStateStageGetActive(PetscLogState, PetscLogStage, PetscBool *); 90 PETSC_EXTERN PetscErrorCode PetscLogStateGetCurrentStage(PetscLogState, PetscLogStage *); 91 92 PETSC_EXTERN PetscErrorCode PetscLogStateEventRegister(PetscLogState, const char[], PetscClassId, PetscLogEvent *); 93 PETSC_EXTERN PetscErrorCode PetscLogStateEventSetCollective(PetscLogState, PetscLogEvent, PetscBool); 94 PETSC_EXTERN PetscErrorCode PetscLogStateEventSetActive(PetscLogState, PetscLogStage, PetscLogEvent, PetscBool); 95 PETSC_EXTERN PetscErrorCode PetscLogStateEventSetActiveAll(PetscLogState, PetscLogEvent, PetscBool); 96 PETSC_EXTERN PetscErrorCode PetscLogStateEventGetActive(PetscLogState, PetscLogStage, PetscLogEvent, PetscBool *); 97 98 PETSC_EXTERN PetscErrorCode PetscLogStateGetEventFromName(PetscLogState, const char[], PetscLogEvent *); 99 PETSC_EXTERN PetscErrorCode PetscLogStateGetStageFromName(PetscLogState, const char[], PetscLogStage *); 100 PETSC_EXTERN PetscErrorCode PetscLogStateGetClassFromName(PetscLogState, const char[], PetscLogClass *); 101 PETSC_EXTERN PetscErrorCode PetscLogStateGetClassFromClassId(PetscLogState, PetscClassId, PetscLogClass *); 102 PETSC_EXTERN PetscErrorCode PetscLogStateGetNumEvents(PetscLogState, PetscInt *); 103 PETSC_EXTERN PetscErrorCode PetscLogStateGetNumStages(PetscLogState, PetscInt *); 104 PETSC_EXTERN PetscErrorCode PetscLogStateGetNumClasses(PetscLogState, PetscInt *); 105 PETSC_EXTERN PetscErrorCode PetscLogStateEventGetInfo(PetscLogState, PetscLogEvent, PetscLogEventInfo *); 106 PETSC_EXTERN PetscErrorCode PetscLogStateStageGetInfo(PetscLogState, PetscLogStage, PetscLogStageInfo *); 107 PETSC_EXTERN PetscErrorCode PetscLogStateClassGetInfo(PetscLogState, PetscLogClass, PetscLogClassInfo *); 108 109 PETSC_EXTERN PetscClassId PETSCLOGHANDLER_CLASSID; 110 111 PETSC_EXTERN PetscFunctionList PetscLogHandlerList; 112 113 PETSC_EXTERN PetscErrorCode PetscLogHandlerRegister(const char[], PetscErrorCode (*)(PetscLogHandler)); 114 PETSC_EXTERN PetscErrorCode PetscLogHandlerCreate(MPI_Comm, PetscLogHandler *); 115 PETSC_EXTERN PetscErrorCode PetscLogHandlerSetType(PetscLogHandler, PetscLogHandlerType); 116 PETSC_EXTERN PetscErrorCode PetscLogHandlerGetType(PetscLogHandler, PetscLogHandlerType *); 117 PETSC_EXTERN PetscErrorCode PetscLogHandlerDestroy(PetscLogHandler *); 118 PETSC_EXTERN PetscErrorCode PetscLogHandlerSetState(PetscLogHandler, PetscLogState); 119 PETSC_EXTERN PetscErrorCode PetscLogHandlerGetState(PetscLogHandler, PetscLogState *); 120 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventBegin(PetscLogHandler, PetscLogEvent, PetscObject, PetscObject, PetscObject, PetscObject); 121 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventEnd(PetscLogHandler, PetscLogEvent, PetscObject, PetscObject, PetscObject, PetscObject); 122 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventSync(PetscLogHandler, PetscLogEvent, MPI_Comm); 123 PETSC_EXTERN PetscErrorCode PetscLogHandlerObjectCreate(PetscLogHandler, PetscObject); 124 PETSC_EXTERN PetscErrorCode PetscLogHandlerObjectDestroy(PetscLogHandler, PetscObject); 125 PETSC_EXTERN PetscErrorCode PetscLogHandlerStagePush(PetscLogHandler, PetscLogStage); 126 PETSC_EXTERN PetscErrorCode PetscLogHandlerStagePop(PetscLogHandler, PetscLogStage); 127 PETSC_EXTERN PetscErrorCode PetscLogHandlerView(PetscLogHandler, PetscViewer); 128 129 PETSC_EXTERN PetscErrorCode PetscLogHandlerGetEventPerfInfo(PetscLogHandler, PetscLogStage, PetscLogEvent, PetscEventPerfInfo **); 130 PETSC_EXTERN PetscErrorCode PetscLogHandlerGetStagePerfInfo(PetscLogHandler, PetscLogStage, PetscEventPerfInfo **); 131 PETSC_EXTERN PetscErrorCode PetscLogHandlerSetLogActions(PetscLogHandler, PetscBool); 132 PETSC_EXTERN PetscErrorCode PetscLogHandlerSetLogObjects(PetscLogHandler, PetscBool); 133 PETSC_EXTERN PetscErrorCode PetscLogHandlerLogObjectState(PetscLogHandler, PetscObject, const char[], ...); 134 PETSC_EXTERN PetscErrorCode PetscLogHandlerGetNumObjects(PetscLogHandler, PetscInt *); 135 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventDeactivatePush(PetscLogHandler, PetscLogStage, PetscLogEvent); 136 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventDeactivatePop(PetscLogHandler, PetscLogStage, PetscLogEvent); 137 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventsPause(PetscLogHandler); 138 PETSC_EXTERN PetscErrorCode PetscLogHandlerEventsResume(PetscLogHandler); 139 PETSC_EXTERN PetscErrorCode PetscLogHandlerDump(PetscLogHandler, const char[]); 140 PETSC_EXTERN PetscErrorCode PetscLogHandlerStageSetVisible(PetscLogHandler, PetscLogStage, PetscBool); 141 PETSC_EXTERN PetscErrorCode PetscLogHandlerStageGetVisible(PetscLogHandler, PetscLogStage, PetscBool *); 142 143 PETSC_EXTERN PetscErrorCode PetscLogHandlerCreateTrace(MPI_Comm, FILE *, PetscLogHandler *); 144 PETSC_EXTERN PetscErrorCode PetscLogHandlerCreateLegacy(MPI_Comm, PetscErrorCode (*)(PetscLogEvent, int, PetscObject, PetscObject, PetscObject, PetscObject), PetscErrorCode (*)(PetscLogEvent, int, PetscObject, PetscObject, PetscObject, PetscObject), PetscErrorCode (*)(PetscObject), PetscErrorCode (*)(PetscObject), PetscLogHandler *); 145 146 /* All events are inactive if an invalid stage is set, like if there have been more stage pops than stage pushes */ 147 #define PetscLogStateStageEventIsActive(state, stage, event) ((stage >= 0) && PetscBTLookup((state)->active, (stage)) && PetscBTLookup((state)->active, (stage) + (event + 1) * (state)->bt_num_stages)) 148 #define PetscLogStateEventCurrentlyActive(state, event) ((state) && PetscLogStateStageEventIsActive(state, (state)->current_stage, event)) 149 150 /* PetscLogHandler with critical methods exposed for speed */ 151 typedef struct _n_PetscLogHandlerHot { 152 PetscLogHandler handler; 153 PetscErrorCode (*eventBegin)(PetscLogHandler, PetscLogEvent, PetscObject, PetscObject, PetscObject, PetscObject); 154 PetscErrorCode (*eventEnd)(PetscLogHandler, PetscLogEvent, PetscObject, PetscObject, PetscObject, PetscObject); 155 PetscErrorCode (*eventSync)(PetscLogHandler, PetscLogEvent, MPI_Comm); 156 PetscErrorCode (*objectCreate)(PetscLogHandler, PetscObject); 157 PetscErrorCode (*objectDestroy)(PetscLogHandler, PetscObject); 158 } PetscLogHandlerHot; 159 160 /* Handle multithreading */ 161 #if defined(PETSC_HAVE_THREADSAFETY) 162 #if defined(__cplusplus) 163 #define PETSC_TLS thread_local 164 #else 165 #define PETSC_TLS _Thread_local 166 #endif 167 #define PETSC_EXTERN_TLS extern PETSC_TLS PETSC_VISIBILITY_PUBLIC 168 #else 169 #define PETSC_EXTERN_TLS PETSC_EXTERN 170 #define PETSC_TLS 171 #endif 172 #if defined(PETSC_HAVE_THREADSAFETY) && defined(PETSC_USE_LOG) 173 PETSC_EXTERN PetscErrorCode PetscAddLogDouble(PetscLogDouble *, PetscLogDouble *, PetscLogDouble); 174 PETSC_EXTERN PetscErrorCode PetscAddLogDoubleCnt(PetscLogDouble *, PetscLogDouble *, PetscLogDouble *, PetscLogDouble *, PetscLogDouble); 175 #else 176 #define PetscAddLogDouble(a, b, c) ((PetscErrorCode)((*(a) += (c), PETSC_SUCCESS) || ((*(b) += (c)), PETSC_SUCCESS))) 177 #define PetscAddLogDoubleCnt(a, b, c, d, e) ((PetscErrorCode)(PetscAddLogDouble(a, c, 1) || PetscAddLogDouble(b, d, e))) 178 #endif 179 180 PETSC_DEPRECATED_FUNCTION(3, 18, 0, "PetscLogObjectParent()", ) static inline PetscErrorCode PetscLogObjectParent(PetscObject o, PetscObject p) 181 { 182 (void)o; 183 (void)p; 184 return PETSC_SUCCESS; 185 } 186 #define PetscLogObjectParents(p, n, d) PetscMacroReturnStandard(for (int _i = 0; _i < (n); ++_i) PetscCall(PetscLogObjectParent((PetscObject)(p), (PetscObject)(d)[_i]));) 187 188 PETSC_DEPRECATED_FUNCTION(3, 18, 0, "PetscLogObjectMemory()", ) static inline PetscErrorCode PetscLogObjectMemory(PetscObject o, PetscLogDouble m) 189 { 190 (void)o; 191 (void)m; 192 return PETSC_SUCCESS; 193 } 194 195 /* Global flop counter */ 196 PETSC_EXTERN PetscLogDouble petsc_TotalFlops; 197 PETSC_EXTERN PetscLogDouble petsc_irecv_ct; 198 PETSC_EXTERN PetscLogDouble petsc_isend_ct; 199 PETSC_EXTERN PetscLogDouble petsc_recv_ct; 200 PETSC_EXTERN PetscLogDouble petsc_send_ct; 201 PETSC_EXTERN PetscLogDouble petsc_irecv_len; 202 PETSC_EXTERN PetscLogDouble petsc_isend_len; 203 PETSC_EXTERN PetscLogDouble petsc_recv_len; 204 PETSC_EXTERN PetscLogDouble petsc_send_len; 205 PETSC_EXTERN PetscLogDouble petsc_allreduce_ct; 206 PETSC_EXTERN PetscLogDouble petsc_gather_ct; 207 PETSC_EXTERN PetscLogDouble petsc_scatter_ct; 208 PETSC_EXTERN PetscLogDouble petsc_wait_ct; 209 PETSC_EXTERN PetscLogDouble petsc_wait_any_ct; 210 PETSC_EXTERN PetscLogDouble petsc_wait_all_ct; 211 PETSC_EXTERN PetscLogDouble petsc_sum_of_waits_ct; 212 213 /* Thread local storage */ 214 PETSC_EXTERN_TLS PetscLogDouble petsc_TotalFlops_th; 215 PETSC_EXTERN_TLS PetscLogDouble petsc_irecv_ct_th; 216 PETSC_EXTERN_TLS PetscLogDouble petsc_isend_ct_th; 217 PETSC_EXTERN_TLS PetscLogDouble petsc_recv_ct_th; 218 PETSC_EXTERN_TLS PetscLogDouble petsc_send_ct_th; 219 PETSC_EXTERN_TLS PetscLogDouble petsc_irecv_len_th; 220 PETSC_EXTERN_TLS PetscLogDouble petsc_isend_len_th; 221 PETSC_EXTERN_TLS PetscLogDouble petsc_recv_len_th; 222 PETSC_EXTERN_TLS PetscLogDouble petsc_send_len_th; 223 PETSC_EXTERN_TLS PetscLogDouble petsc_allreduce_ct_th; 224 PETSC_EXTERN_TLS PetscLogDouble petsc_gather_ct_th; 225 PETSC_EXTERN_TLS PetscLogDouble petsc_scatter_ct_th; 226 PETSC_EXTERN_TLS PetscLogDouble petsc_wait_ct_th; 227 PETSC_EXTERN_TLS PetscLogDouble petsc_wait_any_ct_th; 228 PETSC_EXTERN_TLS PetscLogDouble petsc_wait_all_ct_th; 229 PETSC_EXTERN_TLS PetscLogDouble petsc_sum_of_waits_ct_th; 230 231 /* Global GPU counters */ 232 PETSC_EXTERN PetscLogDouble petsc_ctog_ct; 233 PETSC_EXTERN PetscLogDouble petsc_gtoc_ct; 234 PETSC_EXTERN PetscLogDouble petsc_ctog_sz; 235 PETSC_EXTERN PetscLogDouble petsc_gtoc_sz; 236 PETSC_EXTERN PetscLogDouble petsc_ctog_ct_scalar; 237 PETSC_EXTERN PetscLogDouble petsc_gtoc_ct_scalar; 238 PETSC_EXTERN PetscLogDouble petsc_ctog_sz_scalar; 239 PETSC_EXTERN PetscLogDouble petsc_gtoc_sz_scalar; 240 PETSC_EXTERN PetscLogDouble petsc_gflops; 241 PETSC_EXTERN PetscLogDouble petsc_gtime; 242 PETSC_EXTERN PetscLogDouble petsc_genergy; 243 PETSC_EXTERN PetscLogDouble petsc_genergy_meter; 244 245 /* Thread local storage */ 246 PETSC_EXTERN_TLS PetscLogDouble petsc_ctog_ct_th; 247 PETSC_EXTERN_TLS PetscLogDouble petsc_gtoc_ct_th; 248 PETSC_EXTERN_TLS PetscLogDouble petsc_ctog_sz_th; 249 PETSC_EXTERN_TLS PetscLogDouble petsc_gtoc_sz_th; 250 PETSC_EXTERN_TLS PetscLogDouble petsc_ctog_ct_scalar_th; 251 PETSC_EXTERN_TLS PetscLogDouble petsc_gtoc_ct_scalar_th; 252 PETSC_EXTERN_TLS PetscLogDouble petsc_ctog_sz_scalar_th; 253 PETSC_EXTERN_TLS PetscLogDouble petsc_gtoc_sz_scalar_th; 254 PETSC_EXTERN_TLS PetscLogDouble petsc_gflops_th; 255 PETSC_EXTERN_TLS PetscLogDouble petsc_gtime_th; 256 257 PETSC_EXTERN PetscBool PetscLogMemory; 258 PETSC_EXTERN PetscBool PetscLogSyncOn; /* true if logging synchronization is enabled */ 259 260 PETSC_EXTERN PetscLogState petsc_log_state; 261 262 #define PETSC_LOG_HANDLER_MAX 4 263 PETSC_EXTERN PetscLogHandlerHot PetscLogHandlers[PETSC_LOG_HANDLER_MAX]; 264 265 #if defined(PETSC_USE_LOG) /* --- Logging is turned on --------------------------------*/ 266 PETSC_EXTERN PetscErrorCode PetscGetFlops(PetscLogDouble *); 267 268 PETSC_EXTERN PetscErrorCode PetscLogObjectState(PetscObject, const char[], ...) PETSC_ATTRIBUTE_FORMAT(2, 3); 269 270 /* Initialization functions */ 271 PETSC_EXTERN PetscErrorCode PetscLogDefaultBegin(void); 272 PETSC_EXTERN PetscErrorCode PetscLogNestedBegin(void); 273 PETSC_EXTERN PetscErrorCode PetscLogTraceBegin(FILE *); 274 PETSC_EXTERN PetscErrorCode PetscLogMPEBegin(void); 275 PETSC_EXTERN PetscErrorCode PetscLogPerfstubsBegin(void); 276 PETSC_EXTERN PetscErrorCode PetscLogLegacyCallbacksBegin(PetscErrorCode (*)(PetscLogEvent, int, PetscObject, PetscObject, PetscObject, PetscObject), PetscErrorCode (*)(PetscLogEvent, int, PetscObject, PetscObject, PetscObject, PetscObject), PetscErrorCode (*)(PetscObject), PetscErrorCode (*)(PetscObject)); 277 PETSC_EXTERN PetscErrorCode PetscLogActions(PetscBool); 278 PETSC_EXTERN PetscErrorCode PetscLogObjects(PetscBool); 279 PETSC_EXTERN PetscErrorCode PetscLogSetThreshold(PetscLogDouble, PetscLogDouble *); 280 281 /* Output functions */ 282 PETSC_EXTERN PetscErrorCode PetscLogView(PetscViewer); 283 PETSC_EXTERN PetscErrorCode PetscLogViewFromOptions(void); 284 PETSC_EXTERN PetscErrorCode PetscLogDump(const char[]); 285 PETSC_EXTERN PetscErrorCode PetscLogMPEDump(const char[]); 286 287 PETSC_EXTERN PetscErrorCode PetscLogGetState(PetscLogState *); 288 PETSC_EXTERN PetscErrorCode PetscLogGetDefaultHandler(PetscLogHandler *); 289 PETSC_EXTERN PetscErrorCode PetscLogHandlerStart(PetscLogHandler); 290 PETSC_EXTERN PetscErrorCode PetscLogHandlerStop(PetscLogHandler); 291 292 /* Status checking functions */ 293 PETSC_EXTERN PetscErrorCode PetscLogIsActive(PetscBool *); 294 295 /* Stage functions */ 296 PETSC_EXTERN PetscErrorCode PetscLogStageRegister(const char[], PetscLogStage *); 297 PETSC_EXTERN PetscErrorCode PetscLogStagePush(PetscLogStage); 298 PETSC_EXTERN PetscErrorCode PetscLogStagePop(void); 299 PETSC_EXTERN PetscErrorCode PetscLogStageSetActive(PetscLogStage, PetscBool); 300 PETSC_EXTERN PetscErrorCode PetscLogStageGetActive(PetscLogStage, PetscBool *); 301 PETSC_EXTERN PetscErrorCode PetscLogStageSetVisible(PetscLogStage, PetscBool); 302 PETSC_EXTERN PetscErrorCode PetscLogStageGetVisible(PetscLogStage, PetscBool *); 303 PETSC_EXTERN PetscErrorCode PetscLogStageGetId(const char[], PetscLogStage *); 304 PETSC_EXTERN PetscErrorCode PetscLogStageGetName(PetscLogEvent, const char **); 305 PETSC_EXTERN PetscErrorCode PetscLogStageGetPerfInfo(PetscLogStage, PetscEventPerfInfo *); 306 307 /* Event functions */ 308 PETSC_EXTERN PetscErrorCode PetscLogEventRegister(const char[], PetscClassId, PetscLogEvent *); 309 PETSC_EXTERN PetscErrorCode PetscLogEventSetCollective(PetscLogEvent, PetscBool); 310 PETSC_EXTERN PetscErrorCode PetscLogEventIncludeClass(PetscClassId); 311 PETSC_EXTERN PetscErrorCode PetscLogEventExcludeClass(PetscClassId); 312 PETSC_EXTERN PetscErrorCode PetscLogEventActivate(PetscLogEvent); 313 PETSC_EXTERN PetscErrorCode PetscLogEventDeactivate(PetscLogEvent); 314 PETSC_EXTERN PetscErrorCode PetscLogEventDeactivatePush(PetscLogEvent); 315 PETSC_EXTERN PetscErrorCode PetscLogEventDeactivatePop(PetscLogEvent); 316 PETSC_EXTERN PetscErrorCode PetscLogEventSetActiveAll(PetscLogEvent, PetscBool); 317 PETSC_EXTERN PetscErrorCode PetscLogEventActivateClass(PetscClassId); 318 PETSC_EXTERN PetscErrorCode PetscLogEventDeactivateClass(PetscClassId); 319 PETSC_EXTERN PetscErrorCode PetscLogEventGetId(const char[], PetscLogEvent *); 320 PETSC_EXTERN PetscErrorCode PetscLogEventGetName(PetscLogEvent, const char **); 321 PETSC_EXTERN PetscErrorCode PetscLogEventGetPerfInfo(PetscLogStage, PetscLogEvent, PetscEventPerfInfo *); 322 PETSC_EXTERN PetscErrorCode PetscLogEventSetDof(PetscLogEvent, PetscInt, PetscLogDouble); 323 PETSC_EXTERN PetscErrorCode PetscLogEventSetError(PetscLogEvent, PetscInt, PetscLogDouble); 324 PETSC_EXTERN PetscErrorCode PetscLogEventsPause(void); 325 PETSC_EXTERN PetscErrorCode PetscLogEventsResume(void); 326 327 /* Class functions */ 328 PETSC_EXTERN PetscErrorCode PetscLogClassGetClassId(const char[], PetscClassId *); 329 PETSC_EXTERN PetscErrorCode PetscLogClassIdGetName(PetscClassId, const char **); 330 331 static inline PetscErrorCode PetscLogEventSync(PetscLogEvent e, MPI_Comm comm) 332 { 333 if (PetscLogStateEventCurrentlyActive(petsc_log_state, e)) { 334 for (int i = 0; i < PETSC_LOG_HANDLER_MAX; i++) { 335 PetscLogHandlerHot *h = &PetscLogHandlers[i]; 336 if (h->eventSync) { 337 PetscErrorCode err = (*h->eventSync)(h->handler, e, comm); 338 if (err != PETSC_SUCCESS) return err; 339 } 340 } 341 } 342 return PETSC_SUCCESS; 343 } 344 345 static inline PetscErrorCode PetscLogEventBegin_Internal(PetscLogEvent e, PetscObject o1, PetscObject o2, PetscObject o3, PetscObject o4) 346 { 347 if (PetscLogStateEventCurrentlyActive(petsc_log_state, e)) { 348 for (int i = 0; i < PETSC_LOG_HANDLER_MAX; i++) { 349 PetscLogHandlerHot *h = &PetscLogHandlers[i]; 350 if (h->eventBegin) { 351 PetscErrorCode err = (*h->eventBegin)(h->handler, e, o1, o2, o3, o4); 352 if (err != PETSC_SUCCESS) return err; 353 } 354 } 355 } 356 return PETSC_SUCCESS; 357 } 358 #define PetscLogEventBegin(e, o1, o2, o3, o4) PetscLogEventBegin_Internal(e, (PetscObject)(o1), (PetscObject)(o2), (PetscObject)(o3), (PetscObject)(o4)) 359 360 static inline PetscErrorCode PetscLogEventEnd_Internal(PetscLogEvent e, PetscObject o1, PetscObject o2, PetscObject o3, PetscObject o4) 361 { 362 if (PetscLogStateEventCurrentlyActive(petsc_log_state, e)) { 363 for (int i = 0; i < PETSC_LOG_HANDLER_MAX; i++) { 364 PetscLogHandlerHot *h = &PetscLogHandlers[i]; 365 if (h->eventEnd) { 366 PetscErrorCode err = (*h->eventEnd)(h->handler, e, o1, o2, o3, o4); 367 if (err != PETSC_SUCCESS) return err; 368 } 369 } 370 } 371 return PETSC_SUCCESS; 372 } 373 #define PetscLogEventEnd(e, o1, o2, o3, o4) PetscLogEventEnd_Internal(e, (PetscObject)(o1), (PetscObject)(o2), (PetscObject)(o3), (PetscObject)(o4)) 374 375 /* Object functions */ 376 static inline PetscErrorCode PetscLogObjectCreate(PetscObject o) 377 { 378 if (petsc_log_state) { 379 for (int i = 0; i < PETSC_LOG_HANDLER_MAX; i++) { 380 PetscLogHandlerHot *h = &PetscLogHandlers[i]; 381 if (h->objectCreate) { 382 PetscErrorCode err = (*h->objectCreate)(h->handler, o); 383 if (err != PETSC_SUCCESS) return err; 384 } 385 } 386 } 387 return PETSC_SUCCESS; 388 } 389 390 static inline PetscErrorCode PetscLogObjectDestroy(PetscObject o) 391 { 392 if (petsc_log_state) { 393 for (int i = 0; i < PETSC_LOG_HANDLER_MAX; i++) { 394 PetscLogHandlerHot *h = &PetscLogHandlers[i]; 395 if (h->objectDestroy) { 396 PetscErrorCode err = (*h->objectDestroy)(h->handler, o); 397 if (err != PETSC_SUCCESS) return err; 398 } 399 } 400 } 401 return PETSC_SUCCESS; 402 } 403 404 /* 405 Flop counting: We count each arithmetic operation (e.g., addition, multiplication) separately. 406 407 For the complex numbers version, note that 408 1 complex addition = 2 flops 409 1 complex multiplication = 6 flops, 410 where we define 1 flop as that for a double precision scalar. We roughly approximate 411 flop counting for complex numbers by multiplying the total flops by 4; this corresponds 412 to the assumption that we're counting mostly additions and multiplications -- and 413 roughly the same number of each. More accurate counting could be done by distinguishing 414 among the various arithmetic operations. 415 */ 416 417 #if defined(PETSC_USE_COMPLEX) 418 #define PETSC_FLOPS_PER_OP 4.0 419 #else 420 #define PETSC_FLOPS_PER_OP 1.0 421 #endif 422 423 /*@ 424 PetscLogFlops - Log how many flops are performed in a calculation 425 426 Input Parameter: 427 . flops - the number of flops 428 429 Level: intermediate 430 431 Note: 432 To limit the chance of integer overflow when multiplying by a constant, represent the constant as a double, 433 not an integer. Use `PetscLogFlops`(4.0*n) not `PetscLogFlops`(4*n) 434 435 .seealso: [](ch_profiling), `PetscLogView()`, `PetscLogGpuFlops()` 436 @*/ 437 static inline PetscErrorCode PetscLogFlops(PetscLogDouble n) 438 { 439 PetscAssert(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot log negative flops"); 440 return PetscAddLogDouble(&petsc_TotalFlops, &petsc_TotalFlops_th, PETSC_FLOPS_PER_OP * n); 441 } 442 443 /* 444 These are used internally in the PETSc routines to keep a count of MPI messages and 445 their sizes. 446 447 This does not work for MPI-Uni because our include/petsc/mpiuni/mpi.h file 448 uses macros to defined the MPI operations. 449 450 It does not work correctly from HP-UX because it processes the 451 macros in a way that sometimes it double counts, hence 452 PETSC_HAVE_BROKEN_RECURSIVE_MACRO 453 454 It does not work with Windows because winmpich lacks MPI_Type_size() 455 */ 456 #if !defined(MPIUNI_H) && !defined(PETSC_HAVE_BROKEN_RECURSIVE_MACRO) 457 /* 458 Logging of MPI activities 459 */ 460 static inline PetscErrorCode PetscMPITypeSize(PetscCount count, MPI_Datatype type, PetscLogDouble *length, PetscLogDouble *length_th) 461 { 462 PetscMPIInt typesize; 463 464 if (type == MPI_DATATYPE_NULL) return PETSC_SUCCESS; 465 PetscCallMPI(MPI_Type_size(type, &typesize)); 466 return PetscAddLogDouble(length, length_th, (PetscLogDouble)(count * typesize)); 467 } 468 469 static inline PetscErrorCode PetscMPITypeSizeComm(MPI_Comm comm, const PetscMPIInt *counts, MPI_Datatype type, PetscLogDouble *length, PetscLogDouble *length_th) 470 { 471 PetscMPIInt typesize, size, p; 472 PetscLogDouble l; 473 474 if (type == MPI_DATATYPE_NULL) return PETSC_SUCCESS; 475 PetscCallMPI(MPI_Comm_size(comm, &size)); 476 PetscCallMPI(MPI_Type_size(type, &typesize)); 477 for (p = 0, l = 0.0; p < size; ++p) l += (PetscLogDouble)(counts[p] * typesize); 478 return PetscAddLogDouble(length, length_th, l); 479 } 480 481 /* 482 Returns 1 if the communicator is parallel else zero 483 */ 484 static inline int PetscMPIParallelComm(MPI_Comm comm) 485 { 486 PetscMPIInt size; 487 MPI_Comm_size(comm, &size); 488 return size > 1; 489 } 490 491 #define MPI_Irecv(buf, count, datatype, source, tag, comm, request) \ 492 (PetscAddLogDouble(&petsc_irecv_ct, &petsc_irecv_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_irecv_len), &(petsc_irecv_len_th)) || MPI_Irecv((buf), (count), (datatype), (source), (tag), (comm), (request))) 493 494 #define MPI_Irecv_c(buf, count, datatype, source, tag, comm, request) \ 495 (PetscAddLogDouble(&petsc_irecv_ct, &petsc_irecv_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_irecv_len), &(petsc_irecv_len_th)) || MPI_Irecv_c((buf), (count), (datatype), (source), (tag), (comm), (request))) 496 497 #define MPI_Isend(buf, count, datatype, dest, tag, comm, request) \ 498 (PetscAddLogDouble(&petsc_isend_ct, &petsc_isend_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_isend_len), &(petsc_isend_len_th)) || MPI_Isend((buf), (count), (datatype), (dest), (tag), (comm), (request))) 499 500 #define MPI_Isend_c(buf, count, datatype, dest, tag, comm, request) \ 501 (PetscAddLogDouble(&petsc_isend_ct, &petsc_isend_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_isend_len), &(petsc_isend_len_th)) || MPI_Isend_c((buf), (count), (datatype), (dest), (tag), (comm), (request))) 502 503 #define MPI_Startall_irecv(count, datatype, number, requests) \ 504 (PetscAddLogDouble(&petsc_irecv_ct, &petsc_irecv_ct_th, number) || PetscMPITypeSize((count), (datatype), &(petsc_irecv_len), &(petsc_irecv_len_th)) || ((number) && MPI_Startall((number), (requests)))) 505 506 #define MPI_Startall_isend(count, datatype, number, requests) \ 507 (PetscAddLogDouble(&petsc_isend_ct, &petsc_isend_ct_th, number) || PetscMPITypeSize((count), (datatype), &(petsc_isend_len), &(petsc_isend_len_th)) || ((number) && MPI_Startall((number), (requests)))) 508 509 #define MPI_Start_isend(count, datatype, requests) (PetscAddLogDouble(&petsc_isend_ct, &petsc_isend_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_isend_len), (&petsc_isend_len_th)) || MPI_Start(requests)) 510 511 #define MPI_Recv(buf, count, datatype, source, tag, comm, status) \ 512 (PetscAddLogDouble(&petsc_recv_ct, &petsc_recv_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_recv_len), (&petsc_recv_len_th)) || MPI_Recv((buf), (count), (datatype), (source), (tag), (comm), (status))) 513 514 #define MPI_Recv_c(buf, count, datatype, source, tag, comm, status) \ 515 (PetscAddLogDouble(&petsc_recv_ct, &petsc_recv_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_recv_len), &(petsc_recv_len_th)) || MPI_Recv_c((buf), (count), (datatype), (source), (tag), (comm), (status))) 516 517 #define MPI_Send(buf, count, datatype, dest, tag, comm) \ 518 (PetscAddLogDouble(&petsc_send_ct, &petsc_send_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Send((buf), (count), (datatype), (dest), (tag), (comm))) 519 520 #define MPI_Send_c(buf, count, datatype, dest, tag, comm) \ 521 (PetscAddLogDouble(&petsc_send_ct, &petsc_send_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Send_c((buf), (count), (datatype), (dest), (tag), (comm))) 522 523 #define MPI_Wait(request, status) (PetscAddLogDouble(&petsc_wait_ct, &petsc_wait_ct_th, 1) || PetscAddLogDouble(&petsc_sum_of_waits_ct, &petsc_sum_of_waits_ct_th, 1) || MPI_Wait((request), (status))) 524 525 #define MPI_Waitany(a, b, c, d) (PetscAddLogDouble(&petsc_wait_any_ct, &petsc_wait_any_ct_th, 1) || PetscAddLogDouble(&petsc_sum_of_waits_ct, &petsc_sum_of_waits_ct_th, 1) || MPI_Waitany((a), (b), (c), (d))) 526 527 #define MPI_Waitall(count, array_of_requests, array_of_statuses) \ 528 (PetscAddLogDouble(&petsc_wait_all_ct, &petsc_wait_all_ct_th, 1) || PetscAddLogDouble(&petsc_sum_of_waits_ct, &petsc_sum_of_waits_ct_th, count) || MPI_Waitall((count), (array_of_requests), (array_of_statuses))) 529 530 #define MPI_Allreduce(sendbuf, recvbuf, count, datatype, op, comm) (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || MPI_Allreduce((sendbuf), (recvbuf), (count), (datatype), (op), (comm))) 531 532 #define MPI_Bcast(buffer, count, datatype, root, comm) (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || MPI_Bcast((buffer), (count), (datatype), (root), (comm))) 533 534 #define MPI_Reduce_scatter_block(sendbuf, recvbuf, recvcount, datatype, op, comm) \ 535 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || MPI_Reduce_scatter_block((sendbuf), (recvbuf), (recvcount), (datatype), (op), (comm))) 536 537 #define MPI_Alltoall(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ 538 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Alltoall((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (comm))) 539 540 #define MPI_Alltoallv(sendbuf, sendcnts, sdispls, sendtype, recvbuf, recvcnts, rdispls, recvtype, comm) \ 541 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || PetscMPITypeSizeComm((comm), (sendcnts), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Alltoallv((sendbuf), (sendcnts), (sdispls), (sendtype), (recvbuf), (recvcnts), (rdispls), (recvtype), (comm))) 542 543 #define MPI_Allgather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm) \ 544 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, PetscMPIParallelComm(comm)) || MPI_Allgather((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (comm))) 545 546 #define MPI_Allgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, comm) \ 547 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, PetscMPIParallelComm(comm)) || MPI_Allgatherv((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (displs), (recvtype), (comm))) 548 549 #define MPI_Gather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm) \ 550 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, 1) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Gather((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm))) 551 552 #define MPI_Gatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, root, comm) \ 553 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, 1) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Gatherv((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (displs), (recvtype), (root), (comm))) 554 555 #define MPI_Scatter(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm) \ 556 (PetscAddLogDouble(&petsc_scatter_ct, &petsc_scatter_ct_th, 1) || PetscMPITypeSize((recvcount), (recvtype), (&petsc_recv_len), &(petsc_recv_len_th)) || MPI_Scatter((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm))) 557 558 #define MPI_Scatterv(sendbuf, sendcount, displs, sendtype, recvbuf, recvcount, recvtype, root, comm) \ 559 (PetscAddLogDouble(&petsc_scatter_ct, &petsc_scatter_ct_th, 1) || PetscMPITypeSize((recvcount), (recvtype), (&petsc_recv_len), &(petsc_recv_len_th)) || MPI_Scatterv((sendbuf), (sendcount), (displs), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm))) 560 561 #define MPI_Ialltoall(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm, request) \ 562 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Ialltoall((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (comm), (request))) 563 564 #define MPI_Ialltoallv(sendbuf, sendcnts, sdispls, sendtype, recvbuf, recvcnts, rdispls, recvtype, comm, request) \ 565 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || PetscMPITypeSizeComm((comm), (sendcnts), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Ialltoallv((sendbuf), (sendcnts), (sdispls), (sendtype), (recvbuf), (recvcnts), (rdispls), (recvtype), (comm), (request))) 566 567 #define MPI_Iallgather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, comm, request) \ 568 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, PetscMPIParallelComm(comm)) || MPI_Iallgather((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (comm), (request))) 569 570 #define MPI_Iallgatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, comm, request) \ 571 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, PetscMPIParallelComm(comm)) || MPI_Iallgatherv((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (displs), (recvtype), (comm), (request))) 572 573 #define MPI_Igather(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm, request) \ 574 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, 1) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Igather((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm), (request))) 575 576 #define MPI_Igatherv(sendbuf, sendcount, sendtype, recvbuf, recvcount, displs, recvtype, root, comm, request) \ 577 (PetscAddLogDouble(&petsc_gather_ct, &petsc_gather_ct_th, 1) || PetscMPITypeSize((sendcount), (sendtype), (&petsc_send_len), (&petsc_send_len_th)) || MPI_Igatherv((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (displs), (recvtype), (root), (comm), (request))) 578 579 #define MPI_Iscatter(sendbuf, sendcount, sendtype, recvbuf, recvcount, recvtype, root, comm, request) \ 580 (PetscAddLogDouble(&petsc_scatter_ct, &petsc_scatter_ct_th, 1) || PetscMPITypeSize((recvcount), (recvtype), (&petsc_recv_len), (&petsc_recv_len_th)) || MPI_Iscatter((sendbuf), (sendcount), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm), (request))) 581 582 #define MPI_Iscatterv(sendbuf, sendcount, displs, sendtype, recvbuf, recvcount, recvtype, root, comm, request) \ 583 (PetscAddLogDouble(&petsc_scatter_ct, &petsc_scatter_ct_th, 1) || PetscMPITypeSize((recvcount), (recvtype), (&petsc_recv_len), (&petsc_recv_len_th)) || MPI_Iscatterv((sendbuf), (sendcount), (displs), (sendtype), (recvbuf), (recvcount), (recvtype), (root), (comm), (request))) 584 585 #define MPIX_Send_enqueue(buf, count, datatype, dest, tag, comm) \ 586 (PetscAddLogDouble(&petsc_send_ct, &petsc_send_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_send_len), (&petsc_send_len_th)) || MPIX_Send_enqueue((buf), (count), (datatype), (dest), (tag), (comm))) 587 588 #define MPIX_Recv_enqueue(buf, count, datatype, source, tag, comm, status) \ 589 (PetscAddLogDouble(&petsc_recv_ct, &petsc_recv_ct_th, 1) || PetscMPITypeSize((count), (datatype), (&petsc_recv_len), (&petsc_recv_len_th)) || MPIX_Recv_enqueue((buf), (count), (datatype), (source), (tag), (comm), (status))) 590 591 #define MPIX_Isend_enqueue(buf, count, datatype, dest, tag, comm, request) \ 592 (PetscAddLogDouble(&petsc_isend_ct, &petsc_isend_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_isend_len), &(petsc_isend_len_th)) || MPIX_Isend_enqueue((buf), (count), (datatype), (dest), (tag), (comm), (request))) 593 594 #define MPIX_Irecv_enqueue(buf, count, datatype, source, tag, comm, request) \ 595 (PetscAddLogDouble(&petsc_irecv_ct, &petsc_irecv_ct_th, 1) || PetscMPITypeSize((count), (datatype), &(petsc_irecv_len), &(petsc_irecv_len_th)) || MPIX_Irecv_enqueue((buf), (count), (datatype), (source), (tag), (comm), (request))) 596 597 #define MPIX_Allreduce_enqueue(sendbuf, recvbuf, count, datatype, op, comm) \ 598 (PetscAddLogDouble(&petsc_allreduce_ct, &petsc_allreduce_ct_th, PetscMPIParallelComm(comm)) || MPIX_Allreduce_enqueue((sendbuf), (recvbuf), (count), (datatype), (op), (comm))) 599 600 #define MPIX_Wait_enqueue(request, status) (PetscAddLogDouble(&petsc_wait_ct, &petsc_wait_ct_th, 1) || PetscAddLogDouble(&petsc_sum_of_waits_ct, &petsc_sum_of_waits_ct_th, 1) || MPIX_Wait_enqueue((request), (status))) 601 602 #define MPIX_Waitall_enqueue(count, array_of_requests, array_of_statuses) \ 603 (PetscAddLogDouble(&petsc_wait_all_ct, &petsc_wait_all_ct_th, 1) || PetscAddLogDouble(&petsc_sum_of_waits_ct, &petsc_sum_of_waits_ct_th, count) || MPIX_Waitall_enqueue((count), (array_of_requests), (array_of_statuses))) 604 #else 605 606 #define MPI_Startall_irecv(count, datatype, number, requests) ((number) && MPI_Startall((number), (requests))) 607 608 #define MPI_Startall_isend(count, datatype, number, requests) ((number) && MPI_Startall((number), (requests))) 609 610 #define MPI_Start_isend(count, datatype, requests) (MPI_Start(requests)) 611 612 #endif /* !MPIUNI_H && ! PETSC_HAVE_BROKEN_RECURSIVE_MACRO */ 613 614 #else /* ---Logging is turned off --------------------------------------------*/ 615 616 #define PetscLogGetState(a) (*(a) = NULL, PETSC_SUCCESS) 617 #define PetscLogGetDefaultHandler(a) (*(a) = NULL, PETSC_SUCCESS) 618 #define PetscLogHandlerStart(a) ((void)(a), PETSC_SUCCESS) 619 #define PetscLogHandlerStop(a) ((void)(a), PETSC_SUCCESS) 620 621 #define PetscLogFlops(n) ((void)(n), PETSC_SUCCESS) 622 #define PetscGetFlops(a) (*(a) = 0.0, PETSC_SUCCESS) 623 624 #define PetscLogStageRegister(a, b) ((void)(a), *(b) = -1, PETSC_SUCCESS) 625 #define PetscLogStagePush(a) ((void)(a), PETSC_SUCCESS) 626 #define PetscLogStagePop() PETSC_SUCCESS 627 #define PetscLogStageSetActive(a, b) ((void)(a), (void)(b), PETSC_SUCCESS) 628 #define PetscLogStageGetActive(a, b) ((void)(a), *(b) = PETSC_FALSE, PETSC_SUCCESS) 629 #define PetscLogStageGetVisible(a, b) ((void)(a), *(b) = PETSC_FALSE, PETSC_SUCCESS) 630 #define PetscLogStageSetVisible(a, b) ((void)(a), (void)(b), PETSC_SUCCESS) 631 #define PetscLogStageGetId(a, b) ((void)(a), *(b) = -1, PETSC_SUCCESS) 632 #define PetscLogStageGetName(a, b) ((void)(a), *(b) = NULL, PETSC_SUCCESS) 633 #define PetscLogStageGetPerfInfo(a, b) ((void)(a), *(b) = (const PetscEventPerfInfo){0}, PETSC_SUCCESS) 634 635 #define PetscLogEventRegister(a, b, c) ((void)(a), (void)(b), *(c) = -1, PETSC_SUCCESS) 636 #define PetscLogEventSetCollective(a, b) ((void)(a), (void)(b), PETSC_SUCCESS) 637 #define PetscLogEventIncludeClass(a) ((void)(a), PETSC_SUCCESS) 638 #define PetscLogEventExcludeClass(a) ((void)(a), PETSC_SUCCESS) 639 #define PetscLogEventActivate(a) ((void)(a), PETSC_SUCCESS) 640 #define PetscLogEventDeactivate(a) ((void)(a), PETSC_SUCCESS) 641 #define PetscLogEventDeactivatePush(a) ((void)(a), PETSC_SUCCESS) 642 #define PetscLogEventDeactivatePop(a) ((void)(a), PETSC_SUCCESS) 643 #define PetscLogEventActivateClass(a) ((void)(a), PETSC_SUCCESS) 644 #define PetscLogEventDeactivateClass(a) ((void)(a), PETSC_SUCCESS) 645 #define PetscLogEventSetActiveAll(a, b) ((void)(a), PETSC_SUCCESS) 646 #define PetscLogEventGetId(a, b) ((void)(a), *(b) = -1, PETSC_SUCCESS) 647 #define PetscLogEventGetName(a, b) ((void)(a), *(b) = NULL, PETSC_SUCCESS) 648 #define PetscLogEventGetPerfInfo(a, b, c) ((void)(a), (void)(b), *(c) = (const PetscEventPerfInfo){0}, PETSC_SUCCESS) 649 650 #define PetscLogEventSetDof(a, b, c) ((void)(a), (void)(b), (void)(c), PETSC_SUCCESS) 651 #define PetscLogEventSetError(a, b, c) ((void)(a), (void)(b), (void)(c), PETSC_SUCCESS) 652 #define PetscLogEventsPause() PETSC_SUCCESS 653 #define PetscLogEventsResume() PETSC_SUCCESS 654 655 #define PetscLogClassGetClassId(a, b) (*(b) = -1, PETSC_SUCCESS) 656 #define PetscLogClassIdGetName(a, b) (*(b) = NULL, PETSC_SUCCESS) 657 658 #define PetscLogObjectCreate(h) ((void)(h), PETSC_SUCCESS) 659 #define PetscLogObjectDestroy(h) ((void)(h), PETSC_SUCCESS) 660 #define PetscLogObjectState(h, c, ...) ((void)(h), (void)(c), PETSC_SUCCESS) 661 662 #define PetscLogDefaultBegin() PETSC_SUCCESS 663 #define PetscLogNestedBegin() PETSC_SUCCESS 664 #define PetscLogTraceBegin(file) ((void)(file), PETSC_SUCCESS) 665 #define PetscLogMPEBegin() PETSC_SUCCESS 666 #define PetscLogPerfstubsBegin() PETSC_SUCCESS 667 #define PetscLogLegacyCallbacksBegin(a, b, c, d) ((void)(a), (void)(b), (void)(c), (void)(d), PETSC_SUCCESS) 668 #define PetscLogActions(a) ((void)(a), PETSC_SUCCESS) 669 #define PetscLogObjects(a) ((void)(a), PETSC_SUCCESS) 670 #define PetscLogSetThreshold(a, b) ((void)(a), (void)(b), PETSC_SUCCESS) 671 672 #define PetscLogIsActive(flag) (*(flag) = PETSC_FALSE, PETSC_SUCCESS) 673 674 #define PetscLogView(viewer) ((void)(viewer), PETSC_SUCCESS) 675 #define PetscLogViewFromOptions() PETSC_SUCCESS 676 #define PetscLogDump(c) ((void)(c), PETSC_SUCCESS) 677 #define PetscLogMPEDump(c) ((void)(c), PETSC_SUCCESS) 678 679 #define PetscLogEventSync(e, comm) ((void)(e), (void)(comm), PETSC_SUCCESS) 680 #define PetscLogEventBegin(e, o1, o2, o3, o4) ((void)(e), (void)(o1), (void)(o2), (void)(o3), PETSC_SUCCESS) 681 #define PetscLogEventEnd(e, o1, o2, o3, o4) ((void)(e), (void)(o1), (void)(o2), (void)(o3), PETSC_SUCCESS) 682 683 /* If PETSC_USE_LOG is NOT defined, these still need to be! */ 684 #define MPI_Startall_irecv(count, datatype, number, requests) ((number) && MPI_Startall(number, requests)) 685 #define MPI_Startall_isend(count, datatype, number, requests) ((number) && MPI_Startall(number, requests)) 686 #define MPI_Start_isend(count, datatype, requests) MPI_Start(requests) 687 688 #endif /* PETSC_USE_LOG */ 689 690 #define PetscPreLoadBegin(flag, name) \ 691 do { \ 692 PetscBool PetscPreLoading = flag; \ 693 int PetscPreLoadMax, PetscPreLoadIt; \ 694 PetscLogStage _stageNum; \ 695 PetscCall(PetscOptionsGetBool(NULL, NULL, "-preload", &PetscPreLoading, NULL)); \ 696 PetscPreLoadMax = (int)(PetscPreLoading); \ 697 PetscPreLoadingUsed = PetscPreLoading ? PETSC_TRUE : PetscPreLoadingUsed; \ 698 PetscCall(PetscLogStageGetId(name, &_stageNum)); \ 699 for (PetscPreLoadIt = (_stageNum == -1) ? 0 : PetscPreLoadMax; PetscPreLoadIt <= PetscPreLoadMax; PetscPreLoadIt++) { \ 700 PetscPreLoadingOn = (PetscBool)(PetscPreLoadIt < PetscPreLoadMax); \ 701 PetscCall(PetscBarrier(NULL)); \ 702 if (_stageNum == -1) PetscCall(PetscLogStageRegister(name, &_stageNum)); \ 703 PetscCall(PetscLogStageSetActive(_stageNum, (PetscBool)(PetscPreLoadIt == PetscPreLoadMax))); \ 704 PetscCall(PetscLogStagePush(_stageNum)) 705 706 #define PetscPreLoadEnd() \ 707 PetscCall(PetscLogStagePop()); \ 708 } \ 709 } \ 710 while (0) 711 712 #define PetscPreLoadStage(name) \ 713 do { \ 714 PetscCall(PetscLogStagePop()); \ 715 PetscCall(PetscLogStageGetId(name, &_stageNum)); \ 716 if (_stageNum == -1) PetscCall(PetscLogStageRegister(name, &_stageNum)); \ 717 PetscCall(PetscLogStageSetActive(_stageNum, (PetscBool)(!PetscPreLoadMax || PetscPreLoadIt))); \ 718 PetscCall(PetscLogStagePush(_stageNum)); \ 719 } while (0) 720 721 /* some vars for logging */ 722 PETSC_EXTERN PetscBool PetscPreLoadingUsed; /* true if we are or have done preloading */ 723 PETSC_EXTERN PetscBool PetscPreLoadingOn; /* true if we are currently in a preloading calculation */ 724 725 #if defined(PETSC_USE_LOG) && defined(PETSC_HAVE_DEVICE) 726 727 PETSC_EXTERN PetscErrorCode PetscLogGpuTime(void); 728 PETSC_EXTERN PetscErrorCode PetscLogGpuTimeBegin(void); 729 PETSC_EXTERN PetscErrorCode PetscLogGpuTimeEnd(void); 730 PETSC_EXTERN PetscErrorCode PetscLogGpuEnergy(void); 731 PETSC_EXTERN PetscErrorCode PetscLogGpuEnergyMeter(void); 732 PETSC_EXTERN PetscErrorCode PetscLogGpuEnergyMeterBegin(void); 733 PETSC_EXTERN PetscErrorCode PetscLogGpuEnergyMeterEnd(void); 734 735 /*@ 736 PetscLogGpuFlops - Log how many flops are performed in a calculation on the device 737 738 Input Parameter: 739 . flops - the number of flops 740 741 Level: intermediate 742 743 Notes: 744 To limit the chance of integer overflow when multiplying by a constant, represent the constant as a double, 745 not an integer. Use `PetscLogFlops`(4.0*n) not `PetscLogFlops`(4*n) 746 747 The values are also added to the total flop count for the MPI rank that is set with `PetscLogFlops()`; hence the number of flops 748 just on the CPU would be the value from set from `PetscLogFlops()` minus the value set from `PetscLogGpuFlops()` 749 750 Developer Note: 751 Currently Fortran stub generator cannot run through files in include 752 753 .seealso: [](ch_profiling), `PetscLogView()`, `PetscLogFlops()`, `PetscLogGpuTimeBegin()`, `PetscLogGpuTimeEnd()` 754 @*/ 755 static inline PetscErrorCode PetscLogGpuFlops(PetscLogDouble n) 756 { 757 PetscAssert(n >= 0, PETSC_COMM_SELF, PETSC_ERR_ARG_OUTOFRANGE, "Cannot log negative flops"); 758 PetscCall(PetscAddLogDouble(&petsc_TotalFlops, &petsc_TotalFlops_th, PETSC_FLOPS_PER_OP * n)); 759 PetscCall(PetscAddLogDouble(&petsc_gflops, &petsc_gflops_th, PETSC_FLOPS_PER_OP * n)); 760 return PETSC_SUCCESS; 761 } 762 763 static inline PetscErrorCode PetscLogGpuTimeAdd(PetscLogDouble t) 764 { 765 return PetscAddLogDouble(&petsc_gtime, &petsc_gtime_th, t); 766 } 767 768 static inline PetscErrorCode PetscLogCpuToGpu(PetscLogDouble size) 769 { 770 return PetscAddLogDoubleCnt(&petsc_ctog_ct, &petsc_ctog_sz, &petsc_ctog_ct_th, &petsc_ctog_sz_th, size); 771 } 772 773 static inline PetscErrorCode PetscLogGpuToCpu(PetscLogDouble size) 774 { 775 return PetscAddLogDoubleCnt(&petsc_gtoc_ct, &petsc_gtoc_sz, &petsc_gtoc_ct_th, &petsc_gtoc_sz_th, size); 776 } 777 778 static inline PetscErrorCode PetscLogCpuToGpuScalar(PetscLogDouble size) 779 { 780 return PetscAddLogDoubleCnt(&petsc_ctog_ct_scalar, &petsc_ctog_sz_scalar, &petsc_ctog_ct_scalar_th, &petsc_ctog_sz_scalar_th, size); 781 } 782 783 static inline PetscErrorCode PetscLogGpuToCpuScalar(PetscLogDouble size) 784 { 785 return PetscAddLogDoubleCnt(&petsc_gtoc_ct_scalar, &petsc_gtoc_sz_scalar, &petsc_gtoc_ct_scalar_th, &petsc_gtoc_sz_scalar_th, size); 786 } 787 #else 788 789 #define PetscLogCpuToGpu(a) PETSC_SUCCESS 790 #define PetscLogGpuToCpu(a) PETSC_SUCCESS 791 #define PetscLogCpuToGpuScalar(a) PETSC_SUCCESS 792 #define PetscLogGpuToCpuScalar(a) PETSC_SUCCESS 793 #define PetscLogGpuFlops(a) PETSC_SUCCESS 794 #define PetscLogGpuTime() PETSC_SUCCESS 795 #define PetscLogGpuTimeAdd(a) PETSC_SUCCESS 796 #define PetscLogGpuTimeBegin() PETSC_SUCCESS 797 #define PetscLogGpuTimeEnd() PETSC_SUCCESS 798 799 #define PetscLogGpuEnergy() PETSC_SUCCESS 800 #define PetscLogGpuEnergyMeter() PETSC_SUCCESS 801 #define PetscLogGpuEnergyMeterBegin() PETSC_SUCCESS 802 #define PetscLogGpuEnergyMeterEnd() PETSC_SUCCESS 803 804 #endif /* PETSC_USE_LOG && PETSC_HAVE_DEVICE */ 805 806 /* remove TLS defines */ 807 #undef PETSC_EXTERN_TLS 808 #undef PETSC_TLS 809 810 #include <petsclogdeprecated.h> 811