diff options
author | Lisa Hsu <Lisa.Hsu@amd.com> | 2010-06-03 10:34:40 -0700 |
---|---|---|
committer | Lisa Hsu <Lisa.Hsu@amd.com> | 2010-06-03 10:34:40 -0700 |
commit | aeb6e2e3ec920f8590f36bdb33c03c2968ab5dde (patch) | |
tree | 38e13eb6990c8983dd18044c9467ddd577fe24dc /util/checkpoint-aggregator.py | |
parent | d2186857b15ccdfe58b6d9e1369946253c28fb02 (diff) | |
download | gem5-aeb6e2e3ec920f8590f36bdb33c03c2968ab5dde.tar.xz |
utils: checkpoint aggregator: some physmem files are too big to read at once,
break it up into reading one page at a time. Also, avoid redoing a aggregating a checkpoint that's
already done.
--HG--
rename : util/checkpoint-aggregator.py => util/checkpoint_aggregator.py
Diffstat (limited to 'util/checkpoint-aggregator.py')
-rwxr-xr-x | util/checkpoint-aggregator.py | 167 |
1 files changed, 0 insertions, 167 deletions
diff --git a/util/checkpoint-aggregator.py b/util/checkpoint-aggregator.py deleted file mode 100755 index 6e40db01e..000000000 --- a/util/checkpoint-aggregator.py +++ /dev/null @@ -1,167 +0,0 @@ -# Copyright (c) 2009 The Regents of The University of Michigan -# All rights reserved. -# -# Redistribution and use in source and binary forms, with or without -# modification, are permitted provided that the following conditions are -# met: redistributions of source code must retain the above copyright -# notice, this list of conditions and the following disclaimer; -# redistributions in binary form must reproduce the above copyright -# notice, this list of conditions and the following disclaimer in the -# documentation and/or other materials provided with the distribution; -# neither the name of the copyright holders nor the names of its -# contributors may be used to endorse or promote products derived from -# this software without specific prior written permission. -# -# THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS -# "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT -# LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR -# A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT -# OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, -# SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT -# LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, -# DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY -# THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT -# (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE -# OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. -# -# Authors: Lisa Hsu - -from ConfigParser import ConfigParser -import gzip - -import sys, re, optparse, os - -class myCP(ConfigParser): - def __init__(self): - ConfigParser.__init__(self) - - def optionxform(self, optionstr): - return optionstr - -def aggregate(options, args): - merged = myCP() - page_ptr = 0 - - allfiles = os.listdir(os.getcwd()) - cpts = [] - for arg in args: - found = False - for f in allfiles: - if re.compile("cpt." + arg + ".\d+").search(f): - found = True - cpts.append(f) - break - if not found: - print "missing checkpoint: ", arg - sys.exit(1) - - dirname = "-".join([options.prefix, "cpt"]) - print dirname - agg_name = "-".join(args) - print agg_name - fullpath = os.path.join("..", dirname, "cpt." + agg_name + ".10000") - if not os.path.isdir(fullpath): - os.system("mkdir -p " + fullpath) - - myfile = open(fullpath + "/system.physmem.physmem", "wb+") - merged_mem = gzip.GzipFile(fileobj=myfile, mode="wb") - - max_curtick = 0 - when = 0 - for (i, arg) in enumerate(args): - config = myCP() - config.readfp(open(cpts[i] + "/m5.cpt")) - - for sec in config.sections(): - if re.compile("cpu").search(sec): - newsec = re.sub("cpu", "cpu" + str(i), sec) - merged.add_section(newsec) - if re.compile("workload$").search(sec): - merged.set(newsec, "M5_pid", i) - - items = config.items(sec) - for item in items: - if item[0] == "ppn": - if config.getint(sec, "tag") != 0: - merged.set(newsec, item[0], int(item[1]) + page_ptr) - continue - elif item[0] == "asn": - tmp = re.compile("(.*).Entry(\d+)").search(sec).groups() - if config.has_option(tmp[0], "nlu"): - size = config.getint(tmp[0], "nlu") - if int(tmp[1]) < size: - merged.set(newsec, item[0], i) - continue - else: - merged.set(newsec, item[0], i) - continue - merged.set(newsec, item[0], item[1]) - elif sec == "system": - pass - elif sec == "Globals": - tick = config.getint(sec, "curTick") - if tick > max_curtick: - max_curtick = tick - when = config.getint("system.cpu.tickEvent", "_when") - else: - if i == 0: - print sec - merged.add_section(sec) - for item in config.items(sec): - merged.set(sec, item[0], item[1]) - if item[0] == "curtick": - merged.optionxform(str("curTick")) - elif item[0] == "numevents": - merged.optionxform(str("numEvents")) - - page_ptr = page_ptr + int(config.get("system", "page_ptr")) - - ### memory stuff - f = open(cpts[i] + "/system.physmem.physmem", "rb") - gf = gzip.GzipFile(fileobj=f, mode="rb") - bytes = int(config.get("system", "page_ptr")) << 13 - print "bytes to be read: ", bytes - - bytesRead = gf.read(int(config.get("system", "page_ptr")) << 13) - merged_mem.write(bytesRead) - - gf.close() - f.close() - - merged.add_section("system") - merged.set("system", "page_ptr", page_ptr) - print "WARNING: " - print "Make sure the simulation using this checkpoint has at least " - if page_ptr > (1<<20): - print "8G ", - elif page_ptr > (1<<19): - print "4G ", - elif page_ptr > (1<<18): - print "2G ", - elif page_ptr > (1<<17): - print "1G ", - elif page_ptr > (1<<16): - print "512KB ", - else: - print "this is a small sim, you're probably fine", - print "of memory." - - merged.add_section("Globals") - merged.set("Globals", "curTick", max_curtick) - - for i in xrange(len(args)): - merged.set("system.cpu" + str(i) + ".tickEvent", "_when", when) - - merged.write(file(fullpath + "/m5.cpt", "wb")) - merged_mem.close() - myfile.close() - -if __name__ == "__main__": - - parser = optparse.OptionParser() - parser.add_option("--prefix", type="string", default="agg") - - (options, args) = parser.parse_args() - - aggregate(options, args) - |