Merge pull request #95 from lukepfister/master

Add ability to get handle to CUDA Arrays
inducer · Nov 11, 2015 · 0a4afe0 · 0a4afe0
2 parents 88d143b + 2e0e998
commit 0a4afe0
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/doc/source/driver.rst b/doc/source/driver.rst
@@ -1283,6 +1283,12 @@ Arrays and Textures
         Return a :class:`ArrayDescriptor3D` object for this 3D array,
         like the one that was used to create it.  CUDA 2.0 and above only.
 
+    .. attribute:: handle
+
+       Return an :class:`int` representing the address in device memory where
+       this array resides.
+
+
 .. class:: SurfaceReference()
 
     .. note::

diff --git a/src/cpp/cuda.hpp b/src/cpp/cuda.hpp
@@ -1059,6 +1059,9 @@ namespace pycuda
 
       CUarray handle() const
       { return m_array; }
+
+    intptr_t handle_int() const
+    { return  (intptr_t) m_array; }
   };
 
   // }}}

diff --git a/src/wrapper/wrap_cudadrv.cpp b/src/wrapper/wrap_cudadrv.cpp
@@ -1429,6 +1429,7 @@ BOOST_PYTHON_MODULE(_driver)
       .def(py::init<const CUDA_ARRAY3D_DESCRIPTOR &>())
       .DEF_SIMPLE_METHOD(get_descriptor_3d)
 #endif
+      .add_property("handle", &cl::handle_int)
       ;
   }
   // }}}