summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authorAndreas Sandberg <Andreas.Sandberg@arm.com>2012-11-02 11:32:02 -0500
committerAndreas Sandberg <Andreas.Sandberg@arm.com>2012-11-02 11:32:02 -0500
commitdc01535c7e800189c4cb46cab470f4b958be94c7 (patch)
tree031c5f3b6d8c3e43f493e2efbca413b32359f134
parent196397fea4e25f097b4a1624ee16fbbbc1c45b64 (diff)
downloadgem5-dc01535c7e800189c4cb46cab470f4b958be94c7.tar.xz
python: Rename doDrain()->drain() and make it do the right thing
There is no point in exporting the old drain() method in Simulate.py. It should only be used internally by doDrain(). This patch moves the old drain() method into doDrain() and renames doDrain() to drain().
-rw-r--r--configs/common/Simulation.py4
-rw-r--r--src/python/m5/simulate.py47
2 files changed, 26 insertions, 25 deletions
diff --git a/configs/common/Simulation.py b/configs/common/Simulation.py
index f6273a6f0..53ccf0a14 100644
--- a/configs/common/Simulation.py
+++ b/configs/common/Simulation.py
@@ -242,7 +242,7 @@ def repeatSwitch(testsys, repeat_switch_cpu_list, maxtick, switch_freq):
return exit_cause
print "draining the system"
- m5.doDrain(testsys)
+ m5.drain(testsys)
m5.switchCpus(repeat_switch_cpu_list)
m5.resume(testsys)
@@ -469,7 +469,7 @@ def run(options, root, testsys, cpu_class):
print "Switching CPUS @ tick %s" % (m5.curTick())
print "Simulation ends instruction count:%d" % \
(testsys.switch_cpus_1[0].max_insts_any_thread)
- m5.doDrain(testsys)
+ m5.drain(testsys)
m5.switchCpus(switch_cpu_list1)
m5.resume(testsys)
diff --git a/src/python/m5/simulate.py b/src/python/m5/simulate.py
index 4c6076581..ca2cc4a9c 100644
--- a/src/python/m5/simulate.py
+++ b/src/python/m5/simulate.py
@@ -157,28 +157,29 @@ atexit.register(stats.dump)
# register our C++ exit callback function with Python
atexit.register(internal.core.doExitCleanup)
-# This loops until all objects have been fully drained.
-def doDrain(root):
- all_drained = drain(root)
- while (not all_drained):
- all_drained = drain(root)
-
-# Tries to drain all objects. Draining might not be completed unless
-# all objects return that they are drained on the first call. This is
-# because as objects drain they may cause other objects to no longer
-# be drained.
+# Drain the system in preparation of a checkpoint or memory mode
+# switch.
def drain(root):
- all_drained = False
- dm = internal.drain.createDrainManager()
- unready_objs = sum(obj.drain(dm) for obj in root.descendants())
- # If we've got some objects that can't drain immediately, then simulate
- if unready_objs > 0:
- dm.setCount(unready_objs)
- simulate()
- else:
- all_drained = True
- internal.drain.cleanupDrainManager(dm)
- return all_drained
+ # Try to drain all objects. Draining might not be completed unless
+ # all objects return that they are drained on the first call. This
+ # is because as objects drain they may cause other objects to no
+ # longer be drained.
+ def _drain():
+ all_drained = False
+ dm = internal.drain.createDrainManager()
+ unready_objs = sum(obj.drain(dm) for obj in root.descendants())
+ # If we've got some objects that can't drain immediately, then simulate
+ if unready_objs > 0:
+ dm.setCount(unready_objs)
+ simulate()
+ else:
+ all_drained = True
+ internal.drain.cleanupDrainManager(dm)
+ return all_drained
+
+ all_drained = _drain()
+ while (not all_drained):
+ all_drained = _drain()
def resume(root):
for obj in root.descendants(): obj.drainResume()
@@ -187,7 +188,7 @@ def checkpoint(dir):
root = objects.Root.getInstance()
if not isinstance(root, objects.Root):
raise TypeError, "Checkpoint must be called on a root object."
- doDrain(root)
+ drain(root)
print "Writing checkpoint"
internal.core.serializeAll(dir)
resume(root)
@@ -197,7 +198,7 @@ def changeMemoryMode(system, mode):
raise TypeError, "Parameter of type '%s'. Must be type %s or %s." % \
(type(system), objects.Root, objects.System)
if system.getMemoryMode() != mode:
- doDrain(system)
+ drain(system)
system.setMemoryMode(mode)
else:
print "System already in target mode. Memory mode unchanged."