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 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; 18*e171caa6SJames 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 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)); 37*e171caa6SJames 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