Inorder: Fix compilation of m5.fast.
[gem5.git] / util / regress
index 5d6f321f250c1bb4427ad188dc5e8c4cfd86b010..a74bd09c3bedad28fce32592177bf8fe11672a3f 100755 (executable)
@@ -1,5 +1,5 @@
 #! /usr/bin/env python
-# Copyright (c) 2005-2006 The Regents of The University of Michigan
+# Copyright (c) 2005-2007 The Regents of The University of Michigan
 # All rights reserved.
 #
 # Redistribution and use in source and binary forms, with or without
@@ -31,25 +31,34 @@ import sys
 import os
 import optparse
 import datetime
+from subprocess import call
 
 progname = os.path.basename(sys.argv[0])
 
 optparser = optparse.OptionParser()
-optparser.add_option('-v', '--verbose', dest='verbose', action='store_true',
-                     default=False,
-                     help='echo commands before executing')
-optparser.add_option('--builds', dest='builds',
-                     default='ALPHA_SE,ALPHA_FS,MIPS_SE,SPARC_SE',
-                     help='comma-separated list of build targets to test  '
-                     " (default: '%default')" )
-optparser.add_option('--variants', dest='variants',
-                     default='fast',
-                     help='comma-separated list of build variants to test '
-                     " (default: '%default')" )
-optparser.add_option('--scons-opts', dest='scons_opts', default='',
-                     help='scons options', metavar='OPTS')
-optparser.add_option('-j', '--jobs', type='int', default=1,
-                     help='number of parallel jobs to use')
+add_option = optparser.add_option
+add_option('-v', '--verbose', dest='verbose', action='store_true',
+           default=False,
+           help='echo commands before executing')
+add_option('--builds', dest='builds',
+           default='ALPHA_SE,ALPHA_SE_MOESI_hammer,' \
+           'ALPHA_SE_MESI_CMP_directory,'  \
+           'ALPHA_SE_MOESI_CMP_directory,' \
+           'ALPHA_SE_MOESI_CMP_token,' \
+           'ALPHA_FS,MIPS_SE,POWER_SE,SPARC_SE,SPARC_FS,X86_SE,ARM_SE',
+           help="comma-separated build targets to test (default: '%default')")
+add_option('--variants', dest='variants', default='fast',
+           help="comma-separated build variants to test (default: '%default')")
+add_option('--scons-opts', dest='scons_opts', default='', metavar='OPTS',
+           help='scons options')
+add_option('-j', '--jobs', type='int', default=1,
+           help='number of parallel jobs to use')
+add_option('-k', '--keep-going', action='store_true',
+           help='keep going after errors')
+add_option('-D', '--build-dir', default='',
+           help='build directory location')
+add_option('-n', "--no-exec", default=False, action='store_true',
+           help="don't actually invoke scons, just echo SCons command line")
 
 (options, tests) = optparser.parse_args()
 
@@ -58,16 +67,24 @@ optparser.add_option('-j', '--jobs', type='int', default=1,
 builds = options.builds.split(',')
 variants = options.variants.split(',')
 
+options.build_dir = os.path.join(options.build_dir, 'build')
+
 # Call os.system() and raise exception if return status is non-zero
 def system(cmd):
-    if options.verbose:
-        print cmd
-    status = os.system(cmd)
-    if status != 0:
-        upper = (status & 0xff00) >> 8
-        lower = (status & 0xff)
-        raise OSError, "shell command '%s' failed, status %d:%d" \
-              % (cmd, upper, lower)
+    try:
+        retcode = call(cmd, shell=True)
+        if retcode < 0:
+            print >>sys.stderr, "Child was terminated by signal", -retcode
+            print >>sys.stderr, "When attemping to execute: %s" % cmd
+            sys.exit(1)
+        elif retcode > 0:
+            print >>sys.stderr, "Child returned", retcode
+            print >>sys.stderr, "When attemping to execute: %s" % cmd
+            sys.exit(1)
+    except OSError, e:
+        print >>sys.stderr, "Execution failed:", e
+        print >>sys.stderr, "When attemping to execute: %s" % cmd
+        sys.exit(1)
 
 # Quote string s so it can be passed as a shell arg
 def shellquote(s):
@@ -75,30 +92,50 @@ def shellquote(s):
         s = "'%s'" % s
     return s
 
-try:
-    if not tests:
-        print "No tests specified, just building binaries."
-        targets = ['build/%s/m5.%s' % (build, variant)
-                  for build in builds
-                  for variant in variants]
-    elif 'all' in tests:
-       targets = ['build/%s/tests/%s' % (build, variant)
-                  for build in builds
-                  for variant in variants]
+if not tests:
+    print "No tests specified, just building binaries."
+    targets = ['%s/%s/m5.%s' % (options.build_dir, build, variant)
+               for build in builds
+               for variant in variants]
+elif 'all' in tests:
+    targets = ['%s/%s/tests/%s' % (options.build_dir, build, variant)
+               for build in builds
+               for variant in variants]
+else:
+    # Ugly! Since we don't have any quick SPARC_FS tests remove the SPARC_FS target
+    # If we ever get a quick SPARC_FS test, this code should be removed
+    if 'quick' in tests and 'SPARC_FS' in builds:
+        builds.remove('SPARC_FS')
+    targets = ['%s/%s/tests/%s/%s' % (options.build_dir, build, variant, test)
+               for build in builds
+               for variant in variants
+               for test in tests]
+
+def cpu_count():
+    if 'bsd' in sys.platform or sys.platform == 'darwin':
+        try:
+            return int(os.popen('sysctl -n hw.ncpu').read())
+        except ValueError:
+            pass
     else:
-       targets = ['build/%s/tests/%s/%s' % (build, variant, test)
-                  for build in builds
-                  for variant in variants
-                  for test in tests]
+        try:
+            return os.sysconf('SC_NPROCESSORS_ONLN')
+        except (ValueError, OSError, AttributeError):
+            pass
 
-    scons_opts = options.scons_opts
-    if options.jobs != 1:
-        scons_opts += ' -j %d' % options.jobs
+    raise NotImplementedError('cannot determine number of cpus')
 
-    system('scons %s %s' % (scons_opts, ' '.join(targets)))
+scons_opts = options.scons_opts
+if options.jobs != 1:
+    if options.jobs == 0:
+        options.jobs = cpu_count()
+    scons_opts += ' -j %d' % options.jobs
+if options.keep_going:
+    scons_opts += ' -k'
 
+cmd = 'scons IGNORE_STYLE=True %s %s' % (scons_opts, ' '.join(targets))
+if options.no_exec:
+    print cmd
+else:
+    system(cmd)
     sys.exit(0)
-
-except OSError, exc:
-    print "%s: " % progname, exc
-    sys.exit(1)