show_profile.py 1.1 KB
Newer Older
M
MRXLT 已提交
1 2 3 4 5 6 7 8 9 10 11 12
#coding=utf-8
import sys
import collections

profile_file = sys.argv[1]
thread_num = sys.argv[2]
time_dict = collections.OrderedDict()


def prase(line):
    profile_list = line.split(" ")
    num = len(profile_list)
M
MRXLT 已提交
13
    for idx in range(int(num / 2)):
M
MRXLT 已提交
14 15 16 17 18 19 20
        profile_0_list = profile_list[idx * 2].split(":")
        profile_1_list = profile_list[idx * 2 + 1].split(":")
        if len(profile_0_list[0].split("_")) == 2:
            name = profile_0_list[0].split("_")[0]
        else:
            name = profile_0_list[0].split("_")[0] + "_" + profile_0_list[
                0].split("_")[1]
M
MRXLT 已提交
21
        cost = int(profile_1_list[1]) - int(profile_0_list[1])
M
MRXLT 已提交
22 23 24 25 26 27 28 29 30 31
        if name not in time_dict:
            time_dict[name] = cost
        else:
            time_dict[name] += cost


with open(profile_file) as f:
    for line in f.readlines():
        line = line.strip().split("\t")
        if line[0] == "PROFILE":
M
MRXLT 已提交
32
            prase(line[2])
M
MRXLT 已提交
33

G
gentelyang 已提交
34
print("thread_num: {}".format(thread_num))
M
MRXLT 已提交
35
for name in time_dict:
G
gentelyang 已提交
36
    print("{} cost: {}s in each thread ".format(name, time_dict[name] / (
M
MRXLT 已提交
37
        1000000.0 * float(thread_num))))