perun.backend.rocmsmi

ROCM Backend.

Attributes

Classes

ROCMBackend.

Module Contents

perun.backend.rocmsmi.log

class perun.backend.rocmsmi.ROCMBackend

Bases: perun.backend.backend.Backend

ROCMBackend.

Initialises sensors to get data from AMD GPUs.

id = 'rocm-smi'

name = 'AMD ROCM'

description: str = 'Access GPU information from rocm-smi python bindings.'

setup(): Init rocm object.

close(): Backend cleanup.

availableSensors() → Dict[str, Tuple]

Return string ids of visible devices.

Returns

Set[str]: Set with sensor ids.

getSensors(deviceList: Set[str]) → List[perun.data_model.sensor.Sensor]

Gather sensor object based on a set of device ids.

Parameters

deviceListSet[str]: Set containing divice ids.

Returns

List[Sensor]: List with Sensor objects.

_getPowerSensor(device_idx: int) → perun.data_model.sensor.Sensor

_getPowerCallback(handle: int) → Callable[[], numpy.number]

_getMemSensor(device_idx: int) → perun.data_model.sensor.Sensor

_getMemCallback(handle) → Callable[[], numpy.number]