fix cuda Stream record_event bug (#34285)

d953f8a9 · chentianyu03 · GitHub · 32cb0f5a · d953f8a9
隐藏空白更改
内联并排

Showing with 1 addition and 2 deletion

paddle/fluid/pybind/cuda_streams_py.cc paddle/fluid/pybind/cuda_streams_py.cc +1 -2

未找到文件。
--- a/paddle/fluid/pybind/cuda_streams_py.cc
+++ b/paddle/fluid/pybind/cuda_streams_py.cc
@@ -164,8 +164,7 @@ void BindCudaStream(py::module *m_ptr) {
           [](paddle::platform::stream::CUDAStream &self,
              paddle::platform::CudaEvent *event) {
             if (event == nullptr) {
-               auto event_tmp = paddle::platform::CudaEvent();
-               event = &event_tmp;
+               event = new paddle::platform::CudaEvent();
             }
             event->Record(self);
             return event;