p12346870
/
mindspore
forked from mindspore-Ecosystem/mindspore
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

!25628 Fix get data timeout 

Merge pull request !25628 from luoyang/issues2
This commit is contained in:
i-robot 2021-10-30 01:36:13 +00:00 committed by Gitee
parent b27520b6ae 0030573e38
commit 9bb04d41ae
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/ccsrc/minddata/dataset/engine/datasetops/device_queue_op.cc
View File

@ -612,7 +612,7 @@ Status DeviceQueueOp::SendDataToGPU() {
if (current_row.eoe()) {
MS_LOG(INFO) << "EOE Detected";
TensorRow eoe_flag(TensorRow::kFlagEOE);
RETURN_IF_NOT_OK(receive_queues_[num_buf % num_workers_]->Add(std::move(eoe_flag)));
RETURN_IF_NOT_OK(receive_queues_[num_buf++ % num_workers_]->Add(std::move(eoe_flag)));
}
if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) {