tests/kernbench/kernbench.py

import test, pickle
from autotest_utils import *

class kernbench(test.test):
	version = 1

	def setup(self):
		# http://kernel.org/pub/linux/kernel/v2.6/linux-2.6.14.tar.bz2
		if (os.path.exists(self.bindir + '/linux-2.6.14.tar.bz2')):
			tarball = self.bindir + '/linux-2.6.14.tar.bz2'
		else:
			tarball = '/usr/local/src/linux-2.6.14.tar.bz2'
		print tarball
		kernel = self.job.kernel(self.srcdir, tarball)
		kernel.config('')
		# have to save this off, as we might use it in another run
		kernel.pickle_dump(self.srcdir + '/.pickle')


	def execute(self, iterations = 1, threads = 2 * count_cpus()):
		kernel = pickle.load(open(self.srcdir + '/.pickle', 'r'))
		print "kernbench x %d: %d threads" % (iterations, threads)

		kernel.build_timed(threads)         # warmup run
		for i in range(1, iterations+1):
			logfile = self.resultsdir+'/time.%d' % i
			kernel.build_timed(threads, logfile)

		# Do a profiling run if necessary
		profilers = self.job.profilers
		if profilers.present():
			profilers.start(self)
			logfile = self.resultsdir+'/time.profile'
			kernel.build_timed(threads, logfile)
			profilers.stop(self)
			profilers.report(self)

		kernel.clean()		# Don't leave litter lying around
		os.chdir(self.resultsdir)
		system("grep elapsed time.* > time")