Don't count mapped allocations (#42)

Mapped allocations don't use device memory, so we will not count them with the device memory counter.
libocca · Nov 24, 2016 · 381e886 · 381e886
1 parent 246fff7
commit 381e886
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/base.cpp b/src/base.cpp
@@ -1588,7 +1588,8 @@ namespace occa {
   void memory::free() {
     checkIfInitialized();
 
-    mHandle->dHandle->bytesAllocated -= (mHandle->size);
+    if(!mHandle->isMapped())
+      mHandle->dHandle->bytesAllocated -= (mHandle->size);
 
     if(mHandle->uvaPtr) {
       uvaMap.erase(mHandle->uvaPtr);
@@ -1617,7 +1618,8 @@ namespace occa {
   void memory::detach() {
     checkIfInitialized();
 
-    mHandle->dHandle->bytesAllocated -= (mHandle->size);
+    if(!mHandle->isMapped())
+      mHandle->dHandle->bytesAllocated -= (mHandle->size);
 
     if(mHandle->uvaPtr) {
       uvaMap.erase(mHandle->uvaPtr);
@@ -2260,8 +2262,6 @@ namespace occa {
     mem.mHandle          = dHandle->mappedAlloc(bytes, src);
     mem.mHandle->dHandle = dHandle;
 
-    dHandle->bytesAllocated += bytes;
-
     return mem;
   }