mem: Track TLB entries in the lookup cache as pointers.

Using the architectural page table on x86 and the functional page table
on ARM, both with the twolf benchmark in SE mode, there was no
performance penalty for doing so, and again possibly a performance
improvement. By using a pointer instead of an inline instance, it's
possible for the actual type of the TLB entry to be hidden somewhat,
taking a step towards abstracting away another aspect of the ISAs.

Since the TLB entries are no longer overwritten and now need to be
allocated and freed, this change introduces return types from the
updateCache and eraseCacheEntry functions. These functions will return
the pointer to any entry which has been displaced from the cache which
the caller can either free or ignore, depending on whether the entry
has a purpose outside of the cache.

Because the functional page table stores its entries over a longer time
period, it will generally not delete the pointer returned from those
functions. The "architechtural" page table, ie the one which is backed
by memory, doesn't have any other use for the TlbEntrys and will delete
them. That leads to more news and deletes than there used to be.

To address that, and also to speed up the architectural page table in
general, it would be a good idea to augment the functional page table
with an image of the table in memory, instead of replacing it with one.
The functional page table would provide quick lookups and also avoid
having to translate page table entries to TLB entries, making
performance essentially equivalent to the functional case. The backing
page tables, which are primarily for consumption by the physical
hardware when in KVM, can be updated when mappings change but otherwise
left alone.

If we end up doing that, we could just let the ISA specific process
classes enable whatever additional TLB machinery they need, likely
a backing copy in memory, without any knowledge or involvement from
the ISA agnostic class. We would be able to get rid of the useArchPT
setting and the bits of code in the configs which set it.

Change-Id: I2e21945cd852bb1b3d0740fe6a4c5acbfd9548c5
Reviewed-on: https://gem5-review.googlesource.com/6983
Maintainer: Gabe Black <gabeblack@google.com>
Reviewed-by: Brandon Potter <Brandon.Potter@amd.com>
Reviewed-by: Jason Lowe-Power <jason@lowepower.com>
Reviewed-by: Anthony Gutierrez <anthony.gutierrez@amd.com>
diff --git a/src/mem/multi_level_page_table_impl.hh b/src/mem/multi_level_page_table_impl.hh
index 4ff5f5c..ab7f066 100644
--- a/src/mem/multi_level_page_table_impl.hh
+++ b/src/mem/multi_level_page_table_impl.hh
@@ -164,10 +164,10 @@
             p.write<PageTableEntry>(PTE_addr, PTE);
             DPRINTF(MMU, "New mapping: %#x-%#x\n", vaddr, paddr);
 
-            eraseCacheEntry(vaddr);
-            updateCache(vaddr, TlbEntry(pid, vaddr, paddr,
-                                        flags & Uncacheable,
-                                        flags & ReadOnly));
+            delete eraseCacheEntry(vaddr);
+            delete updateCache(vaddr, new TlbEntry(pid, vaddr, paddr,
+                                                   flags & Uncacheable,
+                                                   flags & ReadOnly));
         }
 
     }
@@ -211,8 +211,9 @@
                 DPRINTF(MMU, "Remapping: %#x-%#x\n", vaddr, new_PTE_addr);
             }
 
-            eraseCacheEntry(vaddr);
-            updateCache(new_vaddr, TlbEntry(pid, new_vaddr, paddr,
+            delete eraseCacheEntry(vaddr);
+            delete updateCache(new_vaddr,
+                               new TlbEntry(pid, new_vaddr, paddr,
                                             pTableISAOps.isUncacheable(PTE),
                                             pTableISAOps.isReadOnly(PTE)));
         } else {
@@ -243,7 +244,7 @@
                 p.write<PageTableEntry>(PTE_addr, PTE);
                 DPRINTF(MMU, "Unmapping: %#x\n", vaddr);
             }
-           eraseCacheEntry(vaddr);
+           delete eraseCacheEntry(vaddr);
         } else {
             fatal("Page fault while unmapping");
         }
