Get error

7 years ago · b6ca3711b4
parent 55e2cc3d87
commit b6ca3711b4
1 changed files with 3 additions and 0 deletions
--- 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));
  }