Parse ytiming.py
Jump to navigation
Jump to search
Save the following text into a file called parse_ytiming.py
#! /usr/bin/env python3
import sys
def parse_report(file_report,fmt_seconds):
#
parse_timing=False
outd={}
outd["parallel_structure"]=""
outd["mpi_tasks"]=1
outd["openmp_tasks"]=1
outd["ncores"]=1
outd["completed"]=False
#
fl=open(file_report,"r")
#print ("parsing "+file_report)
#
for line in fl:
#
# versioning
#
if "Version" in line:
s=line.split()
outd["version"]=s[s.index("Version")+1]
if "Revision" in line:
s=line.split()
outd["revision"]=s[s.index("Revision")+1]
if "Hash" in line:
s=line.split()
outd["hash"]=s[s.index("Hash")+1]
#
# parallel data
#
if "CPU-Threads" in line:
outd["parallel_structure"]=outd["parallel_structure"]+" " \
+line.split(":")[1].replace("\n"," ")
if "MPI CPU" in line:
outd["mpi_tasks"]=int(line.split(":")[-1])
if "THREADS " in line:
outd["openmp_threads"]=int(line.split(":")[-1])
if "THREADS TOT(max)" in line:
outd["ncores"]=int(line.split(":")[-1])
#
# timing
#
if "Timing Overview" in line: parse_timing=True
if "| The users of YAMBO have little formal obligations " in line:
parse_timing=False
outd["completed"]=True
#
if parse_timing and "Timing " in line:
outd["wall_time"] = line.split(":")[1].split("/")[0].replace("-","")
if parse_timing and "io_WF" in line:
outd["io_WF_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "io_HF" in line:
outd["io_HF_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "io_DIPOLES" in line:
outd["io_DIPOLES_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "io_X" in line:
outd["io_X_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "RIM" in line:
outd["RIM_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "Coulomb Cutoff" in line:
outd["cutoff_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "Dipoles :" in line:
outd["dip_time"] = line.split(":")[1].split("CPU")[0]
#if parse_timing and "Dipoles (REDUX)" in line:
# outd["dip_REDUX_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "Xo (procedure)" in line:
outd["Xo_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "Xo (REDUX)" in line:
outd["Xo_REDUX_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "X (procedure)" in line:
outd["X_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "X (REDUX)" in line:
outd["X_REDUX_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "LINEAR ALGEBRA" in line:
outd["LinAlG_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and " HF " in line:
outd["HF_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "HF(REDUX)" in line:
outd["HF_REDUX_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "GW(ppa)" in line:
outd["GW_time"] = line.split(":")[1].split("CPU")[0]
if parse_timing and "GW(REDUX)" in line:
outd["GW_REDUX_time"] = line.split(":")[1].split("CPU")[0]
fl.close()
clean_parsed_data(outd)
#
if fmt_seconds:
outd_sec=convert_to_sec(outd)
outd=outd_sec
#if "dip_time_sec" in outd and "dip_REDUX_time_sec" in outd : outd["dip_tot_time_sec"]=outd["dip_time_sec"]+outd["dip_REDUX_time_sec"]
if "Xo_time_sec" in outd and "Xo_REDUX_time_sec" in outd : outd["Xo_tot_time_sec"]=outd["Xo_time_sec"]+outd["Xo_REDUX_time_sec"]
if "X_time_sec" in outd and "X_REDUX_time_sec" in outd : outd["X_tot_time_sec"]=outd["X_time_sec"]+outd["X_REDUX_time_sec"]
if "HF_time_sec" in outd and "HF_REDUX_time_sec" in outd : outd["HF_tot_time_sec"]=outd["HF_time_sec"]+outd["HF_REDUX_time_sec"]
if "GW_time_sec" in outd and "GW_REDUX_time_sec" in outd : outd["GW_tot_time_sec"]=outd["GW_time_sec"]+outd["GW_REDUX_time_sec"]
#
#print(outd)
#print()
return outd
def convert_to_sec(outd):
outd_new={}
for key,val in outd.items():
if isinstance(val,str) and "_time" in key :
key_new=key+"_sec"
val_new=0
if "d" in val:
val_new+= 86400*int(val.split("d")[0])
val=val.split("d")[1]
if "h" in val:
val_new+= 3600*int(val.split("h")[0])
val=val.split("h")[1]
if "m" in val:
val_new+= 60*int(val.split('m')[0])
val=val.split("m")[1]
if "s" in val:
val_new+=int(float(val.split("s")[0]))
outd_new[key_new]=val_new
else:
outd_new[key]=val
#
return outd_new
def clean_parsed_data(outd):
#
for key in outd:
val=outd[key]
if isinstance(val,str):
val=val.replace(" ","")
outd[key]=val
def print_parsed_data(outd,datafile=None):
print()
if datafile != None:
print(" %20s : %s " % ("file", str(datafile)))
#
for key in outd:
print(" %20s : %s " % (key, str(outd[key])) )
print()
def formatter(val,maxfield=12):
sval=str(val)
return " "*(maxfield-len(sval))+sval
def print_perfGW_data(outd=None,fmt_seconds=False,header=False):
#
if outd==None: return
if header :
strout ="# ncores"
if "mpi_tasks" in outd : strout+=formatter("MPI")
if "openmp_threads"in outd : strout+=formatter("threads")
#
if not fmt_seconds:
if "dip_time" in outd : strout+=formatter("dip")
if "dip_REDUX_time"in outd : strout+=formatter("dip_REDUX")
if "Xo_time" in outd : strout+=formatter("Xo")
if "Xo_REDUX_time" in outd : strout+=formatter("Xo_REDUX")
if "X_time" in outd : strout+=formatter("X")
if "X_REDUX_time" in outd : strout+=formatter("X_REDUX")
if "io_X_time" in outd : strout+=formatter("io_X")
if "io_WF_time" in outd : strout+=formatter("io_WF")
if "HF_time" in outd : strout+=formatter("Sgm_x")
if "HF_REDUX_time" in outd : strout+=formatter("(REDUX)")
if "GW_time" in outd : strout+=formatter("Sgm_c")
if "GW_REDUX_time" in outd : strout+=formatter("(REDUX)")
if "wall_time" in outd : strout+=formatter("WALL_TIME")
else:
if "dip_time_sec" in outd : strout+=formatter("dip_TOT")
if "Xo_time_sec" in outd : strout+=formatter("Xo_TOT")
if "X_time_sec" in outd : strout+=formatter("X_TOT")
if "io_X_time_sec" in outd : strout+=formatter("io_X")
if "io_WF_time_sec"in outd : strout+=formatter("io_WF")
if "HF_time_sec" in outd : strout+=formatter("Sgm_x_TOT")
if "GW_time_sec" in outd : strout+=formatter("Sgm_c_TOT")
if "wall_time_sec" in outd : strout+=formatter("WALL_TIME")
#
print(strout)
return
strout = formatter(outd["ncores"],len("# ncores"))
if "mpi_tasks" in outd : strout+= formatter(outd["mpi_tasks"])
if "openmp_threads"in outd : strout+= formatter(outd["openmp_threads"])
if not outd["completed"]:
print(strout)
return
if not fmt_seconds:
if "dip_time" in outd : strout+= formatter(outd["dip_time"])
if "dip_REDUX_time" in outd : strout+= formatter(outd["dip_REDUX_time"])
if "Xo_time" in outd : strout+= formatter(outd["Xo_time"])
if "Xo_REDUX_time" in outd : strout+= formatter(outd["Xo_REDUX_time"])
if "X_time" in outd : strout+= formatter(outd["X_time"])
if "X_REDUX_time" in outd : strout+= formatter(outd["X_REDUX_time"])
if "io_X_time" in outd : strout+= formatter(outd["io_X_time"])
if "io_WF_time" in outd : strout+= formatter(outd["io_WF_time"])
if "HF_time" in outd : strout+= formatter(outd["HF_time"])
if "HF_REDUX_time" in outd : strout+= formatter(outd["HF_REDUX_time"])
if "GW_time" in outd : strout+= formatter(outd["GW_time"])
if "GW_REDUX_time" in outd : strout+= formatter(outd["GW_REDUX_time"])
if "wall_time" in outd : strout+= formatter(outd["wall_time"])
else:
#if "dip_tot_time_sec"in outd : strout+= formatter(str(outd["dip_tot_time_sec"]))
if "dip_time_sec" in outd : strout+= formatter(str(outd["dip_time_sec"]))
if "Xo_tot_time_sec" in outd : strout+= formatter(str(outd["Xo_tot_time_sec"]))
if "X_tot_time_sec" in outd : strout+= formatter(str(outd["X_tot_time_sec"]))
if "io_X_time_sec" in outd : strout+= formatter(str(outd["io_X_time_sec"]))
if "io_WF_time_sec" in outd : strout+= formatter(str(outd["io_WF_time_sec"]))
if "HF_tot_time_sec" in outd : strout+= formatter(str(outd["HF_tot_time_sec"]))
if "GW_tot_time_sec" in outd : strout+= formatter(str(outd["GW_tot_time_sec"]))
if "wall_time_sec" in outd : strout+= formatter(str(outd["wall_time_sec"]))
print(strout)
return
def main(argv):
#
#
usage_str='''
Usage parse_Yambo.py [-h,--help] [-s, --sec] [-a, --all] file1 file2...
'''
#
verbose=False
fmt_seconds=False
#
if len(argv) == 1:
print(usage_str)
sys.exit(1)
if argv[1]=="-h" or argv[1]=="--help":
print(usage_str)
sys.exit(0)
if argv[1]=="-a" or argv[1]=="--all":
argv.pop(1)
verbose=True
if argv[1]=="-s" or argv[1]=="--sec":
argv.pop(1)
fmt_seconds=True
#
flist=argv
first=True
for f in flist[1:]:
#
outd=parse_report(f,fmt_seconds)
#
if not verbose and first : print_perfGW_data(outd,fmt_seconds,header=True)
#
first=False
#
if verbose:
print_parsed_data(outd,datafile=f)
else:
print_perfGW_data(outd,fmt_seconds)
if __name__ == "__main__":
main(sys.argv)