Add ability to get handle to CUDA Arrays

inducer · Nov 10, 2015 · 86ecf27 · 86ecf27
1 parent 88d143b
commit 86ecf27
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
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)
       ;
   }
   // }}}