perun.backend.rocmsmi
=====================

.. py:module:: perun.backend.rocmsmi

.. autoapi-nested-parse::

   ROCM Backend.



Attributes
----------

.. autoapisummary::

   perun.backend.rocmsmi.log


Classes
-------

.. autoapisummary::

   perun.backend.rocmsmi.ROCMBackend


Module Contents
---------------

.. py:data:: log

.. py:class:: ROCMBackend

   Bases: :py:obj:`perun.backend.backend.Backend`


   ROCMBackend.

   Initialises sensors to get data from AMD GPUs.


   .. py:attribute:: id
      :value: 'rocm-smi'



   .. py:attribute:: name
      :value: 'AMD ROCM'



   .. py:attribute:: description
      :type:  str
      :value: 'Access GPU information from rocm-smi python bindings.'



   .. py:method:: setup()

      Init rocm object.



   .. py:method:: close()

      Backend cleanup.



   .. py:method:: availableSensors() -> Dict[str, Tuple]

      Return string ids of visible devices.

      Returns
      -------
      Set[str]
          Set with sensor ids.



   .. py:method:: getSensors(deviceList: Set[str]) -> List[perun.data_model.sensor.Sensor]

      Gather sensor object based on a set of device ids.

      Parameters
      ----------
      deviceList : Set[str]
          Set containing divice ids.

      Returns
      -------
      List[Sensor]
          List with Sensor objects.



   .. py:method:: _getPowerSensor(device_idx: int) -> perun.data_model.sensor.Sensor


   .. py:method:: _getPowerCallback(handle: int) -> Callable[[], numpy.number]


   .. py:method:: _getMemSensor(device_idx: int) -> perun.data_model.sensor.Sensor


   .. py:method:: _getMemCallback(handle) -> Callable[[], numpy.number]