md fix gpu core dump problem

pull/14107/head
xiefangqi 4 years ago
parent 0d1d043d80
commit de856a1bac

@ -391,7 +391,13 @@ Status DeviceQueueOp::PushDataToGPU() {
break;
}
if (!TaskManager::FindMe()->Interrupted() && !GpuBufferMgr::GetInstance().IsClosed()) {
RETURN_IF_NOT_OK(gpu_item_connector_->Pop(0, &items));
auto rc = gpu_item_connector_->Pop(0, &items);
// If the batches send by dataset are more than gpu calculate, gpu will core for no signal notify.
if (rc.IsError()) {
GpuBufferMgr::GetInstance().Close(handle);
GpuBufferMgr::GetInstance().CloseConfirm();
return rc;
}
} else {
break;
}

Loading…
Cancel
Save