process-wasm-compilation-times.py 3.9 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131
  1. #!/usr/bin/env python3
  2. # Copyright 2021 the V8 project authors. All rights reserved.
  3. # Use of this source code is governed by a BSD-style license that can be
  4. # found in the LICENSE file.
  5. # Processes {stdout} output generated by --trace-wasm-compilation-times
  6. # for easier consumption by human readers.
  7. import sys
  8. def SizeInternal(number, suffix):
  9. if suffix == "": return "%d" % number
  10. if number < 10: return "%.1f%s" % (number, suffix)
  11. return "%d%s" % (number, suffix)
  12. def Size(number):
  13. if (number < 1024): return SizeInternal(number, "")
  14. number /= 1024
  15. if (number < 1024): return SizeInternal(number, "K")
  16. number /= 1024
  17. if (number < 1024): return SizeInternal(number, "M")
  18. number /= 1024
  19. if (number < 1024): return SizeInternal(number, "G")
  20. return SizeInternal(number / 1024, "T")
  21. modules = {}
  22. max_module = 0
  23. total_tf_time = 0
  24. total_tf_size = 0
  25. def RegisterName(raw):
  26. global max_module
  27. parts = raw.split("#")
  28. m = parts[0]
  29. if m not in modules:
  30. modules[m] = max_module
  31. max_module += 1
  32. def Name(raw):
  33. parts = raw.split("#")
  34. if len(modules) == 1: return "#%s" % parts[1]
  35. return "m%d#%s" % (modules[parts[0]], parts[1])
  36. class Function:
  37. def __init__(self, index):
  38. self.index = index
  39. self.has_lo = False
  40. self.has_tf = False
  41. self.time_lo = -1
  42. self.time_tf = -1
  43. self.mem_lo = -1
  44. self.mem_tf_max = -1
  45. self.mem_tf_total = -1
  46. self.name = ""
  47. self.size_wasm = -1
  48. self.size_lo = -1
  49. self.size_tf = -1
  50. def AddLine(self, words):
  51. assert self.index == words[2], "wrong function"
  52. if words[4] == "TurboFan,":
  53. self.AddTFLine(words)
  54. elif words[4] == "Liftoff,":
  55. self.AddLiftoffLine(words)
  56. else:
  57. raise Exception("unknown compiler: %s" % words[4])
  58. def AddTFLine(self, words):
  59. assert not self.has_tf, "duplicate TF line for %s" % self.index
  60. self.has_tf = True
  61. # 0 1 2 3 4 5 6 7 8 9 10 11
  62. # Compiled function #6 using TurboFan, took 0 ms and 14440 / 44656
  63. # 12 13 14 15 16 17 18 19
  64. # max/total bytes; bodysize 12 codesize 24 name wasm-function#6
  65. self.time_tf = int(words[6])
  66. self.mem_tf_max = int(words[9])
  67. self.mem_tf_total = int(words[11])
  68. self.size_tf = int(words[17])
  69. self.name = words[19]
  70. def AddLiftoffLine(self, words):
  71. assert self.index == words[2], "wrong function"
  72. assert not self.has_lo, "duplicate Liftoff line for %s" % self.index
  73. self.has_lo = True
  74. # 0 1 2 3 4 5 6 7 8 9 10 11 12
  75. # Compiled function #6 using Liftoff, took 0 ms and 968 bytes; bodysize 4
  76. # 13 14
  77. # codesize 68
  78. self.time_lo = int(words[6])
  79. self.mem_lo = int(words[9])
  80. self.size_lo = int(words[14])
  81. self.size_wasm = int(words[12])
  82. def __str__(self):
  83. return "%s: time %d %d mem %s %s %s size %s %s %s name %s" % (
  84. Name(self.index), self.time_lo, self.time_tf,
  85. Size(self.mem_lo), Size(self.mem_tf_max), Size(self.mem_tf_total),
  86. Size(self.size_wasm), Size(self.size_lo), Size(self.size_tf), self.name
  87. )
  88. funcs_dict = {}
  89. funcs_list = []
  90. if len(sys.argv) < 2 or sys.argv[1] in ("-h", "--help", "help"):
  91. print("Pass output file (generated with --trace-wasm-compilation-times) as "
  92. "argument")
  93. sys.exit(1)
  94. with open(sys.argv[1], "r") as f:
  95. for line in f.readlines():
  96. words = line.strip().split(" ")
  97. if words[0] != "Compiled" or words[1] != "function":
  98. continue
  99. name = words[2]
  100. RegisterName(name)
  101. if name in funcs_dict:
  102. func = funcs_dict[name]
  103. else:
  104. func = Function(name)
  105. funcs_dict[name] = func
  106. funcs_list.append(func)
  107. func.AddLine(words)
  108. funcs_list.sort(key=lambda fun: fun.time_tf)
  109. for f in funcs_list:
  110. print(f)
  111. if f.time_tf > 0: total_tf_time += f.time_tf
  112. if f.size_tf > 0: total_tf_size += f.size_tf
  113. print("Total TF time: %d" % total_tf_time)
  114. print("Total TF size: %d" % total_tf_size)