| commit | 8aa1cefed880e3ad1a839bdbdc9d1fb9d52f5af3 | [log] [tgz] |
|---|---|---|
| author | Adam Paszke <adam.paszke@gmail.com> | Wed Mar 29 22:19:40 2017 +0200 |
| committer | Soumith Chintala <soumith@gmail.com> | Wed Mar 29 16:19:40 2017 -0400 |
| tree | bd8cd9a91da675c31a06b05542c9198c6a56d193 | |
| parent | 0d908d813b3a70322942df66c7a277c1db558623 [diff] |
Fix deadlock in autograd (#1140)
diff --git a/torch/csrc/autograd/engine.cpp b/torch/csrc/autograd/engine.cpp index 9adc77f..4c02563 100644 --- a/torch/csrc/autograd/engine.cpp +++ b/torch/csrc/autograd/engine.cpp
@@ -98,6 +98,7 @@ } } if (--task.base->outstanding_tasks == 0) { + std::lock_guard<std::mutex> lock(task.base->mutex); task.base->not_done.notify_all(); } }