fs.py revision 3398:7f14cadf3afd
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
43# Benchmark options
44parser.add_option("--dual", action="store_true",
45                  help="Simulate two systems attached with an ethernet link")
46parser.add_option("-b", "--benchmark", action="store", type="string",
47                  dest="benchmark",
48                  help="Specify the benchmark to run. Available benchmarks: %s"\
49                  % DefinedBenchmarks)
50
51# system options
52parser.add_option("-d", "--detailed", action="store_true")
53parser.add_option("-t", "--timing", action="store_true")
54parser.add_option("-n", "--num_cpus", type="int", default=1)
55parser.add_option("--caches", action="store_true")
56
57# Run duration options
58parser.add_option("-m", "--maxtick", type="int")
59parser.add_option("--maxtime", type="float")
60
61# Metafile options
62parser.add_option("--etherdump", action="store", type="string", dest="etherdump",
63                  help="Specify the filename to dump a pcap capture of the" \
64                  "ethernet traffic")
65
66# Checkpointing options
67###Note that performing checkpointing via python script files will override
68###checkpoint instructions built into binaries.
69parser.add_option("--take_checkpoints", action="store", type="string",
70                  help="<M,N> will take checkpoint at cycle M and every N cycles \
71                  thereafter")
72parser.add_option("--max_checkpoints", action="store", type="int",
73                  help="the maximum number of checkpoints to drop",
74                  default=5)
75parser.add_option("--checkpoint_dir", action="store", type="string",
76                  help="Place all checkpoints in this absolute directory")
77parser.add_option("-r", "--checkpoint_restore", action="store", type="int",
78                  help="restore from checkpoint <N>")
79
80# CPU Switching - default switch model goes from a checkpoint
81# to a timing simple CPU with caches to warm up, then to detailed CPU for
82# data measurement
83parser.add_option("-s", "--standard_switch", action="store_true",
84                  help="switch from one cpu mode to another")
85
86(options, args) = parser.parse_args()
87
88if args:
89    print "Error: script doesn't take any positional arguments"
90    sys.exit(1)
91
92class MyCache(BaseCache):
93    assoc = 2
94    block_size = 64
95    latency = 1
96    mshrs = 10
97    tgts_per_mshr = 5
98    protocol = CoherenceProtocol(protocol='moesi')
99
100# driver system CPU is always simple... note this is an assignment of
101# a class, not an instance.
102DriveCPUClass = AtomicSimpleCPU
103drive_mem_mode = 'atomic'
104
105# system under test can be any of these CPUs
106if options.detailed:
107    TestCPUClass = DerivO3CPU
108    test_mem_mode = 'timing'
109elif options.timing:
110    TestCPUClass = TimingSimpleCPU
111    test_mem_mode = 'timing'
112else:
113    TestCPUClass = AtomicSimpleCPU
114    test_mem_mode = 'atomic'
115
116TestCPUClass.clock = '2GHz'
117DriveCPUClass.clock = '2GHz'
118
119if options.benchmark:
120    try:
121        bm = Benchmarks[options.benchmark]
122    except KeyError:
123        print "Error benchmark %s has not been defined." % options.benchmark
124        print "Valid benchmarks are: %s" % DefinedBenchmarks
125        sys.exit(1)
126else:
127    if options.dual:
128        bm = [SysConfig(), SysConfig()]
129    else:
130        bm = [SysConfig()]
131
132test_sys = makeLinuxAlphaSystem(test_mem_mode, bm[0])
133np = options.num_cpus
134test_sys.cpu = [TestCPUClass(cpu_id=i) for i in xrange(np)]
135for i in xrange(np):
136    if options.caches and not options.standard_switch:
137        test_sys.cpu[i].addPrivateSplitL1Caches(MyCache(size = '32kB'),
138                                                  MyCache(size = '64kB'))
139    test_sys.cpu[i].connectMemPorts(test_sys.membus)
140    test_sys.cpu[i].mem = test_sys.physmem
141
142if len(bm) == 2:
143    drive_sys = makeLinuxAlphaSystem(drive_mem_mode, bm[1])
144    drive_sys.cpu = DriveCPUClass(cpu_id=0)
145    drive_sys.cpu.connectMemPorts(drive_sys.membus)
146    drive_sys.cpu.mem = drive_sys.physmem
147    root = makeDualRoot(test_sys, drive_sys, options.etherdump)
148elif len(bm) == 1:
149    root = Root(clock = '1THz', system = test_sys)
150else:
151    print "Error I don't know how to create more than 2 systems."
152    sys.exit(1)
153
154if options.standard_switch:
155    switch_cpus = [TimingSimpleCPU(defer_registration=True, cpu_id=(np+i)) for i in xrange(np)]
156    switch_cpus1 = [DerivO3CPU(defer_registration=True, cpu_id=(2*np+i)) for i in xrange(np)]
157    for i in xrange(np):
158        switch_cpus[i].system =  test_sys
159        switch_cpus1[i].system =  test_sys
160        switch_cpus[i].clock = TestCPUClass.clock
161        switch_cpus1[i].clock = TestCPUClass.clock
162        if options.caches:
163            switch_cpus[i].addPrivateSplitL1Caches(MyCache(size = '32kB'),
164                                                    MyCache(size = '64kB'))
165
166        switch_cpus[i].mem = test_sys.physmem
167        switch_cpus1[i].mem = test_sys.physmem
168        switch_cpus[i].connectMemPorts(test_sys.membus)
169        root.switch_cpus = switch_cpus
170        root.switch_cpus1 = switch_cpus1
171        switch_cpu_list = [(test_sys.cpu[i], switch_cpus[i]) for i in xrange(np)]
172        switch_cpu_list1 = [(switch_cpus[i], switch_cpus1[i]) for i in xrange(np)]
173
174m5.instantiate(root)
175
176if options.checkpoint_dir:
177    cptdir = options.checkpoint_dir
178else:
179    cptdir = os.getcwd()
180
181if options.checkpoint_restore:
182    from os.path import isdir
183    from os import listdir, getcwd
184    import re
185
186    if not isdir(cptdir):
187        m5.panic("checkpoint dir %s does not exist!" % cptdir)
188
189    dirs = listdir(cptdir)
190    expr = re.compile('cpt.([0-9]*)')
191    cpts = []
192    for dir in dirs:
193        match = expr.match(dir)
194        if match:
195            cpts.append(match.group(1))
196
197    cpts.sort(lambda a,b: cmp(long(a), long(b)))
198
199    if options.checkpoint_restore > len(cpts):
200        m5.panic('Checkpoint %d not found' % options.checkpoint_restore)
201
202    m5.restoreCheckpoint(root, "/".join([cptdir, "cpt.%s" % cpts[options.checkpoint_restore - 1]]))
203
204if options.standard_switch:
205    exit_event = m5.simulate(1000)
206    ## when you change to Timing (or Atomic), you halt the system given
207    ## as argument.  When you are finished with the system changes
208    ## (including switchCpus), you must resume the system manually.
209    ## You DON'T need to resume after just switching CPUs if you haven't
210    ## changed anything on the system level.
211    m5.changeToTiming(test_sys)
212    m5.switchCpus(switch_cpu_list)
213    m5.resume(test_sys)
214
215    exit_event = m5.simulate(500000000000)
216    m5.switchCpus(switch_cpu_list1)
217
218if options.maxtick:
219    maxtick = options.maxtick
220elif options.maxtime:
221    simtime = int(options.maxtime * root.clock.value)
222    print "simulating for: ", simtime
223    maxtick = simtime
224else:
225    maxtick = -1
226
227num_checkpoints = 0
228
229exit_cause = ''
230
231if options.take_checkpoints:
232    [when, period] = options.take_checkpoints.split(",", 1)
233    when = int(when)
234    period = int(period)
235
236    exit_event = m5.simulate(when)
237    while exit_event.getCause() == "checkpoint":
238        exit_event = m5.simulate(when - m5.curTick())
239
240    if exit_event.getCause() == "simulate() limit reached":
241        m5.checkpoint(root, cptdir + "cpt.%d")
242        num_checkpoints += 1
243
244    sim_ticks = when
245    exit_cause = "maximum %d checkpoints dropped" % options.max_checkpoints
246    while num_checkpoints < options.max_checkpoints:
247        if (sim_ticks + period) > maxtick and maxtick != -1:
248            exit_event = m5.simulate(maxtick - sim_ticks)
249            exit_cause = exit_event.getCause()
250            break
251        else:
252            exit_event = m5.simulate(period)
253            sim_ticks += period
254            while exit_event.getCause() == "checkpoint":
255                exit_event = m5.simulate(period - m5.curTick())
256            if exit_event.getCause() == "simulate() limit reached":
257                m5.checkpoint(root, cptdir + "cpt.%d")
258                num_checkpoints += 1
259
260else: #no checkpoints being taken via this script
261    exit_event = m5.simulate(maxtick)
262
263    while exit_event.getCause() == "checkpoint":
264        m5.checkpoint(root, cptdir + "cpt.%d")
265        num_checkpoints += 1
266        if num_checkpoints == options.max_checkpoints:
267            exit_cause =  "maximum %d checkpoints dropped" % options.max_checkpoints
268            break
269
270        if maxtick == -1:
271            exit_event = m5.simulate(maxtick)
272        else:
273            exit_event = m5.simulate(maxtick - m5.curTick())
274
275        exit_cause = exit_event.getCause()
276
277if exit_cause == '':
278    exit_cause = exit_event.getCause()
279print 'Exiting @ cycle', m5.curTick(), 'because ', exit_cause
280