1c4762a1bSJed Brown 2c4762a1bSJed Brown static char help[] = "Augmenting PETSc profiling by add events.\n\ 3c4762a1bSJed Brown Run this program with one of the\n\ 4c4762a1bSJed Brown following options to generate logging information: -log, -log_view,\n\ 5c4762a1bSJed Brown -log_all. The PETSc routines automatically log event times and flops,\n\ 6c4762a1bSJed Brown so this monitoring is intended solely for users to employ in application\n\ 7c4762a1bSJed Brown codes.\n\n"; 8c4762a1bSJed Brown 9c4762a1bSJed Brown /*T 10c4762a1bSJed Brown Concepts: PetscLog^user-defined event profiling 11c4762a1bSJed Brown Concepts: profiling^user-defined event 12c4762a1bSJed Brown Concepts: PetscLog^activating/deactivating events for profiling 13c4762a1bSJed Brown Concepts: profiling^activating/deactivating events 14c4762a1bSJed Brown Processors: n 15c4762a1bSJed Brown T*/ 16c4762a1bSJed Brown 17c4762a1bSJed Brown /* 18c4762a1bSJed Brown Include "petscsys.h" so that we can use PETSc profiling routines. 19c4762a1bSJed Brown */ 20c4762a1bSJed Brown #include <petscsys.h> 21c4762a1bSJed Brown #include <petscviewer.h> 22c4762a1bSJed Brown 23c4762a1bSJed Brown int main(int argc,char **argv) 24c4762a1bSJed Brown { 25c4762a1bSJed Brown PetscErrorCode ierr; 26c4762a1bSJed Brown PetscMPIInt rank; 27c4762a1bSJed Brown int i,imax=10000,icount; 28c4762a1bSJed Brown PetscLogEvent USER_EVENT,check_USER_EVENT; 29c4762a1bSJed Brown 30c4762a1bSJed Brown ierr = PetscInitialize(&argc,&argv,NULL,help);if (ierr) return ierr; 31c4762a1bSJed Brown 32c4762a1bSJed Brown /* 33c4762a1bSJed Brown Create a new user-defined event. 34c4762a1bSJed Brown - Note that PetscLogEventRegister() returns to the user a unique 35c4762a1bSJed Brown integer event number, which should then be used for profiling 36c4762a1bSJed Brown the event via PetscLogEventBegin() and PetscLogEventEnd(). 37c4762a1bSJed Brown - The user can also optionally log floating point operations 38c4762a1bSJed Brown with the routine PetscLogFlops(). 39c4762a1bSJed Brown */ 40*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventRegister("User event",PETSC_VIEWER_CLASSID,&USER_EVENT)); 41*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventGetId("User event",&check_USER_EVENT)); 422c71b3e2SJacob Faibussowitsch PetscCheckFalse(USER_EVENT != check_USER_EVENT,PETSC_COMM_SELF,PETSC_ERR_PLIB,"Event Ids do not match"); 43c4762a1bSJed Brown 44*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(USER_EVENT,0,0,0,0)); 45c4762a1bSJed Brown icount = 0; 46c4762a1bSJed Brown for (i=0; i<imax; i++) icount++; 47*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogFlops(imax)); 48*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSleep(0.5)); 49*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(USER_EVENT,0,0,0,0)); 50c4762a1bSJed Brown 51c4762a1bSJed Brown /* 52c4762a1bSJed Brown We disable the logging of an event. 53c4762a1bSJed Brown 54c4762a1bSJed Brown */ 55*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventDeactivate(USER_EVENT)); 56*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(USER_EVENT,0,0,0,0)); 57*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSleep(0.5)); 58*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(USER_EVENT,0,0,0,0)); 59c4762a1bSJed Brown 60c4762a1bSJed Brown /* 61c4762a1bSJed Brown We next enable the logging of an event 62c4762a1bSJed Brown */ 63*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventActivate(USER_EVENT)); 64*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(USER_EVENT,0,0,0,0)); 65*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSleep(0.5)); 66*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(USER_EVENT,0,0,0,0)); 67c4762a1bSJed Brown 68c4762a1bSJed Brown /* 69c4762a1bSJed Brown We test event logging imbalance 70c4762a1bSJed Brown */ 71*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Comm_rank(PETSC_COMM_WORLD,&rank)); 72*5f80ce2aSJacob Faibussowitsch if (rank == 0) CHKERRQ(PetscSleep(0.5)); 73*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventSync(USER_EVENT,PETSC_COMM_WORLD)); 74*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventBegin(USER_EVENT,0,0,0,0)); 75*5f80ce2aSJacob Faibussowitsch CHKERRMPI(MPI_Barrier(PETSC_COMM_WORLD)); 76*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscSleep(0.5)); 77*5f80ce2aSJacob Faibussowitsch CHKERRQ(PetscLogEventEnd(USER_EVENT,0,0,0,0)); 78c4762a1bSJed Brown 79c4762a1bSJed Brown ierr = PetscFinalize(); 80c4762a1bSJed Brown return ierr; 81c4762a1bSJed Brown } 82c4762a1bSJed Brown 83c4762a1bSJed Brown /*TEST 84c4762a1bSJed Brown 85c4762a1bSJed Brown build: 86dfd57a17SPierre Jolivet requires: defined(PETSC_USE_LOG) 87c4762a1bSJed Brown 88c4762a1bSJed Brown test: 89c4762a1bSJed Brown 90c4762a1bSJed Brown TEST*/ 91