fix OrtValue Release condition (microsoft#1182)

CUDAFence::CanRelease should also check wirte_event_
iiSeymour · Jun 22, 2019 · 2bfbcd3 · 2bfbcd3
1 parent 204bd38
commit 2bfbcd3
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/onnxruntime/core/providers/cuda/cuda_fence.cc b/onnxruntime/core/providers/cuda/cuda_fence.cc
@@ -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) {