unilab.utils.device

Functions

get_default_device()

Detect the best available device.

get_device_info_dict()

Collect static hardware metadata (chip, cores, GPU, memory) for run records.

get_device_info_line()

mps_available()

xpu_available()

unilab.utils.device.xpu_available()[source]
Return type:

bool

unilab.utils.device.mps_available()[source]
Return type:

bool

unilab.utils.device.get_default_device()[source]

Detect the best available device.

Return type:

str

unilab.utils.device.get_device_info_dict()[source]

Collect static hardware metadata (chip, cores, GPU, memory) for run records.

Return type:

dict[str, str]

unilab.utils.device.get_device_info_line()[source]
Return type:

str