noether.core.distributed.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

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)