From 7e1cc4a11add5f573883f7bd5c1520ebd3458e8f Mon Sep 17 00:00:00 2001 From: wilfChen Date: Thu, 29 Oct 2020 09:51:20 +0800 Subject: [PATCH] fix gpu queue partial copy issue --- mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc | 1 + 1 file changed, 1 insertion(+) 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_;