Stop Wait NCCL Stream

7 years ago · 260cfe3b86
parent e025e284c6
commit 260cfe3b86
1 changed files with 0 additions and 3 deletions
--- a/paddle/fluid/framework/parallel_executor.cc
+++ b/paddle/fluid/framework/parallel_executor.cc
@ -211,9 +211,6 @@ struct FetchOpHandle : public OpHandle {
    for (auto *input_var : inputs_) {
      input_var->pending_ops_.erase(this);
    }
-    for (auto &pair : dev_ctx_) {
-      pair.second->Wait();
-    }

    // Lazily merge tensors. Will faster code.
    MergeTensors();