llvm/test/Other/opt-bisect-helper.py

   1 #!/usr/bin/env python
   2
   3 from __future__ import print_function
   4
   5 import os
   6 import sys
   7 import argparse
   8 import subprocess
   9
  10 parser = argparse.ArgumentParser()
  11
  12 parser.add_argument("--start", type=int, default=0)
  13 parser.add_argument("--end", type=int, default=(1 << 32))
  14 parser.add_argument("--optcmd", default=("opt"))
  15 parser.add_argument("--filecheckcmd", default=("FileCheck"))
  16 parser.add_argument("--prefix", default=("CHECK-BISECT"))
  17 parser.add_argument("--test", default=(""))
  18
  19 args = parser.parse_args()
  20
  21 start = args.start
  22 end = args.end
  23
  24 opt_command = [args.optcmd, "-O2", "-opt-bisect-limit=%(count)s", "-S", args.test]
  25 check_command = [args.filecheckcmd, args.test, "--check-prefix=%s" % args.prefix]
  26 last = None
  27 while start != end and start != end - 1:
  28     count = int(round(start + (end - start) / 2))
  29     cmd = [x % {"count": count} for x in opt_command]
  30     print("opt: " + str(cmd))
  31     opt_result = subprocess.Popen(cmd, stdout=subprocess.PIPE, stderr=subprocess.PIPE)
  32     filecheck_result = subprocess.Popen(check_command, stdin=opt_result.stdout)
  33     opt_result.stdout.close()
  34     opt_result.stderr.close()
  35     filecheck_result.wait()
  36     if filecheck_result.returncode == 0:
  37         start = count
  38     else:
  39         end = count
  40
  41 print("Last good count: %d" % start)