cuda.rst.txt

torch.cuda

.. currentmodule:: torch.cuda

.. automodule:: torch.cuda
   :members:

Communication collectives

.. autofunction:: torch.cuda.comm.broadcast

.. autofunction:: torch.cuda.comm.reduce_add

.. autofunction:: torch.cuda.comm.scatter

.. autofunction:: torch.cuda.comm.gather

Streams and events

.. autoclass:: Stream
   :members:

.. autoclass:: Event
   :members: