fs.py revision 3183:bd8f3870620f 
1# Copyright (c) 2006 The Regents of The University of Michigan
2# All rights reserved.
3#
4# Redistribution and use in source and binary forms, with or without
5# modification, are permitted provided that the following conditions are
6# met: redistributions of source code must retain the above copyright
7# notice, this list of conditions and the following disclaimer;
8# redistributions in binary form must reproduce the above copyright
9# notice, this list of conditions and the following disclaimer in the
10# documentation and/or other materials provided with the distribution;
11# neither the name of the copyright holders nor the names of its
12# contributors may be used to endorse or promote products derived from
13# this software without specific prior written permission.
14#
15# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
16# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
17# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
18# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
19# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
20# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
21# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
25# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26#
27# Authors: Ali Saidi
28
29import optparse, os, sys
30
31import m5
32from m5.objects import *
33m5.AddToPath('../common')
34from FSConfig import *
35from SysPaths import *
36from Benchmarks import *
37
38if not m5.build_env['FULL_SYSTEM']:
39    m5.panic("This script requires full-system mode (ALPHA_FS).")
40
41parser = optparse.OptionParser()
42
43parser.add_option("-d", "--detailed", action="store_true")
44parser.add_option("-t", "--timing", action="store_true")
45parser.add_option("-m", "--maxtick", type="int")
46parser.add_option("--maxtime", type="float")
47parser.add_option("--dual", action="store_true",
48                  help="Simulate two systems attached with an ethernet link")
49parser.add_option("-b", "--benchmark", action="store", type="string",
50                  dest="benchmark",
51                  help="Specify the benchmark to run. Available benchmarks: %s"\
52                  % DefinedBenchmarks)
53parser.add_option("--etherdump", action="store", type="string", dest="etherdump",
54                  help="Specify the filename to dump a pcap capture of the" \
55                  "ethernet traffic")
56parser.add_option("--checkpoint_dir", action="store", type="string",
57                  help="Place all checkpoints in this absolute directory")
58parser.add_option("-c", "--checkpoint", action="store", type="int",
59                  help="restore from checkpoint <N>")
60
61(options, args) = parser.parse_args()
62
63if args:
64    print "Error: script doesn't take any positional arguments"
65    sys.exit(1)
66
67if options.detailed:
68    cpu = DetailedO3CPU()
69    cpu2 = DetailedO3CPU()
70    mem_mode = 'timing'
71elif options.timing:
72    cpu = TimingSimpleCPU()
73    cpu2 = TimingSimpleCPU()
74    mem_mode = 'timing'
75else:
76    cpu = AtomicSimpleCPU()
77    cpu2 = AtomicSimpleCPU()
78    mem_mode = 'atomic'
79
80cpu.clock = '2GHz'
81cpu2.clock = '2GHz'
82cpu.cpu_id = 0
83cpu2.cpu_id = 0
84
85if options.benchmark:
86    if options.benchmark not in Benchmarks:
87        print "Error benchmark %s has not been defined." % options.benchmark
88        print "Valid benchmarks are: %s" % DefinedBenchmarks
89        sys.exit(1)
90
91    bm = Benchmarks[options.benchmark]
92else:
93    if options.dual:
94        bm = [Machine(), Machine()]
95    else:
96        bm = [Machine()]
97
98if len(bm) == 2:
99    s1 = makeLinuxAlphaSystem(mem_mode, bm[0])
100    s1.cpu = cpu
101    cpu.connectMemPorts(s1.membus)
102    cpu.mem = s1.physmem
103    s2 = makeLinuxAlphaSystem(mem_mode, bm[1])
104    s2.cpu = cpu2
105    cpu2.connectMemPorts(s2.membus)
106    cpu2.mem = s2.physmem
107    root = makeDualRoot(s1, s2, options.etherdump)
108elif len(bm) == 1:
109    root = Root(clock = '1THz',
110                system = makeLinuxAlphaSystem(mem_mode, bm[0]))
111    root.system.cpu = cpu
112    cpu.connectMemPorts(root.system.membus)
113    cpu.mem = root.system.physmem
114else:
115    print "Error I don't know how to create more than 2 systems."
116    sys.exit(1)
117
118m5.instantiate(root)
119
120if options.checkpoint:
121    from os.path import isdir
122    from os import listdir, getcwd
123    import re
124    if options.checkpoint_dir:
125        cptdir = options.checkpoint_dir
126    else:
127        cptdir = getcwd()
128
129    if not isdir(cptdir):
130        m5.panic("checkpoint dir %s does not exist!" % cptdir)
131
132    dirs = listdir(cptdir)
133    expr = re.compile('cpt.([0-9]*)')
134    cpts = []
135    for dir in dirs:
136        match = expr.match(dir)
137        if match:
138            cpts.append(match.group(1))
139
140    if options.checkpoint > len(cpts):
141        m5.panic('Checkpoint %d not found' % options.checkpoint)
142
143    m5.restoreCheckpoint(root, "/".join([cptdir, "cpt.%s" % cpts[options.checkpoint - 1]]))
144
145if options.maxtick:
146    maxtick = options.maxtick
147elif options.maxtime:
148    simtime = int(options.maxtime * root.clock.value)
149    print "simulating for: ", simtime
150    maxtick = simtime
151else:
152    maxtick = -1
153
154exit_event = m5.simulate(maxtick)
155
156while exit_event.getCause() == "checkpoint":
157    if options.checkpoint_dir:
158        m5.checkpoint(root, "/".join([options.checkpoint_dir, "cpt.%d"]))
159    else:
160        m5.checkpoint(root, "cpt.%d")
161
162    if maxtick == -1:
163        exit_event = m5.simulate(maxtick)
164    else:
165        exit_event = m5.simulate(maxtick - m5.curTick())
166
167print 'Exiting @ cycle', m5.curTick(), 'because', exit_event.getCause()
168