expose cuda stream to users

paddle/fluid/pybind/cuda_streams_py.cc

@@ -202,6 +202,28 @@ void BindCudaStream(py::module *m_ptr) {
 
            )DOC",
            py::arg("event") = nullptr)
+      .def_property_readonly(
+          "cuda_stream",
+          [](paddle::platform::stream::CUDAStream &self) {
+            VLOG(10) << self.raw_stream();
+            return reinterpret_cast<std::uintptr_t>(self.raw_stream());
+          },
+          R"DOC(
+      retrun the raw cuda stream of type cudaStream_t as type int.
+
+      Examples:
+        .. code-block:: python
+
+            # required: gpu
+            import paddle
+            import ctypes
+            cuda_stream = paddle.device.cuda.current_stream().cuda_stream
+            print(cuda_stream)
+
+            ptr = ctypes.c_void_p(cuda_stream)  # convert back to void*
+            print(ptr)
+
+           )DOC")
 #endif
       .def("__init__",
            [](paddle::platform::stream::CUDAStream &self,

python/paddle/fluid/tests/unittests/test_cuda_stream_event.py

@@ -14,6 +14,7 @@
 
 from paddle.device import cuda
 import paddle
+import ctypes
 
 import unittest
 import numpy as np
@@ -156,5 +157,14 @@ def test_set_current_stream_raise_error(self):
                               None)
 
 
+class TestRawStream(unittest.TestCase):
+    def test_cuda_stream(self):
+        if paddle.is_compiled_with_cuda():
+            cuda_stream = paddle.device.cuda.current_stream().cuda_stream
+            print(cuda_stream)
+            self.assertTrue(type(cuda_stream) is int)
+            ptr = ctypes.c_void_p(cuda_stream)
+
+
 if __name__ == "__main__":
     unittest.main()