contrib/analyze_brprob_spec.py

   1 #!/usr/bin/env python3
   2
   3 # This file is part of GCC.
   4 #
   5 # GCC is free software; you can redistribute it and/or modify it under
   6 # the terms of the GNU General Public License as published by the Free
   7 # Software Foundation; either version 3, or (at your option) any later
   8 # version.
   9 #
  10 # GCC is distributed in the hope that it will be useful, but WITHOUT ANY
  11 # WARRANTY; without even the implied warranty of MERCHANTABILITY or
  12 # FITNESS FOR A PARTICULAR PURPOSE.  See the GNU General Public License
  13 # for more details.
  14 #
  15 # You should have received a copy of the GNU General Public License
  16 # along with GCC; see the file COPYING3.  If not see
  17 # <http://www.gnu.org/licenses/>.  */
  18
  19 import sys
  20 import os
  21 import subprocess
  22 import tempfile
  23 import argparse
  24
  25 script_location = os.path.realpath(__file__)
  26
  27 parser = argparse.ArgumentParser()
  28 parser.add_argument('location', metavar = 'dump_file',
  29     help = 'Location with SPEC benchmarks')
  30 parser.add_argument('-s', '--sorting', dest = 'sorting',
  31     choices = ['branches', 'branch-hitrate', 'hitrate', 'coverage', 'name'],
  32     default = 'branches')
  33
  34 args = parser.parse_args()
  35
  36 benchmarks = os.listdir(args.location)
  37
  38 for b in sorted(benchmarks):
  39     dumps = []
  40     for root, dirs, files in os.walk(os.path.join(args.location, b)):
  41         for x in files:
  42             if x.endswith('.profile'):
  43                 dumps.append(os.path.join(root, x))
  44
  45     if len(dumps) == 0:
  46         continue
  47
  48     temp = tempfile.NamedTemporaryFile(delete = False)
  49     for d in dumps:
  50         temp.write(open(d, 'rb').read())
  51
  52     temp.close()
  53
  54     print()
  55     print(b)
  56     sys.stdout.flush()
  57     p = [os.path.join(os.path.dirname(script_location), 'analyze_brprob.py'),
  58         temp.name, '--sorting', args.sorting]
  59     p = subprocess.check_call(p)
  60     sys.stdout.flush()
  61
  62     os.remove(temp.name)