1 # Copyright (c) 2005 The Regents of The University of Michigan
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.
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.
27 # Authors: Nathan Binkert
30 import sys
, os
, time
, atexit
, optparse
32 # import the SWIG-wrapped main C++ functions
34 # import a few SWIG-wrapped items (those that are likely to be used
35 # directly by user scripts) completely into this module for
37 from cc_main
import simulate
, SimLoopExitEvent
, setCheckpointDir
39 # import the m5 compile options
42 # define this here so we can use it right away if necessary
44 print >>sys
.stderr
, 'panic:', string
47 # Prepend given directory to system module search path. We may not
48 # need this anymore if we can structure our config library more like a
51 # if it's a relative path and we know what directory the current
52 # python script is in, make the path relative to that directory.
53 if not os
.path
.isabs(path
) and sys
.path
[0]:
54 path
= os
.path
.join(sys
.path
[0], path
)
55 path
= os
.path
.realpath(path
)
56 # sys.path[0] should always refer to the current script's directory,
57 # so place the new dir right after that.
58 sys
.path
.insert(1, path
)
61 # The m5 module's pointer to the parsed options object
65 # User should call this function after calling parse_args() to pass
66 # parsed standard option values back into the m5 module for
68 def setStandardOptions(_options
):
69 # Set module global var
72 # tell C++ about output directory
73 cc_main
.setOutputDir(options
.outdir
)
75 # Callback to set trace flags. Not necessarily the best way to do
76 # things in the long run (particularly if we change how these global
77 # options are handled).
78 def setTraceFlags(option
, opt_str
, value
, parser
):
79 objects
.Trace
.flags
= value
81 def setTraceStart(option
, opt_str
, value
, parser
):
82 objects
.Trace
.start
= value
84 def setTraceFile(option
, opt_str
, value
, parser
):
85 objects
.Trace
.file = value
87 def noPCSymbol(option
, opt_str
, value
, parser
):
88 objects
.ExecutionTrace
.pc_symbol
= False
90 def noPrintCycle(option
, opt_str
, value
, parser
):
91 objects
.ExecutionTrace
.print_cycle
= False
93 def noPrintOpclass(option
, opt_str
, value
, parser
):
94 objects
.ExecutionTrace
.print_opclass
= False
96 def noPrintThread(option
, opt_str
, value
, parser
):
97 objects
.ExecutionTrace
.print_thread
= False
99 def noPrintEA(option
, opt_str
, value
, parser
):
100 objects
.ExecutionTrace
.print_effaddr
= False
102 def noPrintData(option
, opt_str
, value
, parser
):
103 objects
.ExecutionTrace
.print_data
= False
105 def printFetchseq(option
, opt_str
, value
, parser
):
106 objects
.ExecutionTrace
.print_fetchseq
= True
108 def printCpseq(option
, opt_str
, value
, parser
):
109 objects
.ExecutionTrace
.print_cpseq
= True
111 def dumpOnExit(option
, opt_str
, value
, parser
):
112 objects
.Trace
.dump_on_exit
= True
114 def debugBreak(option
, opt_str
, value
, parser
):
115 objects
.Debug
.break_cycles
= value
117 def statsTextFile(option
, opt_str
, value
, parser
):
118 objects
.Statistics
.text_file
= value
120 # Standard optparse options. Need to be explicitly included by the
121 # user script when it calls optparse.OptionParser().
123 optparse
.make_option("--outdir", type="string", default
="."),
124 optparse
.make_option("--traceflags", type="string", action
="callback",
125 callback
=setTraceFlags
),
126 optparse
.make_option("--tracestart", type="int", action
="callback",
127 callback
=setTraceStart
),
128 optparse
.make_option("--tracefile", type="string", action
="callback",
129 callback
=setTraceFile
),
130 optparse
.make_option("--nopcsymbol",
131 action
="callback", callback
=noPCSymbol
,
132 help="Disable PC symbols in trace output"),
133 optparse
.make_option("--noprintcycle",
134 action
="callback", callback
=noPrintCycle
,
135 help="Don't print cycle numbers in trace output"),
136 optparse
.make_option("--noprintopclass",
137 action
="callback", callback
=noPrintOpclass
,
138 help="Don't print op class type in trace output"),
139 optparse
.make_option("--noprintthread",
140 action
="callback", callback
=noPrintThread
,
141 help="Don't print thread number in trace output"),
142 optparse
.make_option("--noprinteffaddr",
143 action
="callback", callback
=noPrintEA
,
144 help="Don't print effective address in trace output"),
145 optparse
.make_option("--noprintdata",
146 action
="callback", callback
=noPrintData
,
147 help="Don't print result data in trace output"),
148 optparse
.make_option("--printfetchseq",
149 action
="callback", callback
=printFetchseq
,
150 help="Print fetch sequence numbers in trace output"),
151 optparse
.make_option("--printcpseq",
152 action
="callback", callback
=printCpseq
,
153 help="Print correct path sequence numbers in trace output"),
154 optparse
.make_option("--dumponexit",
155 action
="callback", callback
=dumpOnExit
,
156 help="Dump trace buffer on exit"),
157 optparse
.make_option("--debugbreak", type="int", metavar
="CYCLE",
158 action
="callback", callback
=debugBreak
,
159 help="Cycle to create a breakpoint"),
160 optparse
.make_option("--statsfile", type="string", action
="callback",
161 callback
=statsTextFile
, metavar
="FILE",
162 help="Sets the output file for the statistics")
165 # make a SmartDict out of the build options for our local use
167 build_env
= smartdict
.SmartDict()
168 build_env
.update(defines
.m5_build_env
)
170 # make a SmartDict out of the OS environment too
171 env
= smartdict
.SmartDict()
172 env
.update(os
.environ
)
175 # Function to provide to C++ so it can look up instances based on paths
176 def resolveSimObject(name
):
177 obj
= config
.instanceDict
[name
]
178 return obj
.getCCObject()
180 # The final hook to generate .ini files. Called from the user script
181 # once the config is built.
182 def instantiate(root
):
183 config
.ticks_per_sec
= float(root
.clock
.frequency
)
184 # ugly temporary hack to get output to config.ini
185 sys
.stdout
= file(os
.path
.join(options
.outdir
, 'config.ini'), 'w')
187 sys
.stdout
.close() # close config.ini
188 sys
.stdout
= sys
.__stdout
__ # restore to original
189 cc_main
.loadIniFile(resolveSimObject
) # load config.ini into C++
190 root
.createCCObject()
193 noDot
= True # temporary until we fix dot
196 instance
.outputDot(dot
)
197 dot
.orientation
= "portrait"
199 dot
.ranksep
="equally"
201 dot
.write("config.dot")
202 dot
.write_ps("config.ps")
204 # Export curTick to user script.
206 return cc_main
.cvar
.curTick
208 # register our C++ exit callback function with Python
209 atexit
.register(cc_main
.doExitCleanup
)
211 # This import allows user scripts to reference 'm5.objects.Foo' after
212 # just doing an 'import m5' (without an 'import m5.objects'). May not
213 # matter since most scripts will probably 'from m5.objects import *'.
217 drain_event
= cc_main
.createCountedDrain()
218 unready_objects
= root
.startDrain(drain_event
, True)
219 # If we've got some objects that can't drain immediately, then simulate
220 if unready_objects
> 0:
221 drain_event
.setCount(unready_objects
)
223 cc_main
.cleanupCountedDrain(drain_event
)
228 def checkpoint(root
):
229 if not isinstance(root
, objects
.Root
):
230 raise TypeError, "Object is not a root object. Checkpoint must be called on a root object."
232 print "Writing checkpoint"
233 cc_main
.serializeAll()
236 def restoreCheckpoint(root
):
237 print "Restoring from checkpoint"
238 cc_main
.unserializeAll()
240 def changeToAtomic(system
):
241 if not isinstance(system
, objects
.Root
) and not isinstance(system
, System
):
242 raise TypeError, "Object is not a root or system object. Checkpoint must be "
243 "called on a root object."
245 print "Changing memory mode to atomic"
246 system
.changeTiming(cc_main
.SimObject
.Atomic
)
249 def changeToTiming(system
):
250 if not isinstance(system
, objects
.Root
) and not isinstance(system
, System
):
251 raise TypeError, "Object is not a root or system object. Checkpoint must be "
252 "called on a root object."
254 print "Changing memory mode to timing"
255 system
.changeTiming(cc_main
.SimObject
.Timing
)
258 def switchCpus(cpuList
):
259 if not isinstance(cpuList
, list):
260 raise RuntimeError, "Must pass a list to this function"
262 if not isinstance(i
, tuple):
263 raise RuntimeError, "List must have tuples of (oldCPU,newCPU)"
265 [old_cpus
, new_cpus
] = zip(*cpuList
)
268 if not isinstance(cpu
, objects
.BaseCPU
):
269 raise TypeError, "%s is not of type BaseCPU", cpu
271 if not isinstance(cpu
, objects
.BaseCPU
):
272 raise TypeError, "%s is not of type BaseCPU", cpu
274 # Drain all of the individual CPUs
275 drain_event
= cc_main
.createCountedDrain()
277 for old_cpu
in old_cpus
:
278 unready_cpus
+= old_cpu
.startDrain(drain_event
, False)
279 # If we've got some objects that can't drain immediately, then simulate
281 drain_event
.setCount(unready_cpus
)
283 cc_main
.cleanupCountedDrain(drain_event
)
284 # Now all of the CPUs are ready to be switched out
285 for old_cpu
in old_cpus
:
286 old_cpu
._ccObject
.switchOut()
288 print "Switching CPUs"
289 for new_cpu
in new_cpus
:
290 new_cpu
.takeOverFrom(old_cpus
[index
])
291 new_cpu
._ccObject
.resume()