123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131 |
- #!/usr/bin/env python3
- # Copyright 2021 the V8 project authors. All rights reserved.
- # Use of this source code is governed by a BSD-style license that can be
- # found in the LICENSE file.
- # Processes {stdout} output generated by --trace-wasm-compilation-times
- # for easier consumption by human readers.
- import sys
- def SizeInternal(number, suffix):
- if suffix == "": return "%d" % number
- if number < 10: return "%.1f%s" % (number, suffix)
- return "%d%s" % (number, suffix)
- def Size(number):
- if (number < 1024): return SizeInternal(number, "")
- number /= 1024
- if (number < 1024): return SizeInternal(number, "K")
- number /= 1024
- if (number < 1024): return SizeInternal(number, "M")
- number /= 1024
- if (number < 1024): return SizeInternal(number, "G")
- return SizeInternal(number / 1024, "T")
- modules = {}
- max_module = 0
- total_tf_time = 0
- total_tf_size = 0
- def RegisterName(raw):
- global max_module
- parts = raw.split("#")
- m = parts[0]
- if m not in modules:
- modules[m] = max_module
- max_module += 1
- def Name(raw):
- parts = raw.split("#")
- if len(modules) == 1: return "#%s" % parts[1]
- return "m%d#%s" % (modules[parts[0]], parts[1])
- class Function:
- def __init__(self, index):
- self.index = index
- self.has_lo = False
- self.has_tf = False
- self.time_lo = -1
- self.time_tf = -1
- self.mem_lo = -1
- self.mem_tf_max = -1
- self.mem_tf_total = -1
- self.name = ""
- self.size_wasm = -1
- self.size_lo = -1
- self.size_tf = -1
- def AddLine(self, words):
- assert self.index == words[2], "wrong function"
- if words[4] == "TurboFan,":
- self.AddTFLine(words)
- elif words[4] == "Liftoff,":
- self.AddLiftoffLine(words)
- else:
- raise Exception("unknown compiler: %s" % words[4])
- def AddTFLine(self, words):
- assert not self.has_tf, "duplicate TF line for %s" % self.index
- self.has_tf = True
- # 0 1 2 3 4 5 6 7 8 9 10 11
- # Compiled function #6 using TurboFan, took 0 ms and 14440 / 44656
- # 12 13 14 15 16 17 18 19
- # max/total bytes; bodysize 12 codesize 24 name wasm-function#6
- self.time_tf = int(words[6])
- self.mem_tf_max = int(words[9])
- self.mem_tf_total = int(words[11])
- self.size_tf = int(words[17])
- self.name = words[19]
- def AddLiftoffLine(self, words):
- assert self.index == words[2], "wrong function"
- assert not self.has_lo, "duplicate Liftoff line for %s" % self.index
- self.has_lo = True
- # 0 1 2 3 4 5 6 7 8 9 10 11 12
- # Compiled function #6 using Liftoff, took 0 ms and 968 bytes; bodysize 4
- # 13 14
- # codesize 68
- self.time_lo = int(words[6])
- self.mem_lo = int(words[9])
- self.size_lo = int(words[14])
- self.size_wasm = int(words[12])
- def __str__(self):
- return "%s: time %d %d mem %s %s %s size %s %s %s name %s" % (
- Name(self.index), self.time_lo, self.time_tf,
- Size(self.mem_lo), Size(self.mem_tf_max), Size(self.mem_tf_total),
- Size(self.size_wasm), Size(self.size_lo), Size(self.size_tf), self.name
- )
- funcs_dict = {}
- funcs_list = []
- if len(sys.argv) < 2 or sys.argv[1] in ("-h", "--help", "help"):
- print("Pass output file (generated with --trace-wasm-compilation-times) as "
- "argument")
- sys.exit(1)
- with open(sys.argv[1], "r") as f:
- for line in f.readlines():
- words = line.strip().split(" ")
- if words[0] != "Compiled" or words[1] != "function":
- continue
- name = words[2]
- RegisterName(name)
- if name in funcs_dict:
- func = funcs_dict[name]
- else:
- func = Function(name)
- funcs_dict[name] = func
- funcs_list.append(func)
- func.AddLine(words)
- funcs_list.sort(key=lambda fun: fun.time_tf)
- for f in funcs_list:
- print(f)
- if f.time_tf > 0: total_tf_time += f.time_tf
- if f.size_tf > 0: total_tf_size += f.size_tf
- print("Total TF time: %d" % total_tf_time)
- print("Total TF size: %d" % total_tf_size)
|