tprofiler.utils.device
=================================================

.. currentmodule:: tprofiler.utils.device

.. automodule:: tprofiler.utils.device


is_cuda_available
----------------------------------------------------------

.. autofunction:: is_cuda_available




get_device_name
----------------------------------------------------------

.. autofunction:: get_device_name




get_torch_device
----------------------------------------------------------

.. autofunction:: get_torch_device




get_device_id
----------------------------------------------------------

.. autofunction:: get_device_id




get_nccl_backend
----------------------------------------------------------

.. autofunction:: get_nccl_backend




set_expandable_segments
----------------------------------------------------------

.. autofunction:: set_expandable_segments