From b6ca3711b4efad23afb13d5d3ca72d462550d7b0 Mon Sep 17 00:00:00 2001
From: Yu Yang <yuyang18@baidu.com>
Date: Tue, 27 Mar 2018 13:52:16 +0800
Subject: [PATCH] Get error

---
 paddle/fluid/framework/details/op_handle_base.cc | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/paddle/fluid/framework/details/op_handle_base.cc b/paddle/fluid/framework/details/op_handle_base.cc
index 63affb70542..07a4b892175 100644
--- a/paddle/fluid/framework/details/op_handle_base.cc
+++ b/paddle/fluid/framework/details/op_handle_base.cc
@@ -33,6 +33,9 @@ std::string OpHandleBase::DebugString() const {
 
 OpHandleBase::~OpHandleBase() {
 #ifdef PADDLE_WITH_CUDA
+  for (auto &ctx : dev_ctx_) {
+    ctx.second->Wait();
+  }
   for (auto &ev : events_) {
     PADDLE_ENFORCE(cudaEventDestroy(ev.second));
   }
-- 
GitLab