Browse Source

fix gpu queue partial copy issue

tags/v1.1.0
wilfChen 5 years ago
parent
commit
7e1cc4a11a
1 changed files with 1 additions and 0 deletions
  1. +1
    -0
      mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc

+ 1
- 0
mindspore/ccsrc/runtime/device/gpu/blocking_queue.cc View File

@@ -58,6 +58,7 @@ BlockQueueStatus_T GpuQueue::Push(const std::vector<DataItemGpu> &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_;


Loading…
Cancel
Save