mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
28 lines
458 B
ReStructuredText
28 lines
458 B
ReStructuredText
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:
|