From 8ae4f45bc4782b4ab1dc95dbca183e2cd926fc5b Mon Sep 17 00:00:00 2001
From: Kevin Lim <ktlim@umich.edu>
Date: Thu, 6 Jul 2006 16:06:00 -0400
Subject: Fixes for draining.

src/cpu/simple/timing.cc:
    Update for changed return values.
src/python/m5/__init__.py:
    Loop in order to make sure all objects are really drained.  Objects may become undrained as other objects become drained (e.g. a bus-bridge has a packet, while a bus is empty, and the first drain() will cause the bus-bridge to give the packet to the bus).

    The only case we know every object is actually drained is if they all return immediately that they are drained.

--HG--
extra : convert_revision : 80057a1d6d30381bd0b67b23549bd202f447c5cb
---
 src/python/m5/__init__.py | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

(limited to 'src/python')

diff --git a/src/python/m5/__init__.py b/src/python/m5/__init__.py
index 579785a46..7d35ee8b8 100644
--- a/src/python/m5/__init__.py
+++ b/src/python/m5/__init__.py
@@ -213,14 +213,28 @@ atexit.register(cc_main.doExitCleanup)
 # matter since most scripts will probably 'from m5.objects import *'.
 import objects
 
+# 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.
+def drain(root):
+    all_drained = False
     drain_event = cc_main.createCountedDrain()
     unready_objects = root.startDrain(drain_event, True)
     # If we've got some objects that can't drain immediately, then simulate
     if unready_objects > 0:
         drain_event.setCount(unready_objects)
         simulate()
+    else:
+        all_drained = True
     cc_main.cleanupCountedDrain(drain_event)
+    return all_drained
 
 def resume(root):
     root.resume()
-- 
cgit v1.2.3


From 6872b99c29cd4263062bb8b3ef15aa5a9f2532d4 Mon Sep 17 00:00:00 2001
From: Kevin Lim <ktlim@umich.edu>
Date: Thu, 6 Jul 2006 23:16:22 -0400
Subject: Be sure to call resume after restoring from a checkpoint.

--HG--
extra : convert_revision : 4d672917038779a23f4ce7eb5d4e3039c1f5d726
---
 src/python/m5/__init__.py | 1 +
 1 file changed, 1 insertion(+)

(limited to 'src/python')

diff --git a/src/python/m5/__init__.py b/src/python/m5/__init__.py
index 7d35ee8b8..dc3af7000 100644
--- a/src/python/m5/__init__.py
+++ b/src/python/m5/__init__.py
@@ -250,6 +250,7 @@ def checkpoint(root):
 def restoreCheckpoint(root):
     print "Restoring from checkpoint"
     cc_main.unserializeAll()
+    resume(root)
 
 def changeToAtomic(system):
     if not isinstance(system, objects.Root) and not isinstance(system, System):
-- 
cgit v1.2.3