erase empty elements to prevent the map size growing

close issue ROCm#48
yan-ming · May 4, 2016 · 783ae6a · 783ae6a
1 parent f168e44
commit 783ae6a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/lib/hsa/mcwamp_hsa.cpp b/lib/hsa/mcwamp_hsa.cpp
@@ -653,6 +653,7 @@ class HSAQueue final : public KalmarQueue
 
         // clear data in kernelBufferMap
         kernelBufferMap[ker].clear();
+        kernelBufferMap.erase(ker);
 
         delete(dispatch);
     }
@@ -697,6 +698,7 @@ class HSAQueue final : public KalmarQueue
 
         // clear data in kernelBufferMap
         kernelBufferMap[ker].clear();
+        kernelBufferMap.erase(ker);
 
         return sp_dispatch;
     }