mirror of https://gitlab.com/scemama/irpf90.git
93 lines
2.0 KiB
Python
93 lines
2.0 KiB
Python
#!/usr/bin/python
|
|
|
|
rdtsc = """
|
|
#define uint64_t unsigned long
|
|
//#ifdef __i386
|
|
uint64_t irp_rdtsc_() {
|
|
uint64_t x;
|
|
__asm__ volatile ("rdtsc" : "=A" (x));
|
|
return x;
|
|
}
|
|
#if __amd64
|
|
uint64_t irp_rdtsc_() {
|
|
uint64_t a, d;
|
|
__asm__ volatile ("rdtsc" : "=a" (a), "=d" (d));
|
|
return (d<<32) | a;
|
|
}
|
|
#endif
|
|
"""
|
|
|
|
import subprocess
|
|
import tempfile
|
|
import os
|
|
from variables import variables
|
|
|
|
def build_rdtsc():
|
|
file,filename = tempfile.mkstemp()
|
|
filename += ".c"
|
|
file = open(filename,'w')
|
|
file.write(rdtsc)
|
|
file.close()
|
|
p = subprocess.Popen(["gcc","-O3",filename,"-c","-o","IRPF90_temp/irp_rdtsc.o"])
|
|
p.communicate()
|
|
os.remove(filename)
|
|
|
|
def build_module():
|
|
data = """
|
|
module irp_timer
|
|
double precision :: irp_profile(2,%(n)d)
|
|
character*(64) :: irp_profile_label(%(n)d)
|
|
end module
|
|
|
|
subroutine irp_init_timer
|
|
use irp_timer
|
|
implicit none
|
|
irp_profile = 0.
|
|
%(text)s
|
|
end
|
|
|
|
subroutine irp_set_timer(i,value)
|
|
use irp_timer
|
|
implicit none
|
|
integer, intent(in) :: i
|
|
integer*8, intent(in) :: value
|
|
irp_profile(1,i) = dble(value)
|
|
irp_profile(2,i) = irp_profile(2,i)+1.d0
|
|
end
|
|
|
|
subroutine irp_print_timer()
|
|
use irp_timer
|
|
implicit none
|
|
integer :: i
|
|
print '(16X,5(2X,A16))', 'Num Calls', 'Tot Cycles', 'Avge Cycles', &
|
|
'Tot Secs(1GHz)', 'Avge Secs(1GHz)'
|
|
do i=1,%(n)d
|
|
if (irp_profile(2,i) > 0.) then
|
|
print '(A16,5(2X,F16.4))', irp_profile_label(i), irp_profile(2,i), &
|
|
irp_profile(1,i), irp_profile(1,i)/irp_profile(2,i), &
|
|
irp_profile(1,i)*1.d-9, 1.d-9*irp_profile(1,i)/irp_profile(2,i)
|
|
endif
|
|
enddo
|
|
end
|
|
"""
|
|
label = {}
|
|
for i in variables:
|
|
vi = variables[i]
|
|
label[vi.label] = vi.name
|
|
text = []
|
|
for l in label:
|
|
text.append(" irp_profile_label(%d) = '%s'"%(l,label[l]))
|
|
text.sort()
|
|
text = '\n'.join(text)
|
|
data = data%{'text': text, 'n':len(label.keys())}
|
|
file = open("IRPF90_temp/irp_profile.irp.F90",'w')
|
|
file.write(data)
|
|
file.close()
|
|
|
|
def run():
|
|
build_module()
|
|
build_rdtsc()
|
|
|
|
if __name__ == "__main__":
|
|
build_rdtsc()
|