Add compare_set in distributed docs (#61351)

H-Huang · facebook-github-bot · commit cdc027679b5d · 2021-07-08T12:30:32.000-07:00
Summary: Pull Request resolved: #61351 Test Plan: Imported from OSS Reviewed By: mrshenli Differential Revision: D29588206 Pulled By: H-Huang fbshipit-source-id: 9db48e7b6de29503275f10616470ad2d66b075f9
diff --git a/docs/source/distributed.rst b/docs/source/distributed.rst
@@ -298,6 +298,7 @@ Key-Value Stores: :class:`~torch.distributed.TCPStore`,
 .. autofunction:: torch.distributed.Store.set
 .. autofunction:: torch.distributed.Store.get
 .. autofunction:: torch.distributed.Store.add
+.. autofunction:: torch.distributed.Store.compare_set
 .. autofunction:: torch.distributed.Store.wait
 .. autofunction:: torch.distributed.Store.num_keys
 .. autofunction:: torch.distributed.Store.delete_key
