===============
fairseq2.device
===============

.. automodule:: fairseq2.device
    :no-members:

ABCs
====

.. autoclass:: CudaContext
.. autoclass:: DeviceContext
.. autoclass:: SupportsDeviceTransfer

Functions
=========

.. autofunction:: detect_default_device
.. autofunction:: get_current_device
.. autofunction:: set_device

Exceptions
==========

.. autoclass:: LocalRankOutOfRangeError
