xref: /honee/src/log_events.c (revision 8a8cb6e06ce4728cc6d80ca92f8de31da49852e5)
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 FLUIDS_CeedOperatorApply;
10 PetscLogEvent FLUIDS_CeedOperatorAssemble;
11 PetscLogEvent FLUIDS_CeedOperatorAssembleDiagonal;
12 PetscLogEvent FLUIDS_CeedOperatorAssemblePointBlockDiagonal;
13 PetscLogEvent FLUIDS_SmartRedis_Init;
14 PetscLogEvent FLUIDS_SmartRedis_Meta;
15 PetscLogEvent FLUIDS_SmartRedis_Train;
16 PetscLogEvent FLUIDS_TrainDataCompute;
17 PetscLogEvent FLUIDS_DifferentialFilter;
18 PetscLogEvent FLUIDS_VelocityGradientProjection;
19 PetscLogEvent FLUIDS_SgsModel;
20 PetscLogEvent FLUIDS_SgsModelDDInference;
21 PetscLogEvent FLUIDS_SgsModelDDData;
22 
23 PetscErrorCode RegisterLogEvents() {
24   PetscFunctionBeginUser;
25   PetscCall(PetscClassIdRegister("libCEED", &libCEED_classid));
26   PetscCall(PetscLogEventRegister("CeedOpApply", libCEED_classid, &FLUIDS_CeedOperatorApply));
27   PetscCall(PetscLogEventRegister("CeedOpAsm", libCEED_classid, &FLUIDS_CeedOperatorAssemble));
28   PetscCall(PetscLogEventRegister("CeedOpAsmD", libCEED_classid, &FLUIDS_CeedOperatorAssembleDiagonal));
29   PetscCall(PetscLogEventRegister("CeedOpAsmPBD", libCEED_classid, &FLUIDS_CeedOperatorAssemblePointBlockDiagonal));
30 
31   PetscCall(PetscClassIdRegister("onlineTrain", &onlineTrain_classid));
32   PetscCall(PetscLogEventRegister("SmartRedis_Init", onlineTrain_classid, &FLUIDS_SmartRedis_Init));
33   PetscCall(PetscLogEventRegister("SmartRedis_Meta", onlineTrain_classid, &FLUIDS_SmartRedis_Meta));
34   PetscCall(PetscLogEventRegister("SmartRedis_Train", onlineTrain_classid, &FLUIDS_SmartRedis_Train));
35   PetscCall(PetscLogEventRegister("TrainDataCompute", onlineTrain_classid, &FLUIDS_TrainDataCompute));
36 
37   PetscCall(PetscClassIdRegister("SGS Model", &sgs_model_classid));
38   PetscCall(PetscLogEventRegister("SgsModel", sgs_model_classid, &FLUIDS_SgsModel));
39   PetscCall(PetscLogEventRegister("SgsModelDDInfer", sgs_model_classid, &FLUIDS_SgsModelDDInference));
40   PetscCall(PetscLogEventRegister("SgsModelDDData", sgs_model_classid, &FLUIDS_SgsModelDDData));
41 
42   PetscCall(PetscClassIdRegister("Miscellaneous", &misc_classid));
43   PetscCall(PetscLogEventRegister("DiffFilter", misc_classid, &FLUIDS_DifferentialFilter));
44   PetscCall(PetscLogEventRegister("VeloGradProj", misc_classid, &FLUIDS_VelocityGradientProjection));
45   PetscFunctionReturn(PETSC_SUCCESS);
46 }
47