saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-06-06 00:03:22 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

fix OrtValue Release condition (#1182)

Browse source 
CUDAFence::CanRelease should also check wirte_event_
This commit is contained in:
Yufeng Li 2019-06-22 11:29:23 -07:00 committed by GitHub
parent 204bd38d6a
commit 2bfbcd323a
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

3
onnxruntime/core/providers/cuda/cuda_fence.cc
View file

@ -43,7 +43,8 @@ void CUDAFence::BeforeUsingAsOutput(onnxruntime::ProviderType provider_type, int
}
bool CUDAFence::CanRelease() {
return cudaEventQuery(read_event_) == cudaSuccess;
return cudaEventQuery(read_event_) == cudaSuccess &&
cudaEventQuery(write_event_) == cudaSuccess;
}
void CUDAFence::AfterUsedAsInput(int queue_id) {