1 files changed, 132 insertions, 0 deletions
diff --git a/src/mem/probes/mem_footprint.cc b/src/mem/probes/mem_footprint.cc
new file mode 100644
index 000000000..0c14389ed
--- /dev/null
+++ b/src/mem/probes/mem_footprint.cc
@@ -0,0 +1,132 @@
+/*
+ * Copyright (c) 2016 Google Inc.
+ * All rights reserved.
+ *
+ * The license below extends only to copyright in the software and
+ * shall not be construed as granting a license to any other
+ * intellectual property including but not limited to intellectual
+ * property relating to a hardware implementation of the
+ * functionality of the software licensed hereunder.  You may use the
+ * software subject to the license terms below provided that you
+ * ensure that this notice is replicated unmodified and in its
+ * entirety in all distributions of the software, modified or
+ * unmodified, in source code or in binary form.
+ *
+ * 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: Rahul Thakur
+ */
+
+#include "mem/probes/mem_footprint.hh"
+
+#include "base/intmath.hh"
+#include "params/MemFootprintProbe.hh"
+
+MemFootprintProbe::MemFootprintProbe(MemFootprintProbeParams *p)
+    : BaseMemProbe(p),
+      cacheLineSizeLg2(floorLog2(p->system->cacheLineSize())),
+      pageSizeLg2(floorLog2(p->page_size)),
+      totalCacheLinesInMem(p->system->memSize() / p->system->cacheLineSize()),
+      totalPagesInMem(p->system->memSize() / p->page_size),
+      cacheLines(),
+      cacheLinesAll(),
+      pages(),
+      pagesAll(),
+      system(p->system)
+{
+    fatal_if(!isPowerOf2(system->cacheLineSize()),
+             "MemFootprintProbe expects cache line size is power of 2.");
+    fatal_if(!isPowerOf2(p->page_size),
+             "MemFootprintProbe expects page size parameter is power of 2");
+}
+
+void
+MemFootprintProbe::regStats()
+{
+    BaseMemProbe::regStats();
+
+    using namespace Stats;
+    // clang-format off
+    fpCacheLine.name(name() + ".cacheline")
+        .desc("Memory footprint at cache line granularity")
+        .flags(nozero | nonan);
+    fpCacheLineTotal.name(name() + ".cacheline_total")
+        .desc("Total memory footprint at cache line granularity since "
+              "simulation begin")
+        .flags(nozero | nonan);
+    fpPage.name(name() + ".page")
+        .desc("Memory footprint at page granularity")
+        .flags(nozero | nonan);
+    fpPageTotal.name(name() + ".page_total")
+        .desc("Total memory footprint at page granularity since simulation "
+              "begin")
+        .flags(nozero | nonan);
+    // clang-format on
+
+    registerResetCallback(
+        new MakeCallback<MemFootprintProbe, &MemFootprintProbe::statReset>(
+            this));
+}
+
+void
+MemFootprintProbe::insertAddr(Addr addr, AddrSet *set, uint64_t limit)
+{
+    set->insert(addr);
+    assert(set->size() <= limit);
+}
+
+void
+MemFootprintProbe::handleRequest(const ProbePoints::PacketInfo &pi)
+{
+    if (!pi.cmd.isRequest() || !system->isMemAddr(pi.addr))
+        return;
+
+    const Addr cl_addr = (pi.addr >> cacheLineSizeLg2) << cacheLineSizeLg2;
+    const Addr page_addr = (pi.addr >> pageSizeLg2) << pageSizeLg2;
+    insertAddr(cl_addr, &cacheLines, totalCacheLinesInMem);
+    insertAddr(cl_addr, &cacheLinesAll, totalCacheLinesInMem);
+    insertAddr(page_addr, &pages, totalPagesInMem);
+    insertAddr(page_addr, &pagesAll, totalPagesInMem);
+
+    assert(cacheLines.size() <= cacheLinesAll.size());
+    assert(pages.size() <= pagesAll.size());
+
+    fpCacheLine = cacheLines.size() << cacheLineSizeLg2;
+    fpCacheLineTotal = cacheLinesAll.size() << cacheLineSizeLg2;
+    fpPage = pages.size() << pageSizeLg2;
+    fpPageTotal = pagesAll.size() << pageSizeLg2;
+}
+
+void
+MemFootprintProbe::statReset()
+{
+    cacheLines.clear();
+    pages.clear();
+}
+
+MemFootprintProbe *
+MemFootprintProbeParams::create()
+{
+    return new MemFootprintProbe(this);
+}