configs/example/se.py

   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: Steve Reinhardt
  28
  29 # Simple test script
  30 #
  31 # "m5 test.py"
  32
  33 import m5
  34 from m5.objects import *
  35 import os, optparse, sys
  36 m5.AddToPath('../common')
  37
  38 parser = optparse.OptionParser()
  39
  40 # Benchmark options
  41 parser.add_option("-c", "--cmd",
  42                   default="../../tests/test-progs/hello/bin/alpha/linux/hello",
  43                   help="The binary to run in syscall emulation mode.")
  44 parser.add_option("-o", "--options", default="",
  45                   help="The options to pass to the binary, use \" \" around the entire\
  46                         string.")
  47 parser.add_option("-i", "--input", default="",
  48                   help="A file of input to give to the binary.")
  49
  50 # System options
  51 parser.add_option("-d", "--detailed", action="store_true")
  52 parser.add_option("-t", "--timing", action="store_true")
  53 parser.add_option("--caches", action="store_true")
  54
  55 # Run duration options
  56 parser.add_option("-m", "--maxtick", type="int")
  57 parser.add_option("--maxtime", type="float")
  58
  59 #Checkpointing options
  60 ###Note that performing checkpointing via python script files will override
  61 ###checkpoint instructions built into binaries.
  62 parser.add_option("--take_checkpoints", action="store", type="string",
  63                   help="<M,N> will take checkpoint at cycle M and every N cycles \
  64                   thereafter")
  65 parser.add_option("--max_checkpoints", action="store", type="int",
  66                   help="the maximum number of checkpoints to drop",
  67                   default=5)
  68 parser.add_option("--checkpoint_dir", action="store", type="string",
  69                   help="Place all checkpoints in this absolute directory")
  70 parser.add_option("-r", "--checkpoint_restore", action="store", type="int",
  71                   help="restore from checkpoint <N>")
  72
  73 #CPU Switching - default switch model generally goes from a checkpoint
  74 #to a timing simple CPU with caches to warm up, then to detailed CPU for
  75 #data measurement
  76 parser.add_option("-s", "--standard_switch", action="store_true",
  77                   help="switch from one cpu mode to another")
  78
  79 (options, args) = parser.parse_args()
  80
  81 if args:
  82     print "Error: script doesn't take any positional arguments"
  83     sys.exit(1)
  84
  85 class MyCache(BaseCache):
  86     assoc = 2
  87     block_size = 64
  88     latency = 1
  89     mshrs = 10
  90     tgts_per_mshr = 5
  91
  92 process = LiveProcess()
  93 process.executable = options.cmd
  94 process.cmd = options.cmd + " " + options.options
  95 if options.input != "":
  96     process.input = options.input
  97
  98 if options.detailed:
  99     #check for SMT workload
 100     workloads = options.cmd.split(';')
 101     if len(workloads) > 1:
 102         process = []
 103         smt_idx = 0
 104         inputs = []
 105
 106         if options.input != "":
 107             inputs = options.input.split(';')
 108
 109         for wrkld in workloads:
 110             smt_process = LiveProcess()
 111             smt_process.executable = wrkld
 112             smt_process.cmd = wrkld + " " + options.options
 113             if inputs and inputs[smt_idx]:
 114                 smt_process.input = inputs[smt_idx]
 115             process += [smt_process, ]
 116             smt_idx += 1
 117
 118
 119 if options.timing:
 120     cpu = TimingSimpleCPU()
 121 elif options.detailed:
 122     cpu = DerivO3CPU()
 123 else:
 124     cpu = AtomicSimpleCPU()
 125
 126 cpu.workload = process
 127 cpu.cpu_id = 0
 128
 129 system = System(cpu = cpu,
 130                 physmem = PhysicalMemory(range=AddrRange("512MB")),
 131                 membus = Bus())
 132
 133 if options.caches and not options.standard_switch:
 134     system.cpu.addPrivateSplitL1Caches(MyCache(size = '32kB'),
 135                                        MyCache(size = '64kB'))
 136
 137 system.physmem.port = system.membus.port
 138 system.cpu.connectMemPorts(system.membus)
 139 system.cpu.mem = system.physmem
 140 system.cpu.clock = '2GHz'
 141 root = Root(system = system)
 142
 143 if options.timing or options.detailed:
 144     root.system.mem_mode = 'timing'
 145
 146 if options.standard_switch:
 147     switch_cpu = TimingSimpleCPU(defer_registration=True, cpu_id=1)
 148     switch_cpu1 = DerivO3CPU(defer_registration=True, cpu_id=2)
 149     switch_cpu.system =  system
 150     switch_cpu1.system =  system
 151     switch_cpu.clock = cpu.clock
 152     switch_cpu1.clock = cpu.clock
 153     if options.caches:
 154         switch_cpu.addPrivateSplitL1Caches(MyCache(size = '32kB'),
 155                                            MyCache(size = '64kB'))
 156
 157     switch_cpu.workload = process
 158     switch_cpu1.workload = process
 159     switch_cpu.mem = system.physmem
 160     switch_cpu1.mem = system.physmem
 161     switch_cpu.connectMemPorts(system.membus)
 162     root.switch_cpu = switch_cpu
 163     root.switch_cpu1 = switch_cpu1
 164     switch_cpu_list = [(system.cpu, switch_cpu)]
 165     switch_cpu_list1 = [(switch_cpu, switch_cpu1)]
 166
 167 # instantiate configuration
 168 m5.instantiate(root)
 169
 170 if options.checkpoint_dir:
 171     cptdir = options.checkpoint_dir
 172 else:
 173     cptdir = os.getcwd()
 174
 175 if options.checkpoint_restore:
 176     from os.path import isdir
 177     from os import listdir, getcwd
 178     import re
 179
 180     if not isdir(cptdir):
 181         m5.panic("checkpoint dir %s does not exist!" % cptdir)
 182
 183     dirs = listdir(cptdir)
 184     expr = re.compile('cpt.([0-9]*)')
 185     cpts = []
 186     for dir in dirs:
 187         match = expr.match(dir)
 188         if match:
 189             cpts.append(match.group(1))
 190
 191     cpts.sort(lambda a,b: cmp(long(a), long(b)))
 192
 193     if options.checkpoint_restore > len(cpts):
 194         m5.panic('Checkpoint %d not found' % options.checkpoint_restore)
 195
 196     print "restoring checkpoint from ","/".join([cptdir, "cpt.%s" % cpts[options.checkpoint_restore - 1]])
 197     m5.restoreCheckpoint(root, "/".join([cptdir, "cpt.%s" % cpts[options.checkpoint_restore - 1]]))
 198
 199 if options.standard_switch:
 200     exit_event = m5.simulate(10000)
 201     ## when you change to Timing (or Atomic), you halt the system given
 202     ## as argument.  When you are finished with the system changes
 203     ## (including switchCpus), you must resume the system manually.
 204     ## You DON'T need to resume after just switching CPUs if you haven't
 205     ## changed anything on the system level.
 206     m5.changeToTiming(system)
 207     m5.switchCpus(switch_cpu_list)
 208     m5.resume(system)
 209
 210     exit_event = m5.simulate(500000000000)
 211     m5.switchCpus(switch_cpu_list1)
 212
 213 if options.maxtick:
 214     maxtick = options.maxtick
 215 elif options.maxtime:
 216     simtime = int(options.maxtime * root.clock.value)
 217     print "simulating for: ", simtime
 218     maxtick = simtime
 219 else:
 220     maxtick = -1
 221
 222 num_checkpoints = 0
 223
 224 exit_cause = ''
 225
 226 if options.take_checkpoints:
 227     [when, period] = options.take_checkpoints.split(",", 1)
 228     when = int(when)
 229     period = int(period)
 230
 231     exit_event = m5.simulate(when)
 232     while exit_event.getCause() == "checkpoint":
 233         exit_event = m5.simulate(when - m5.curTick())
 234
 235     if exit_event.getCause() == "simulate() limit reached":
 236         m5.checkpoint(root, cptdir + "cpt.%d")
 237         num_checkpoints += 1
 238
 239     sim_ticks = when
 240     exit_cause = "maximum %d checkpoints dropped" % options.max_checkpoints
 241     while num_checkpoints < options.max_checkpoints:
 242         if (sim_ticks + period) > maxtick and maxtick != -1:
 243             exit_event = m5.simulate(maxtick - sim_ticks)
 244             exit_cause = exit_event.getCause()
 245             break
 246         else:
 247             exit_event = m5.simulate(period)
 248             sim_ticks += period
 249             while exit_event.getCause() == "checkpoint":
 250                 exit_event = m5.simulate(period - m5.curTick())
 251             if exit_event.getCause() == "simulate() limit reached":
 252                 m5.checkpoint(root, cptdir + "cpt.%d")
 253                 num_checkpoints += 1
 254
 255 else: #no checkpoints being taken via this script
 256     exit_event = m5.simulate(maxtick)
 257
 258     while exit_event.getCause() == "checkpoint":
 259         m5.checkpoint(root, cptdir + "cpt.%d")
 260         num_checkpoints += 1
 261         if num_checkpoints == options.max_checkpoints:
 262             exit_cause =  "maximum %d checkpoints dropped" % options.max_checkpoints
 263             break
 264
 265         if maxtick == -1:
 266             exit_event = m5.simulate(maxtick)
 267         else:
 268             exit_event = m5.simulate(maxtick - m5.curTick())
 269
 270         exit_cause = exit_event.getCause()
 271
 272 if exit_cause == '':
 273     exit_cause = exit_event.getCause()
 274 print 'Exiting @ cycle', m5.curTick(), 'because ', exit_cause
 275
 276