1 files changed, 5 insertions, 16 deletions
diff --git a/src/python/swig/stats.i b/src/python/swig/stats.i
index 3bdd2c9aa..41115445c 100644
--- a/src/python/swig/stats.i
+++ b/src/python/swig/stats.i
@@ -43,6 +43,7 @@
 #include "base/statistics.hh"
 #include "sim/core.hh"
 #include "sim/stat_control.hh"
+#include "sim/stat_register.hh"
 
 namespace Stats {
 template <class T>
@@ -64,20 +65,6 @@ Stats_Info_flags_set(Info *info, FlagsType flags)
     info->flags = flags;
 }
 
-inline void
-processResetQueue()
-{
-    extern CallbackQueue resetQueue;
-    resetQueue.process();
-}
-
-inline void
-processDumpQueue()
-{
-    extern CallbackQueue dumpQueue;
-    dumpQueue.process();
-}
-
 inline char *
 PCC(const char *string)
 {
@@ -102,13 +89,13 @@ call_module_function(const char *module_name, const char *func_name)
 }
 
 void
-dump()
+pythonDump()
 {
     call_module_function("m5.stats", "dump");
 }
 
 void
-reset()
+pythonReset()
 {
     call_module_function("m5.stats", "reset");
 }
@@ -150,6 +137,8 @@ template <class T> T cast_info(Info *info);
 void initSimStats();
 Output *initText(const std::string &filename, bool desc);
 
+void registerPythonStatsHandlers();
+
 void schedStatEvent(bool dump, bool reset,
                     Tick when = curTick(), Tick repeat = 0);