ksuit.distributed.run.utils =========================== .. py:module:: ksuit.distributed.run.utils Attributes ---------- .. autoapisummary:: ksuit.distributed.run.utils.logger Functions --------- .. autoapisummary:: ksuit.distributed.run.utils.check_single_device_visible ksuit.distributed.run.utils.get_backend ksuit.distributed.run.utils.accelerator_to_device ksuit.distributed.run.utils.parse_devices ksuit.distributed.run.utils.log_device_info Module Contents --------------- .. py:data:: logger .. py:function:: check_single_device_visible(accelerator) .. py:function:: get_backend(accelerator, device_ids=None) .. py:function:: accelerator_to_device(accelerator) .. py:function:: parse_devices(accelerator, devices) .. py:function:: log_device_info(accelerator, device_ids)