xref: /honee/src/log_events.c (revision ae2b091fac884a554e48acc4b4c187524c2a2818)
1*ae2b091fSJames Wright // SPDX-FileCopyrightText: Copyright (c) 2017-2024, HONEE contributors.
2*ae2b091fSJames Wright // SPDX-License-Identifier: Apache-2.0 OR BSD-2-Clause
3b40a7e63SJames Wright 
4b40a7e63SJames Wright #include <log_events.h>
5b40a7e63SJames Wright #include <petsc.h>
6b40a7e63SJames Wright 
7b40a7e63SJames Wright static PetscClassId libCEED_classid, onlineTrain_classid, sgs_model_classid, misc_classid;
8b40a7e63SJames Wright 
9b40a7e63SJames Wright PetscLogEvent FLUIDS_CeedOperatorApply;
10b40a7e63SJames Wright PetscLogEvent FLUIDS_CeedOperatorAssemble;
11b40a7e63SJames Wright PetscLogEvent FLUIDS_CeedOperatorAssembleDiagonal;
12b40a7e63SJames Wright PetscLogEvent FLUIDS_CeedOperatorAssemblePointBlockDiagonal;
13b40a7e63SJames Wright PetscLogEvent FLUIDS_SmartRedis_Init;
14b40a7e63SJames Wright PetscLogEvent FLUIDS_SmartRedis_Meta;
15b40a7e63SJames Wright PetscLogEvent FLUIDS_SmartRedis_Train;
16b40a7e63SJames Wright PetscLogEvent FLUIDS_TrainDataCompute;
17b40a7e63SJames Wright PetscLogEvent FLUIDS_DifferentialFilter;
18b40a7e63SJames Wright PetscLogEvent FLUIDS_VelocityGradientProjection;
19b40a7e63SJames Wright PetscLogEvent FLUIDS_SgsModel;
20b40a7e63SJames Wright PetscLogEvent FLUIDS_SgsModelDDInference;
21b40a7e63SJames Wright PetscLogEvent FLUIDS_SgsModelDDData;
22b40a7e63SJames Wright 
23b40a7e63SJames Wright PetscErrorCode RegisterLogEvents() {
24b40a7e63SJames Wright   PetscFunctionBeginUser;
25b40a7e63SJames Wright   PetscCall(PetscClassIdRegister("libCEED", &libCEED_classid));
26b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("CeedOpApply", libCEED_classid, &FLUIDS_CeedOperatorApply));
27b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("CeedOpAsm", libCEED_classid, &FLUIDS_CeedOperatorAssemble));
28b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("CeedOpAsmD", libCEED_classid, &FLUIDS_CeedOperatorAssembleDiagonal));
29b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("CeedOpAsmPBD", libCEED_classid, &FLUIDS_CeedOperatorAssemblePointBlockDiagonal));
30b40a7e63SJames Wright 
31b40a7e63SJames Wright   PetscCall(PetscClassIdRegister("onlineTrain", &onlineTrain_classid));
32b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SmartRedis_Init", onlineTrain_classid, &FLUIDS_SmartRedis_Init));
33b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SmartRedis_Meta", onlineTrain_classid, &FLUIDS_SmartRedis_Meta));
34b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SmartRedis_Train", onlineTrain_classid, &FLUIDS_SmartRedis_Train));
35b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("TrainDataCompute", onlineTrain_classid, &FLUIDS_TrainDataCompute));
36b40a7e63SJames Wright 
37b40a7e63SJames Wright   PetscCall(PetscClassIdRegister("SGS Model", &sgs_model_classid));
38b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SgsModel", sgs_model_classid, &FLUIDS_SgsModel));
39b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SgsModelDDInfer", sgs_model_classid, &FLUIDS_SgsModelDDInference));
40b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("SgsModelDDData", sgs_model_classid, &FLUIDS_SgsModelDDData));
41b40a7e63SJames Wright 
42b40a7e63SJames Wright   PetscCall(PetscClassIdRegister("Miscellaneous", &misc_classid));
43b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("DiffFilter", misc_classid, &FLUIDS_DifferentialFilter));
44b40a7e63SJames Wright   PetscCall(PetscLogEventRegister("VeloGradProj", misc_classid, &FLUIDS_VelocityGradientProjection));
45b40a7e63SJames Wright   PetscFunctionReturn(PETSC_SUCCESS);
46b40a7e63SJames Wright }
47