#! /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
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()
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):
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)