[arrayfire] 163/284: Added getMappedPtr and unmapPtr functions in opencl memory

[Ghislain Vaillant]

This is an automated email from the git hooks/post-receive script.

ghisvail-guest pushed a commit to branch debian/experimental
in repository arrayfire.

commit 4275f5f2dda6a089fcbae77418204a5c1c76a2f3
Author: Shehzan Mohammed <shehzan at arrayfire.com>
Date:   Thu Jan 7 15:08:51 2016 -0500

    Added getMappedPtr and unmapPtr functions in opencl memory
---
 src/backend/opencl/memory.cpp | 26 ++++++++++++++++++++++++++
 src/backend/opencl/memory.hpp |  3 +++
 2 files changed, 29 insertions(+)

diff --git a/src/backend/opencl/memory.cpp b/src/backend/opencl/memory.cpp
index 141610d..924e370 100644
--- a/src/backend/opencl/memory.cpp
+++ b/src/backend/opencl/memory.cpp
@@ -281,6 +281,29 @@ namespace opencl
         return bufferPush((cl::Buffer *)ptr);
     }
 
+    template<typename T>
+    T *getMappedPtr(const cl::Buffer *buf)
+    {
+        int n = getActiveDeviceId();
+        mem_iter iter = memory_maps[n].find(const_cast<cl::Buffer*>(buf));
+
+        if (iter == memory_maps[n].end()) {
+            // Buffer not found in memory manager
+            // Very Very Bad
+            return NULL;
+        }
+        size_t alloc_bytes = iter->second.bytes;
+
+        T *ptr = (T*)getQueue().enqueueMapBuffer(
+                     *buf, true, CL_MAP_READ, 0, alloc_bytes);
+        return ptr;
+    }
+
+    void unmapPtr(const cl::Buffer *buf, void *ptr)
+    {
+        getQueue().enqueueUnmapMemObject(*buf, ptr);
+    }
+
     // pinned memory manager
     typedef struct {
         cl::Buffer *buf;
@@ -403,6 +426,7 @@ namespace opencl
     template void memPush(const T* ptr);                        \
     template T* pinnedAlloc(const size_t &elements);            \
     template void pinnedFree(T* ptr);                           \
+    template T* getMappedPtr(const cl::Buffer *buf);            \
 
     INSTANTIATE(float)
     INSTANTIATE(cfloat)
@@ -416,4 +440,6 @@ namespace opencl
     INSTANTIATE(uintl)
     INSTANTIATE(short)
     INSTANTIATE(ushort)
+
+    template void* getMappedPtr(const cl::Buffer *buf);
 }
diff --git a/src/backend/opencl/memory.hpp b/src/backend/opencl/memory.hpp
index 96292cd..f337a7a 100644
--- a/src/backend/opencl/memory.hpp
+++ b/src/backend/opencl/memory.hpp
@@ -32,6 +32,9 @@ namespace opencl
     template<typename T> void memPop(const T *ptr);
     template<typename T> void memPush(const T *ptr);
 
+    template<typename T> T *getMappedPtr(const cl::Buffer *buf);
+    void unmapPtr(const cl::Buffer *buf, void *ptr);
+
     template<typename T> T* pinnedAlloc(const size_t &elements);
     template<typename T> void pinnedFree(T* ptr);
 

-- 
Alioth's /usr/local/bin/git-commit-notice on /srv/git.debian.org/git/debian-science/packages/arrayfire.git