提交 260cfe3b 编写于 作者: Y Yu Yang

Stop Wait NCCL Stream

上级 e025e284
...@@ -211,9 +211,6 @@ struct FetchOpHandle : public OpHandle { ...@@ -211,9 +211,6 @@ struct FetchOpHandle : public OpHandle {
for (auto *input_var : inputs_) { for (auto *input_var : inputs_) {
input_var->pending_ops_.erase(this); input_var->pending_ops_.erase(this);
} }
for (auto &pair : dev_ctx_) {
pair.second->Wait();
}
// Lazily merge tensors. Will faster code. // Lazily merge tensors. Will faster code.
MergeTensors(); MergeTensors();
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册