fs.py revision 3304
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") 452917SN/Aparser.add_option("-m", "--maxtick", type="int") 462948SN/Aparser.add_option("--maxtime", type="float") 472995SN/Aparser.add_option("--dual", action="store_true", 482995SN/A help="Simulate two systems attached with an ethernet link") 492995SN/Aparser.add_option("-b", "--benchmark", action="store", type="string", 502995SN/A dest="benchmark", 512995SN/A help="Specify the benchmark to run. Available benchmarks: %s"\ 523143Shsul@eecs.umich.edu % DefinedBenchmarks) 533025Ssaidi@eecs.umich.eduparser.add_option("--etherdump", action="store", type="string", dest="etherdump", 543143Shsul@eecs.umich.edu help="Specify the filename to dump a pcap capture of the" \ 553143Shsul@eecs.umich.edu "ethernet traffic") 563143Shsul@eecs.umich.eduparser.add_option("--checkpoint_dir", action="store", type="string", 573143Shsul@eecs.umich.edu help="Place all checkpoints in this absolute directory") 583183Shsul@eecs.umich.eduparser.add_option("-c", "--checkpoint", action="store", type="int", 593183Shsul@eecs.umich.edu help="restore from checkpoint <N>") 602710SN/A 612710SN/A(options, args) = parser.parse_args() 622710SN/A 632710SN/Aif args: 642710SN/A print "Error: script doesn't take any positional arguments" 652710SN/A sys.exit(1) 662710SN/A 673304Sstever@eecs.umich.edu# client system CPU is always simple... note this is an assignment of 683304Sstever@eecs.umich.edu# a class, not an instance. 693304Sstever@eecs.umich.eduClientCPUClass = AtomicSimpleCPU 703304Sstever@eecs.umich.educlient_mem_mode = 'atomic' 713304Sstever@eecs.umich.edu 722934SN/Aif options.detailed: 733304Sstever@eecs.umich.edu ServerCPUClass = DerivO3CPU 743304Sstever@eecs.umich.edu server_mem_mode = 'timing' 752934SN/Aelif options.timing: 763304Sstever@eecs.umich.edu ServerCPUClass = TimingSimpleCPU 773304Sstever@eecs.umich.edu server_mem_mode = 'timing' 782934SN/Aelse: 793304Sstever@eecs.umich.edu ServerCPUClass = AtomicSimpleCPU 803304Sstever@eecs.umich.edu server_mem_mode = 'atomic' 812566SN/A 823304Sstever@eecs.umich.eduServerCPUClass.clock = '2GHz' 833304Sstever@eecs.umich.eduClientCPUClass.clock = '2GHz' 842995SN/A 852995SN/Aif options.benchmark: 863304Sstever@eecs.umich.edu try: 873304Sstever@eecs.umich.edu bm = Benchmarks[options.benchmark] 883304Sstever@eecs.umich.edu except KeyError: 892995SN/A print "Error benchmark %s has not been defined." % options.benchmark 902995SN/A print "Valid benchmarks are: %s" % DefinedBenchmarks 912995SN/A sys.exit(1) 922917SN/Aelse: 932995SN/A if options.dual: 943304Sstever@eecs.umich.edu bm = [SysConfig(), SysConfig()] 952995SN/A else: 963304Sstever@eecs.umich.edu bm = [SysConfig()] 973304Sstever@eecs.umich.edu 983304Sstever@eecs.umich.eduserver_sys = makeLinuxAlphaSystem(server_mem_mode, bm[0]) 993304Sstever@eecs.umich.eduserver_sys.cpu = ServerCPUClass(cpu_id=0) 1003304Sstever@eecs.umich.eduserver_sys.cpu.connectMemPorts(server_sys.membus) 1013304Sstever@eecs.umich.eduserver_sys.cpu.mem = server_sys.physmem 1023005Sstever@eecs.umich.edu 1033005Sstever@eecs.umich.eduif len(bm) == 2: 1043304Sstever@eecs.umich.edu client_sys = makeLinuxAlphaSystem(client_mem_mode, bm[1]) 1053304Sstever@eecs.umich.edu client_sys.cpu = ClientCPUClass(cpu_id=0) 1063304Sstever@eecs.umich.edu client_sys.cpu.connectMemPorts(client_sys.membus) 1073304Sstever@eecs.umich.edu client_sys.cpu.mem = client_sys.physmem 1083304Sstever@eecs.umich.edu root = makeDualRoot(server_sys, client_sys, options.etherdump) 1093005Sstever@eecs.umich.eduelif len(bm) == 1: 1103304Sstever@eecs.umich.edu root = Root(clock = '1THz', system = server_sys) 1113005Sstever@eecs.umich.eduelse: 1123005Sstever@eecs.umich.edu print "Error I don't know how to create more than 2 systems." 1133005Sstever@eecs.umich.edu sys.exit(1) 1142566SN/A 1152710SN/Am5.instantiate(root) 1162710SN/A 1173183Shsul@eecs.umich.eduif options.checkpoint: 1183183Shsul@eecs.umich.edu from os.path import isdir 1193183Shsul@eecs.umich.edu from os import listdir, getcwd 1203183Shsul@eecs.umich.edu import re 1213183Shsul@eecs.umich.edu if options.checkpoint_dir: 1223183Shsul@eecs.umich.edu cptdir = options.checkpoint_dir 1233183Shsul@eecs.umich.edu else: 1243183Shsul@eecs.umich.edu cptdir = getcwd() 1253183Shsul@eecs.umich.edu 1263183Shsul@eecs.umich.edu if not isdir(cptdir): 1273183Shsul@eecs.umich.edu m5.panic("checkpoint dir %s does not exist!" % cptdir) 1283183Shsul@eecs.umich.edu 1293183Shsul@eecs.umich.edu dirs = listdir(cptdir) 1303183Shsul@eecs.umich.edu expr = re.compile('cpt.([0-9]*)') 1313183Shsul@eecs.umich.edu cpts = [] 1323183Shsul@eecs.umich.edu for dir in dirs: 1333183Shsul@eecs.umich.edu match = expr.match(dir) 1343183Shsul@eecs.umich.edu if match: 1353183Shsul@eecs.umich.edu cpts.append(match.group(1)) 1363183Shsul@eecs.umich.edu 1373183Shsul@eecs.umich.edu if options.checkpoint > len(cpts): 1383183Shsul@eecs.umich.edu m5.panic('Checkpoint %d not found' % options.checkpoint) 1393183Shsul@eecs.umich.edu 1403183Shsul@eecs.umich.edu m5.restoreCheckpoint(root, "/".join([cptdir, "cpt.%s" % cpts[options.checkpoint - 1]])) 1413183Shsul@eecs.umich.edu 1422917SN/Aif options.maxtick: 1433046Sstever@eecs.umich.edu maxtick = options.maxtick 1442948SN/Aelif options.maxtime: 1452948SN/A simtime = int(options.maxtime * root.clock.value) 1462948SN/A print "simulating for: ", simtime 1473046Sstever@eecs.umich.edu maxtick = simtime 1482917SN/Aelse: 1493046Sstever@eecs.umich.edu maxtick = -1 1503022Shsul@eecs.umich.edu 1513046Sstever@eecs.umich.eduexit_event = m5.simulate(maxtick) 1523022Shsul@eecs.umich.edu 1533022Shsul@eecs.umich.eduwhile exit_event.getCause() == "checkpoint": 1543143Shsul@eecs.umich.edu if options.checkpoint_dir: 1553143Shsul@eecs.umich.edu m5.checkpoint(root, "/".join([options.checkpoint_dir, "cpt.%d"])) 1563143Shsul@eecs.umich.edu else: 1573143Shsul@eecs.umich.edu m5.checkpoint(root, "cpt.%d") 1583143Shsul@eecs.umich.edu 1593133Shsul@eecs.umich.edu if maxtick == -1: 1603133Shsul@eecs.umich.edu exit_event = m5.simulate(maxtick) 1613133Shsul@eecs.umich.edu else: 1623133Shsul@eecs.umich.edu exit_event = m5.simulate(maxtick - m5.curTick()) 1632710SN/A 1642740SN/Aprint 'Exiting @ cycle', m5.curTick(), 'because', exit_event.getCause() 165