noether.core.distributed.utils

Attributes

logger

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

noether.core.distributed.utils.logger

noether.core.distributed.utils.check_single_device_visible(accelerator)

noether.core.distributed.utils.get_backend(accelerator, device_ids=None)

noether.core.distributed.utils.accelerator_to_device(accelerator)

noether.core.distributed.utils.parse_devices(accelerator, devices)

Parameters:

• accelerator (str)

• devices (str | None)

noether.core.distributed.utils.log_device_info(accelerator, device_ids)