Module ivit_i.utils.device

Classes

class iDevice

Let user could capture the temperature and other device information

Usage

  1. get_all_device: return a dictoinary include all device information
  2. get_device_info: return target device information
  3. get_device_id: return the index of the device
  4. get_available_device: return a list of available device name

Ancestors

Methods

def get_available_device(self) ‑> List[str]

Get availabe devices

Returns

List[str]
a list of the availabe devices
def get_device_info(self, uuids: Optional[List[str]] = None) ‑> Dict[str, dict]

Return Accelerator Information with Dict format which key is hardware name and value is content with loading, memory, temperate

Args

uuids : Union[List[str], None], optional
the device uuid, you can get it from get_availabe_device function. Defaults to None.

Raises

KeyError
get not expected key.

Returns

Dict[str, dict]
key is hardware uuid (name) and value is dictionaray with loading, memory, temperate.

Inherited members

class iDeviceWrapper

Helper class that provides a standard way to create an ABC using inheritance.

Ancestors

  • abc.ABC

Subclasses

Static methods

def get_framework(...) ‑> str

Return current optimization framework, like tensorrt ( dgpu, jetson ), openvino ( intel ), vitis-ai ( xilinx ).

Returns

str
current optimization framework with lowercase.

Methods

def get_available_device(self) ‑> list
def get_device_info(self, uuid: Optional[str]) ‑> dict