ksuit.distributed.run.utils¶

Attributes¶

Functions¶

`check_single_device_visible`(accelerator)
`get_backend`(accelerator[, device_ids])
`accelerator_to_device`(accelerator)
`parse_devices`(accelerator, devices)
`log_device_info`(accelerator, device_ids)

Module Contents¶

ksuit.distributed.run.utils.logger¶

ksuit.distributed.run.utils.check_single_device_visible(accelerator)¶

ksuit.distributed.run.utils.get_backend(accelerator, device_ids=None)¶

ksuit.distributed.run.utils.accelerator_to_device(accelerator)¶

ksuit.distributed.run.utils.parse_devices(accelerator, devices)¶

Parameters:

accelerator (str)
devices (str | None)

ksuit.distributed.run.utils.log_device_info(accelerator, device_ids)¶