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