1 | MODULE profiling_mod |
---|
2 | IMPLICIT NONE |
---|
3 | SAVE |
---|
4 | PRIVATE |
---|
5 | |
---|
6 | INTEGER, PARAMETER :: max_id=20, max_depth=10 |
---|
7 | INTEGER :: nb_id=-10, depth=-10, current_id(max_depth) |
---|
8 | CHARACTER(10), DIMENSION(max_id) :: name |
---|
9 | REAL :: chrono(max_depth), elapsed(max_id) |
---|
10 | |
---|
11 | PUBLIC :: init_profiling, register_id, enter_profile, exit_profile, print_profile |
---|
12 | |
---|
13 | CONTAINS |
---|
14 | |
---|
15 | SUBROUTINE init_profiling |
---|
16 | !$OMP MASTER |
---|
17 | nb_id=0 |
---|
18 | depth=0 |
---|
19 | elapsed(:)=0 |
---|
20 | !$OMP END MASTER |
---|
21 | END SUBROUTINE init_profiling |
---|
22 | |
---|
23 | SUBROUTINE register_id(thename, id) |
---|
24 | CHARACTER(*), INTENT(IN) :: thename |
---|
25 | INTEGER, INTENT(OUT) :: id |
---|
26 | !$OMP MASTER |
---|
27 | nb_id = nb_id+1 |
---|
28 | id = nb_id |
---|
29 | name(id)=thename |
---|
30 | !$OMP END MASTER |
---|
31 | END SUBROUTINE register_id |
---|
32 | |
---|
33 | FUNCTION get_elapsed(start) |
---|
34 | INTEGER(kind=8) :: count, count_rate |
---|
35 | REAL :: start,get_elapsed |
---|
36 | CALL SYSTEM_CLOCK(count,count_rate) |
---|
37 | get_elapsed = (1.*count)/(1.*count_rate) - start |
---|
38 | IF(get_elapsed<0.) get_elapsed=0. |
---|
39 | END FUNCTION get_elapsed |
---|
40 | |
---|
41 | SUBROUTINE enter_profile(id) |
---|
42 | INTEGER, INTENT(IN) :: id |
---|
43 | REAL :: my_chrono |
---|
44 | !$OMP MASTER |
---|
45 | depth = depth+1 |
---|
46 | chrono(depth) = get_elapsed(0.) |
---|
47 | current_id(depth) = id |
---|
48 | !$OMP END MASTER |
---|
49 | END SUBROUTINE enter_profile |
---|
50 | |
---|
51 | SUBROUTINE exit_profile(id) |
---|
52 | INTEGER, INTENT(IN) :: id |
---|
53 | INTEGER :: parent_id |
---|
54 | REAL :: my_elapsed |
---|
55 | !$OMP MASTER |
---|
56 | IF(depth<=0) THEN |
---|
57 | PRINT *, 'exit_profile called at depth=0 !!' |
---|
58 | STOP |
---|
59 | END IF |
---|
60 | IF(id /= current_id(depth)) THEN |
---|
61 | PRINT *,' exit_profile : at depth ', depth, ' exiting ', TRIM(name(id)), ' after entering ', TRIM(name(current_id(depth))) |
---|
62 | STOP |
---|
63 | END IF |
---|
64 | my_elapsed = get_elapsed(chrono(depth)) |
---|
65 | ! add elapsed to current profile |
---|
66 | elapsed(id) = elapsed(id) + my_elapsed |
---|
67 | depth = depth-1 |
---|
68 | ! and substract from parent profile |
---|
69 | IF(depth>0) THEN |
---|
70 | parent_id = current_id(depth) |
---|
71 | elapsed(parent_id) = elapsed(parent_id) - my_elapsed |
---|
72 | END IF |
---|
73 | !$OMP END MASTER |
---|
74 | END SUBROUTINE exit_profile |
---|
75 | |
---|
76 | SUBROUTINE print_profile |
---|
77 | INTEGER :: i |
---|
78 | REAL :: total |
---|
79 | !$OMP MASTER |
---|
80 | PRINT *, '---------------------- Profiling -----------------------' |
---|
81 | total = SUM(elapsed(1:nb_id)) |
---|
82 | PRINT *, 'Total (s) : ', total |
---|
83 | DO i=1,nb_id |
---|
84 | PRINT *, name(i), INT(elapsed(i)), INT(1000.*elapsed(i)/total) |
---|
85 | END DO |
---|
86 | PRINT *, '---------------------- Profiling -----------------------' |
---|
87 | !$OMP END MASTER |
---|
88 | END SUBROUTINE print_profile |
---|
89 | |
---|
90 | END MODULE profiling_mod |
---|