Add support for attaching a managed array to a stream.

inducer · Feb 18, 2014 · ef401e2 · ef401e2
1 parent 401b1ba
commit ef401e2
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/src/cpp/cuda.hpp b/src/cpp/cuda.hpp
@@ -1982,6 +1982,14 @@ namespace pycuda
       {
         return m_devptr;
       }
+
+      void attach(py::object stream_py, unsigned flags)
+      {
+        PYCUDA_PARSE_STREAM_PY;
+
+        CUDAPP_CALL_GUARDED(cuStreamAttachMemAsync, (s_handle, m_devptr, 0, flags));
+      }
+
   };
 #endif
 

diff --git a/src/wrapper/wrap_cudadrv.cpp b/src/wrapper/wrap_cudadrv.cpp
@@ -1141,6 +1141,8 @@ BOOST_PYTHON_MODULE(_driver)
 
     wrp
       .DEF_SIMPLE_METHOD(get_device_pointer)
+      .def("attach", &cl::attach, 
+        (py::arg("stream"), py::arg("mem_flags")))
       ;
   }
 #endif