|
|
|
@@ -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_; |
|
|
|
|