diff --git a/mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc b/mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc index ab1f8c5886..228bfe7b34 100644 --- a/mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc +++ b/mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc @@ -58,6 +58,7 @@ BlockQueueStatus_T GpuQueue::Push(const std::vector &data) { node_info_[tail_].event_.reset(new cudaEvent_t()); CHECK_CUDA_RET_WITH_ERROR(cudaEventCreate(&(*(node_info_[tail_].event_))), "Cuda Create Event Failed"); + CHECK_CUDA_RET_WITH_ERROR(cudaEventRecord(*(node_info_[tail_].event_), stream_), "Cuda Create Event Failed"); node_info_[tail_].data_ = data; tail_ = (tail_ + 1) % (capacity_); ++size_;