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> 5b40a7e63SJames Wright #include <petsc.h> 6b40a7e63SJames Wright 7b40a7e63SJames Wright static PetscClassId libCEED_classid, onlineTrain_classid, sgs_model_classid, misc_classid; 8b40a7e63SJames Wright 9*ea615d4cSJames Wright PetscLogEvent HONEE_CeedOperatorApply; 10*ea615d4cSJames Wright PetscLogEvent HONEE_CeedOperatorAssemble; 11*ea615d4cSJames Wright PetscLogEvent HONEE_CeedOperatorAssembleDiagonal; 12*ea615d4cSJames Wright PetscLogEvent HONEE_CeedOperatorAssemblePointBlockDiagonal; 13*ea615d4cSJames Wright PetscLogEvent HONEE_DivDiffFluxProjection; 14*ea615d4cSJames Wright PetscLogEvent HONEE_SmartRedis_Init; 15*ea615d4cSJames Wright PetscLogEvent HONEE_SmartRedis_Meta; 16*ea615d4cSJames Wright PetscLogEvent HONEE_SmartRedis_Train; 17*ea615d4cSJames Wright PetscLogEvent HONEE_TrainDataCompute; 18*ea615d4cSJames Wright PetscLogEvent HONEE_DifferentialFilter; 19*ea615d4cSJames Wright PetscLogEvent HONEE_VelocityGradientProjection; 20*ea615d4cSJames Wright PetscLogEvent HONEE_SgsModel; 21*ea615d4cSJames Wright PetscLogEvent HONEE_SgsModelDDInference; 22*ea615d4cSJames Wright PetscLogEvent HONEE_SgsModelDDData; 23b40a7e63SJames Wright 24b40a7e63SJames Wright PetscErrorCode RegisterLogEvents() { 25b40a7e63SJames Wright PetscFunctionBeginUser; 26b40a7e63SJames Wright PetscCall(PetscClassIdRegister("libCEED", &libCEED_classid)); 27*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("CeedOpApply", libCEED_classid, &HONEE_CeedOperatorApply)); 28*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("CeedOpAsm", libCEED_classid, &HONEE_CeedOperatorAssemble)); 29*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("CeedOpAsmD", libCEED_classid, &HONEE_CeedOperatorAssembleDiagonal)); 30*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("CeedOpAsmPBD", libCEED_classid, &HONEE_CeedOperatorAssemblePointBlockDiagonal)); 31b40a7e63SJames Wright 32b40a7e63SJames Wright PetscCall(PetscClassIdRegister("onlineTrain", &onlineTrain_classid)); 33*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SmartRedis_Init", onlineTrain_classid, &HONEE_SmartRedis_Init)); 34*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SmartRedis_Meta", onlineTrain_classid, &HONEE_SmartRedis_Meta)); 35*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SmartRedis_Train", onlineTrain_classid, &HONEE_SmartRedis_Train)); 36*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("TrainDataCompute", onlineTrain_classid, &HONEE_TrainDataCompute)); 37b40a7e63SJames Wright 38b40a7e63SJames Wright PetscCall(PetscClassIdRegister("SGS Model", &sgs_model_classid)); 39*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SgsModel", sgs_model_classid, &HONEE_SgsModel)); 40*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SgsModelDDInfer", sgs_model_classid, &HONEE_SgsModelDDInference)); 41*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("SgsModelDDData", sgs_model_classid, &HONEE_SgsModelDDData)); 42b40a7e63SJames Wright 43b40a7e63SJames Wright PetscCall(PetscClassIdRegister("Miscellaneous", &misc_classid)); 44*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("DiffFilter", misc_classid, &HONEE_DifferentialFilter)); 45*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("VeloGradProj", misc_classid, &HONEE_VelocityGradientProjection)); 46*ea615d4cSJames Wright PetscCall(PetscLogEventRegister("DivDiffFluxProj", misc_classid, &HONEE_DivDiffFluxProjection)); 47b40a7e63SJames Wright PetscFunctionReturn(PETSC_SUCCESS); 48b40a7e63SJames Wright } 49