[pytorch][PR] update comment in autograd.h for locking (#32222)
Summary:
Just update the comment to make it accurate.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/32222
Differential Revision: D19410428
Pulled By: albanD
fbshipit-source-id: ad13596382613c2728e674a47049ea4f563964b9
diff --git a/torch/csrc/autograd/engine.h b/torch/csrc/autograd/engine.h
index 6f7c986..6b4a604 100644
--- a/torch/csrc/autograd/engine.h
+++ b/torch/csrc/autograd/engine.h
@@ -48,7 +48,7 @@
bool grad_mode_;
// To protect reads/writes to not_ready_, dependencies_, captured_vars_,
- // has_error_ and future_result_.
+ // has_error_, future_result_ and leaf_streams.
std::mutex mutex_;
std::unordered_map<Node*, InputBuffer> not_ready_;
std::unordered_map<Node*, int> dependencies_;
