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