From 6569fba06d576a78f09ab3d3887243e46bde1934 Mon Sep 17 00:00:00 2001
From: foreman <dl.constructicon@amd.com>
Date: Fri, 15 Apr 2016 19:22:30 -0400
Subject: [PATCH] P4 to Git Change 1258779 by gandryey@gera-ocl on 2016/04/15
 19:16:15

	SWDEV-92245 - [CQE OCL][QR][G] Multiple 32/64 bit Bolt sample crashes on all CPUs. FaultyCL#1257532
	- Bolt calls map/unmap with the same region twice. Add a counter for the same region to track that case

Affected files ...

... //depot/stg/opencl/drivers/opencl/runtime/device/device.cpp#195 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/device.hpp#272 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/gpu/gpuvirtual.cpp#401 edit
... //depot/stg/opencl/drivers/opencl/runtime/device/pal/palvirtual.cpp#6 edit
---
 rocclr/runtime/device/device.cpp         | 12 +++++++-----
 rocclr/runtime/device/device.hpp         |  9 ++++++---
 rocclr/runtime/device/gpu/gpuvirtual.cpp |  6 +++++-
 rocclr/runtime/device/pal/palvirtual.cpp |  6 +++++-
 4 files changed, 23 insertions(+), 10 deletions(-)

diff --git a/rocclr/runtime/device/device.cpp b/rocclr/runtime/device/device.cpp
index 0d96f97977..c299f2cc74 100644
--- a/rocclr/runtime/device/device.cpp
+++ b/rocclr/runtime/device/device.cpp
@@ -689,11 +689,7 @@ Memory::saveMapInfo(
     WriteMapInfo* pInfo = &info;
     auto it = writeMapInfo_.find(mapAddress);
     if (it != writeMapInfo_.end()) {
-        LogWarning("Double map of the same region!");
-    }
-    else {
-        writeMapInfo_.insert(std::pair<const void*, WriteMapInfo>(mapAddress, info));
-        it  = writeMapInfo_.find(mapAddress);
+        LogWarning("Double map of the same or overlapped region!");
         pInfo = &it->second;
     }
 
@@ -707,6 +703,12 @@ Memory::saveMapInfo(
         pInfo->unmapRead_ = true;
     }
     pInfo->baseMip_ = baseMip;
+
+    // Insert into the map if it's the first region
+    if (++pInfo->count_ == 1) {
+        writeMapInfo_.insert(std::pair<const void*, WriteMapInfo>(mapAddress, info));
+    }
+
 }
 
 Program::Program(amd::Device& device)
diff --git a/rocclr/runtime/device/device.hpp b/rocclr/runtime/device/device.hpp
index 239882ab32..901e42049a 100644
--- a/rocclr/runtime/device/device.hpp
+++ b/rocclr/runtime/device/device.hpp
@@ -629,13 +629,14 @@ public:
         SyncFlags(): value_(0) {}
     };
 
-    struct WriteMapInfo: public amd::EmbeddedObject
+    struct WriteMapInfo: public amd::HeapObject
     {
         amd::Coord3D    origin_;    //!< Origin of the map location
         amd::Coord3D    region_;    //!< Mapped region
         amd::Image*     baseMip_;   //!< The base mip level for images
         union {
             struct {
+                uint32_t    count_: 8;      //!< The same map region counter
                 uint32_t    unmapWrite_: 1; //!< Unmap write operation
                 uint32_t    unmapRead_: 1;  //!< Unmap read operation
                 uint32_t    entire_: 1;     //!< Process the entire memory
@@ -766,7 +767,7 @@ public:
         auto it = writeMapInfo_.find(mapAddress);
         if (it == writeMapInfo_.end()) {
             if (writeMapInfo_.size() == 0) {
-                assert(false && "Unmap() call without map!");
+                LogError("Unmap is a NOP!");
                 return nullptr;
             }
             LogWarning("Unknown unmap signature!");
@@ -786,7 +787,9 @@ public:
             // Get the first map info
             it = writeMapInfo_.begin();
         }
-        writeMapInfo_.erase(it);
+        if (--it->second.count_ == 0) {
+            writeMapInfo_.erase(it);
+        }
     }
 
     //! Returns state of memory direct access flag
diff --git a/rocclr/runtime/device/gpu/gpuvirtual.cpp b/rocclr/runtime/device/gpu/gpuvirtual.cpp
index 311b8c31ff..ef9773d503 100644
--- a/rocclr/runtime/device/gpu/gpuvirtual.cpp
+++ b/rocclr/runtime/device/gpu/gpuvirtual.cpp
@@ -1179,12 +1179,16 @@ VirtualGPU::submitUnmapMemory(amd::UnmapMemoryCommand& vcmd)
     // Make sure VirtualGPU has an exclusive access to the resources
     amd::ScopedLock lock(execution());
 
-    profilingBegin(vcmd, true);
     gpu::Memory* memory = dev().getGpuMemory(&vcmd.memory());
     amd::Memory* owner = memory->owner();
     bool    unmapMip = false;
     const device::Memory::WriteMapInfo* writeMapInfo =
         memory->writeMapInfo(vcmd.mapPtr());
+    if (nullptr == writeMapInfo) {
+        LogError("Unmap without map call");
+        return;
+    }
+    profilingBegin(vcmd, true);
 
     // Check if image is a mipmap and assign a saved view
     amd::Image* amdImage = owner->asImage();
diff --git a/rocclr/runtime/device/pal/palvirtual.cpp b/rocclr/runtime/device/pal/palvirtual.cpp
index 8909c0172a..2912ec7afc 100644
--- a/rocclr/runtime/device/pal/palvirtual.cpp
+++ b/rocclr/runtime/device/pal/palvirtual.cpp
@@ -1422,12 +1422,16 @@ VirtualGPU::submitUnmapMemory(amd::UnmapMemoryCommand& vcmd)
     // Make sure VirtualGPU has an exclusive access to the resources
     amd::ScopedLock lock(execution());
 
-    profilingBegin(vcmd, true);
     pal::Memory* memory = dev().getGpuMemory(&vcmd.memory());
     amd::Memory* owner = memory->owner();
     bool    unmapMip = false;
     const device::Memory::WriteMapInfo* writeMapInfo =
         memory->writeMapInfo(vcmd.mapPtr());
+    if (nullptr == writeMapInfo) {
+        LogError("Unmap without map call");
+        return;
+    }
+    profilingBegin(vcmd, true);
 
     // Check if image is a mipmap and assign a saved view
     amd::Image* amdImage = owner->asImage();