From 44cb10540d9010f028cc36da32e118830ae8fe21 Mon Sep 17 00:00:00 2001
From: Liangfu Chen <liangfu.chen@icloud.com>
Date: Sun, 22 Dec 2019 06:19:56 +0800
Subject: [PATCH] [VTA] improved virtual memory mapping (#4545)

* [VTA] improved virtual memory mapping

* Update virtual_memory.cc
---
 vta/src/vmem/virtual_memory.cc | 14 ++++++++++++--
 1 file changed, 12 insertions(+), 2 deletions(-)

diff --git a/vta/src/vmem/virtual_memory.cc b/vta/src/vmem/virtual_memory.cc
index 20ffd00b9814..0bf2382e155e 100644
--- a/vta/src/vmem/virtual_memory.cc
+++ b/vta/src/vmem/virtual_memory.cc
@@ -66,9 +66,19 @@ void* VirtualMemoryManager::GetAddr(uint64_t phy_addr) {
 vta_phy_addr_t VirtualMemoryManager::GetPhyAddr(void* buf) {
   std::lock_guard<std::mutex> lock(mutex_);
   auto it = pmap_.find(buf);
-  CHECK(it != pmap_.end());
+  uint64_t offset = 0;
+  if (it == pmap_.end()) {
+    for (it = pmap_.begin(); it != pmap_.end(); it++) {
+      uint64_t bytes = it->second->num_pages << kPageBits;
+      if ((buf >= it->first) && (buf < static_cast<char*>(it->first) + bytes)) {
+        offset = static_cast<char*>(buf) - static_cast<char*>(it->first);
+        break;
+      }
+    }
+    CHECK(it != pmap_.end());
+  }
   Page* p = it->second.get();
-  return (p->ptable_begin + 1) << kPageBits;
+  return ((p->ptable_begin + 1) << kPageBits) + offset;
 }
 
 /*!