@@ -277,16 +278,16 @@
 {
     Addr page_addr = pageAlign(vaddr);
 
-    if (pTableCache[0].valid && pTableCache[0].vaddr == page_addr) {
-        entry = pTableCache[0].entry;
+    if (pTableCache[0].entry && pTableCache[0].vaddr == page_addr) {
+        entry = *pTableCache[0].entry;
         return true;
     }
-    if (pTableCache[1].valid && pTableCache[1].vaddr == page_addr) {
-        entry = pTableCache[1].entry;
+    if (pTableCache[1].entry && pTableCache[1].vaddr == page_addr) {
+        entry = *pTableCache[1].entry;
         return true;
     }
-    if (pTableCache[2].valid && pTableCache[2].vaddr == page_addr) {
-        entry = pTableCache[2].entry;
+    if (pTableCache[2].entry && pTableCache[2].vaddr == page_addr) {
+        entry = *pTableCache[2].entry;
         return true;
     }
 
@@ -299,10 +300,11 @@
         if (pnum == 0)
             return false;
 
-        entry = TlbEntry(pid, vaddr, pnum << PageShift,
-                         pTableISAOps.isUncacheable(PTE),
-                         pTableISAOps.isReadOnly(PTE));
-        updateCache(page_addr, entry);
+        TlbEntry *new_entry = new TlbEntry(pid, vaddr, pnum << PageShift,
+                                           pTableISAOps.isUncacheable(PTE),
+                                           pTableISAOps.isReadOnly(PTE));
+        entry = *new_entry;
+        delete updateCache(page_addr, new_entry);
     } else {
         return false;
     }
diff --git a/src/mem/page_table.cc b/src/mem/page_table.cc
index 0abe8ea..4f06c29 100644
--- a/src/mem/page_table.cc
+++ b/src/mem/page_table.cc
@@ -56,6 +56,8 @@
 
 FuncPageTable::~FuncPageTable()
 {
+    for (auto &iter : pTable)
+        delete iter.second;
 }
 
 void
@@ -68,12 +70,20 @@
     DPRINTF(MMU, "Allocating Page: %#x-%#x\n", vaddr, vaddr+ size);
 
     for (; size > 0; size -= pageSize, vaddr += pageSize, paddr += pageSize) {
-        if (!clobber && (pTable.find(vaddr) != pTable.end())) {
-            // already mapped
-            fatal("FuncPageTable::allocate: addr 0x%x already mapped", vaddr);
+        auto it = pTable.find(vaddr);
+        if (it != pTable.end()) {
+            if (clobber) {
+                delete it->second;
+            } else {
+                // already mapped
+                fatal("FuncPageTable::allocate: addr %#x already mapped",
+                      vaddr);
+            }
+        } else {
+            it = pTable.emplace(vaddr, nullptr).first;
         }
 
-        pTable[vaddr] = TheISA::TlbEntry(pid, vaddr, paddr,
+        it->second = new TheISA::TlbEntry(pid, vaddr, paddr,
                                          flags & Uncacheable,
                                          flags & ReadOnly);
         eraseCacheEntry(vaddr);
@@ -93,13 +103,15 @@
     for (; size > 0;
          size -= pageSize, vaddr += pageSize, new_vaddr += pageSize)
     {
-        assert(pTable.find(vaddr) != pTable.end());
+        auto new_it = pTable.find(new_vaddr);
+        auto old_it = pTable.find(vaddr);
+        assert(old_it != pTable.end() && new_it == pTable.end());
 
-        pTable[new_vaddr] = pTable[vaddr];
-        pTable.erase(vaddr);
+        new_it->second = old_it->second;
+        pTable.erase(old_it);
         eraseCacheEntry(vaddr);
-        pTable[new_vaddr].updateVaddr(new_vaddr);
-        updateCache(new_vaddr, pTable[new_vaddr]);
+        new_it->second->updateVaddr(new_vaddr);
+        updateCache(new_vaddr, new_it->second);
     }
 }
 
@@ -107,7 +119,7 @@
 FuncPageTable::getMappings(std::vector<std::pair<Addr, Addr>> *addr_maps)
 {
     for (auto &iter : pTable)
-        addr_maps->push_back(make_pair(iter.first, iter.second.pageStart()));
+        addr_maps->push_back(make_pair(iter.first, iter.second->pageStart()));
 }
 
 void
@@ -118,9 +130,11 @@
     DPRINTF(MMU, "Unmapping page: %#x-%#x\n", vaddr, vaddr+ size);
 
     for (; size > 0; size -= pageSize, vaddr += pageSize) {
-        assert(pTable.find(vaddr) != pTable.end());
-        pTable.erase(vaddr);
+        auto it = pTable.find(vaddr);
+        assert(it != pTable.end());
         eraseCacheEntry(vaddr);
+        delete it->second;
+        pTable.erase(it);
     }
 
 }
@@ -145,16 +159,16 @@
 {
     Addr page_addr = pageAlign(vaddr);
 
-    if (pTableCache[0].valid && pTableCache[0].vaddr == page_addr) {
-        entry = pTableCache[0].entry;
+    if (pTableCache[0].entry && pTableCache[0].vaddr == page_addr) {
+        entry = *pTableCache[0].entry;
         return true;
     }
-    if (pTableCache[1].valid && pTableCache[1].vaddr == page_addr) {
-        entry = pTableCache[1].entry;
+    if (pTableCache[1].entry && pTableCache[1].vaddr == page_addr) {
+        entry = *pTableCache[1].entry;
         return true;
     }
-    if (pTableCache[2].valid && pTableCache[2].vaddr == page_addr) {
-        entry = pTableCache[2].entry;
+    if (pTableCache[2].entry && pTableCache[2].vaddr == page_addr) {
+        entry = *pTableCache[2].entry;
         return true;
     }
 
@@ -165,7 +179,7 @@
     }
 
     updateCache(page_addr, iter->second);
-    entry = iter->second;
+    entry = *iter->second;
     return true;
 }
 
@@ -209,7 +223,7 @@
         ScopedCheckpointSection sec(cp, csprintf("Entry%d", count++));
 
         paramOut(cp, "vaddr", pte.first);
-        pte.second.serialize(cp);
+        pte.second->serialize(cp);
     }
     assert(count == pTable.size());
 }
@@ -223,14 +237,13 @@
     for (int i = 0; i < count; ++i) {
         ScopedCheckpointSection sec(cp, csprintf("Entry%d", i));
 
-        std::unique_ptr<TheISA::TlbEntry> entry;
-        Addr vaddr;
-
-        paramIn(cp, "vaddr", vaddr);
-        entry.reset(new TheISA::TlbEntry());
+        TheISA::TlbEntry *entry = new TheISA::TlbEntry();
         entry->unserialize(cp);
 
-        pTable[vaddr] = *entry;
+        Addr vaddr;
+        paramIn(cp, "vaddr", vaddr);
+
+        pTable[vaddr] = entry;
     }
 }
 
diff --git a/src/mem/page_table.hh b/src/mem/page_table.hh
index f784b21..fa58487 100644
--- a/src/mem/page_table.hh
+++ b/src/mem/page_table.hh
@@ -58,9 +58,8 @@
 {
   protected:
     struct cacheElement {
-        bool valid;
         Addr vaddr;
-        TheISA::TlbEntry entry;
+        TheISA::TlbEntry *entry;
     };
 
     struct cacheElement pTableCache[3];
@@ -78,9 +77,9 @@
               pid(_pid), _name(__name)
     {
         assert(isPowerOf2(pageSize));
-        pTableCache[0].valid = false;
-        pTableCache[1].valid = false;
-        pTableCache[2].valid = false;
+        pTableCache[0].entry = nullptr;
+        pTableCache[1].entry = nullptr;
+        pTableCache[2].entry = nullptr;
     }
 
     virtual ~PageTableBase() {};
@@ -162,36 +161,46 @@
      * Update the page table cache.
      * @param vaddr virtual address (page aligned) to check
      * @param pte page table entry to return
+     * @return A pointer to any entry which is displaced from the cache.
      */
-    inline void updateCache(Addr vaddr, TheISA::TlbEntry entry)
+    TheISA::TlbEntry *
+    updateCache(Addr vaddr, TheISA::TlbEntry *entry)
     {
+        TheISA::TlbEntry *evicted = pTableCache[2].entry;
+
         pTableCache[2].entry = pTableCache[1].entry;
         pTableCache[2].vaddr = pTableCache[1].vaddr;
-        pTableCache[2].valid = pTableCache[1].valid;
 
         pTableCache[1].entry = pTableCache[0].entry;
         pTableCache[1].vaddr = pTableCache[0].vaddr;
-        pTableCache[1].valid = pTableCache[0].valid;
 
         pTableCache[0].entry = entry;
         pTableCache[0].vaddr = vaddr;
-        pTableCache[0].valid = true;
+
+        return evicted;
     }
 
     /**
      * Erase an entry from the page table cache.
      * @param vaddr virtual address (page aligned) to check
+     * @return A pointer to the entry (if any) which is kicked out.
      */
-    inline void eraseCacheEntry(Addr vaddr)
+    TheISA::TlbEntry *
+    eraseCacheEntry(Addr vaddr)
     {
+        TheISA::TlbEntry *evicted = nullptr;
         // Invalidate cached entries if necessary
-        if (pTableCache[0].valid && pTableCache[0].vaddr == vaddr) {
-            pTableCache[0].valid = false;
-        } else if (pTableCache[1].valid && pTableCache[1].vaddr == vaddr) {
-            pTableCache[1].valid = false;
-        } else if (pTableCache[2].valid && pTableCache[2].vaddr == vaddr) {
-            pTableCache[2].valid = false;
+        if (pTableCache[0].entry && pTableCache[0].vaddr == vaddr) {
+            evicted = pTableCache[0].entry;
+            pTableCache[0].entry = nullptr;
+        } else if (pTableCache[1].entry && pTableCache[1].vaddr == vaddr) {
+            evicted = pTableCache[1].entry;
+            pTableCache[1].entry = nullptr;
+        } else if (pTableCache[2].entry && pTableCache[2].vaddr == vaddr) {
+            evicted = pTableCache[2].entry;
+            pTableCache[2].entry = nullptr;
         }
+        return evicted;
     }
 
     virtual void getMappings(std::vector<std::pair<Addr, Addr>>
@@ -204,7 +213,7 @@
 class FuncPageTable : public PageTableBase
 {
   private:
-    typedef std::unordered_map<Addr, TheISA::TlbEntry> PTable;
+    typedef std::unordered_map<Addr, TheISA::TlbEntry *> PTable;
     typedef PTable::iterator PTableItr;
     PTable pTable;