fs.py revision 3312
13005Sstever@eecs.umich.edu# Copyright (c) 2006 The Regents of The University of Michigan 23005Sstever@eecs.umich.edu# All rights reserved. 33005Sstever@eecs.umich.edu# 43005Sstever@eecs.umich.edu# Redistribution and use in source and binary forms, with or without 53005Sstever@eecs.umich.edu# modification, are permitted provided that the following conditions are 63005Sstever@eecs.umich.edu# met: redistributions of source code must retain the above copyright 73005Sstever@eecs.umich.edu# notice, this list of conditions and the following disclaimer; 83005Sstever@eecs.umich.edu# redistributions in binary form must reproduce the above copyright 93005Sstever@eecs.umich.edu# notice, this list of conditions and the following disclaimer in the 103005Sstever@eecs.umich.edu# documentation and/or other materials provided with the distribution; 113005Sstever@eecs.umich.edu# neither the name of the copyright holders nor the names of its 123005Sstever@eecs.umich.edu# contributors may be used to endorse or promote products derived from 133005Sstever@eecs.umich.edu# this software without specific prior written permission. 143005Sstever@eecs.umich.edu# 153005Sstever@eecs.umich.edu# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS 163005Sstever@eecs.umich.edu# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT 173005Sstever@eecs.umich.edu# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR 183005Sstever@eecs.umich.edu# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT 193005Sstever@eecs.umich.edu# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, 203005Sstever@eecs.umich.edu# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT 213005Sstever@eecs.umich.edu# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 223005Sstever@eecs.umich.edu# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 233005Sstever@eecs.umich.edu# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 243005Sstever@eecs.umich.edu# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE 253005Sstever@eecs.umich.edu# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 263005Sstever@eecs.umich.edu# 273005Sstever@eecs.umich.edu# Authors: Ali Saidi 283005Sstever@eecs.umich.edu 292889SN/Aimport optparse, os, sys 302889SN/A 312710SN/Aimport m5 322710SN/Afrom m5.objects import * 332934SN/Am5.AddToPath('../common') 342934SN/Afrom FSConfig import * 352549SN/Afrom SysPaths import * 362995SN/Afrom Benchmarks import * 372549SN/A 383088Sstever@eecs.umich.eduif not m5.build_env['FULL_SYSTEM']: 393088Sstever@eecs.umich.edu m5.panic("This script requires full-system mode (ALPHA_FS).") 403088Sstever@eecs.umich.edu 412889SN/Aparser = optparse.OptionParser() 422710SN/A 432917SN/Aparser.add_option("-d", "--detailed", action="store_true") 442710SN/Aparser.add_option("-t", "--timing", action="store_true") 453312Sstever@eecs.umich.eduparser.add_option("-n", "--num_cpus", type="int", default=1) 462917SN/Aparser.add_option("-m", "--maxtick", type="int") 472948SN/Aparser.add_option("--maxtime", type="float") 482995SN/Aparser.add_option("--dual", action="store_true", 492995SN/A help="Simulate two systems attached with an ethernet link") 502995SN/Aparser.add_option("-b", "--benchmark", action="store", type="string", 512995SN/A dest="benchmark", 522995SN/A help="Specify the benchmark to run. Available benchmarks: %s"\ 533143Shsul@eecs.umich.edu % DefinedBenchmarks) 543025Ssaidi@eecs.umich.eduparser.add_option("--etherdump", action="store", type="string", dest="etherdump", 553143Shsul@eecs.umich.edu help="Specify the filename to dump a pcap capture of the" \ 563143Shsul@eecs.umich.edu "ethernet traffic") 573143Shsul@eecs.umich.eduparser.add_option("--checkpoint_dir", action="store", type="string", 583143Shsul@eecs.umich.edu help="Place all checkpoints in this absolute directory") 593183Shsul@eecs.umich.eduparser.add_option("-c", "--checkpoint", action="store", type="int", 603183Shsul@eecs.umich.edu help="restore from checkpoint <N>") 612710SN/A 622710SN/A(options, args) = parser.parse_args() 632710SN/A 642710SN/Aif args: 652710SN/A print "Error: script doesn't take any positional arguments" 662710SN/A sys.exit(1) 672710SN/A 683304Sstever@eecs.umich.edu# client system CPU is always simple... note this is an assignment of 693304Sstever@eecs.umich.edu# a class, not an instance. 703304Sstever@eecs.umich.eduClientCPUClass = AtomicSimpleCPU 713304Sstever@eecs.umich.educlient_mem_mode = 'atomic' 723304Sstever@eecs.umich.edu 732934SN/Aif options.detailed: 743304Sstever@eecs.umich.edu ServerCPUClass = DerivO3CPU 753304Sstever@eecs.umich.edu server_mem_mode = 'timing' 762934SN/Aelif options.timing: 773304Sstever@eecs.umich.edu ServerCPUClass = TimingSimpleCPU 783304Sstever@eecs.umich.edu server_mem_mode = 'timing' 792934SN/Aelse: 803304Sstever@eecs.umich.edu ServerCPUClass = AtomicSimpleCPU 813304Sstever@eecs.umich.edu server_mem_mode = 'atomic' 822566SN/A 833304Sstever@eecs.umich.eduServerCPUClass.clock = '2GHz' 843304Sstever@eecs.umich.eduClientCPUClass.clock = '2GHz' 852995SN/A 862995SN/Aif options.benchmark: 873304Sstever@eecs.umich.edu try: 883304Sstever@eecs.umich.edu bm = Benchmarks[options.benchmark] 893304Sstever@eecs.umich.edu except KeyError: 902995SN/A print "Error benchmark %s has not been defined." % options.benchmark 912995SN/A print "Valid benchmarks are: %s" % DefinedBenchmarks 922995SN/A sys.exit(1) 932917SN/Aelse: 942995SN/A if options.dual: 953304Sstever@eecs.umich.edu bm = [SysConfig(), SysConfig()] 962995SN/A else: 973304Sstever@eecs.umich.edu bm = [SysConfig()] 983304Sstever@eecs.umich.edu 993304Sstever@eecs.umich.eduserver_sys = makeLinuxAlphaSystem(server_mem_mode, bm[0]) 1003312Sstever@eecs.umich.edunp = options.num_cpus 1013312Sstever@eecs.umich.eduserver_sys.cpu = [ServerCPUClass(cpu_id=i) for i in xrange(np)] 1023312Sstever@eecs.umich.edufor i in xrange(np): 1033312Sstever@eecs.umich.edu server_sys.cpu[i].connectMemPorts(server_sys.membus) 1043312Sstever@eecs.umich.edu server_sys.cpu[i].mem = server_sys.physmem 1053005Sstever@eecs.umich.edu 1063005Sstever@eecs.umich.eduif len(bm) == 2: 1073304Sstever@eecs.umich.edu client_sys = makeLinuxAlphaSystem(client_mem_mode, bm[1]) 1083304Sstever@eecs.umich.edu client_sys.cpu = ClientCPUClass(cpu_id=0) 1093304Sstever@eecs.umich.edu client_sys.cpu.connectMemPorts(client_sys.membus) 1103304Sstever@eecs.umich.edu client_sys.cpu.mem = client_sys.physmem 1113304Sstever@eecs.umich.edu root = makeDualRoot(server_sys, client_sys, options.etherdump) 1123005Sstever@eecs.umich.eduelif len(bm) == 1: 1133304Sstever@eecs.umich.edu root = Root(clock = '1THz', system = server_sys) 1143005Sstever@eecs.umich.eduelse: 1153005Sstever@eecs.umich.edu print "Error I don't know how to create more than 2 systems." 1163005Sstever@eecs.umich.edu sys.exit(1) 1172566SN/A 1182710SN/Am5.instantiate(root) 1192710SN/A 1203183Shsul@eecs.umich.eduif options.checkpoint: 1213183Shsul@eecs.umich.edu from os.path import isdir 1223183Shsul@eecs.umich.edu from os import listdir, getcwd 1233183Shsul@eecs.umich.edu import re 1243183Shsul@eecs.umich.edu if options.checkpoint_dir: 1253183Shsul@eecs.umich.edu cptdir = options.checkpoint_dir 1263183Shsul@eecs.umich.edu else: 1273183Shsul@eecs.umich.edu cptdir = getcwd() 1283183Shsul@eecs.umich.edu 1293183Shsul@eecs.umich.edu if not isdir(cptdir): 1303183Shsul@eecs.umich.edu m5.panic("checkpoint dir %s does not exist!" % cptdir) 1313183Shsul@eecs.umich.edu 1323183Shsul@eecs.umich.edu dirs = listdir(cptdir) 1333183Shsul@eecs.umich.edu expr = re.compile('cpt.([0-9]*)') 1343183Shsul@eecs.umich.edu cpts = [] 1353183Shsul@eecs.umich.edu for dir in dirs: 1363183Shsul@eecs.umich.edu match = expr.match(dir) 1373183Shsul@eecs.umich.edu if match: 1383183Shsul@eecs.umich.edu cpts.append(match.group(1)) 1393183Shsul@eecs.umich.edu 1403183Shsul@eecs.umich.edu if options.checkpoint > len(cpts): 1413183Shsul@eecs.umich.edu m5.panic('Checkpoint %d not found' % options.checkpoint) 1423183Shsul@eecs.umich.edu 1433183Shsul@eecs.umich.edu m5.restoreCheckpoint(root, "/".join([cptdir, "cpt.%s" % cpts[options.checkpoint - 1]])) 1443183Shsul@eecs.umich.edu 1452917SN/Aif options.maxtick: 1463046Sstever@eecs.umich.edu maxtick = options.maxtick 1472948SN/Aelif options.maxtime: 1482948SN/A simtime = int(options.maxtime * root.clock.value) 1492948SN/A print "simulating for: ", simtime 1503046Sstever@eecs.umich.edu maxtick = simtime 1512917SN/Aelse: 1523046Sstever@eecs.umich.edu maxtick = -1 1533022Shsul@eecs.umich.edu 1543046Sstever@eecs.umich.eduexit_event = m5.simulate(maxtick) 1553022Shsul@eecs.umich.edu 1563022Shsul@eecs.umich.eduwhile exit_event.getCause() == "checkpoint": 1573143Shsul@eecs.umich.edu if options.checkpoint_dir: 1583143Shsul@eecs.umich.edu m5.checkpoint(root, "/".join([options.checkpoint_dir, "cpt.%d"])) 1593143Shsul@eecs.umich.edu else: 1603143Shsul@eecs.umich.edu m5.checkpoint(root, "cpt.%d") 1613143Shsul@eecs.umich.edu 1623133Shsul@eecs.umich.edu if maxtick == -1: 1633133Shsul@eecs.umich.edu exit_event = m5.simulate(maxtick) 1643133Shsul@eecs.umich.edu else: 1653133Shsul@eecs.umich.edu exit_event = m5.simulate(maxtick - m5.curTick()) 1662710SN/A 1672740SN/Aprint 'Exiting @ cycle', m5.curTick(), 'because', exit_event.getCause() 168