xref: /honee/src/log_events.c (revision 4ea616f4b9e156c5dc0183ae10bc12e5088c8c31)
1 // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors.
2 // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause
3 
4 #include <log_events.h>
5 #include <petsc.h>
6 
7 static PetscClassId libCEED_classid, onlineTrain_classid, sgs_model_classid, misc_classid;
8 
9 PetscLogEvent HONEE_CeedOperatorApply;
10 PetscLogEvent HONEE_CeedOperatorAssemble;
11 PetscLogEvent HONEE_CeedOperatorAssembleDiagonal;
12 PetscLogEvent HONEE_CeedOperatorAssemblePointBlockDiagonal;
13 PetscLogEvent HONEE_DivDiffFluxProjection;
14 PetscLogEvent HONEE_StrongBCInsert;
15 PetscLogEvent HONEE_StrongBCCeed;
16 PetscLogEvent HONEE_SmartRedis_Init;
17 PetscLogEvent HONEE_SmartRedis_Meta;
18 PetscLogEvent HONEE_SmartRedis_Train;
19 PetscLogEvent HONEE_TrainDataCompute;
20 PetscLogEvent HONEE_DifferentialFilter;
21 PetscLogEvent HONEE_VelocityGradientProjection;
22 PetscLogEvent HONEE_SgsModel;
23 PetscLogEvent HONEE_SgsModelDDInference;
24 PetscLogEvent HONEE_SgsModelDDData;
25 
26 PetscErrorCode RegisterLogEvents() {
27   PetscFunctionBeginUser;
28   PetscCall(PetscClassIdRegister("libCEED", &libCEED_classid));
29   PetscCall(PetscLogEventRegister("CeedOpApply", libCEED_classid, &HONEE_CeedOperatorApply));
30   PetscCall(PetscLogEventRegister("CeedOpAsm", libCEED_classid, &HONEE_CeedOperatorAssemble));
31   PetscCall(PetscLogEventRegister("CeedOpAsmD", libCEED_classid, &HONEE_CeedOperatorAssembleDiagonal));
32   PetscCall(PetscLogEventRegister("CeedOpAsmPBD", libCEED_classid, &HONEE_CeedOperatorAssemblePointBlockDiagonal));
33 
34   PetscCall(PetscClassIdRegister("onlineTrain", &onlineTrain_classid));
35   PetscCall(PetscLogEventRegister("SmartRedis_Init", onlineTrain_classid, &HONEE_SmartRedis_Init));
36   PetscCall(PetscLogEventRegister("SmartRedis_Meta", onlineTrain_classid, &HONEE_SmartRedis_Meta));
37   PetscCall(PetscLogEventRegister("SmartRedis_Train", onlineTrain_classid, &HONEE_SmartRedis_Train));
38   PetscCall(PetscLogEventRegister("TrainDataCompute", onlineTrain_classid, &HONEE_TrainDataCompute));
39 
40   PetscCall(PetscClassIdRegister("SGS Model", &sgs_model_classid));
41   PetscCall(PetscLogEventRegister("SgsModel", sgs_model_classid, &HONEE_SgsModel));
42   PetscCall(PetscLogEventRegister("SgsModelDDInfer", sgs_model_classid, &HONEE_SgsModelDDInference));
43   PetscCall(PetscLogEventRegister("SgsModelDDData", sgs_model_classid, &HONEE_SgsModelDDData));
44 
45   PetscCall(PetscClassIdRegister("Miscellaneous", &misc_classid));
46   PetscCall(PetscLogEventRegister("DiffFilter", misc_classid, &HONEE_DifferentialFilter));
47   PetscCall(PetscLogEventRegister("VeloGradProj", misc_classid, &HONEE_VelocityGradientProjection));
48   PetscCall(PetscLogEventRegister("DivDiffFluxProj", misc_classid, &HONEE_DivDiffFluxProjection));
49   PetscCall(PetscLogEventRegister("StrongBCInsert", misc_classid, &HONEE_StrongBCInsert));
50   PetscCall(PetscLogEventRegister("StrongBCCeed", misc_classid, &HONEE_StrongBCCeed));
51   PetscFunctionReturn(PETSC_SUCCESS);
52 }
53