o 8wik @sfddlmZddlZddlmZddlmZddlmZGdddeZ dee e fd e fd d Z dS) )UnionN)override) Accelerator)_AcceleratorRegistryc@seZdZdZedejddfddZedddZe ed e e e fde fd d Z e ed e e e fdeejfd d Ze ede fddZe edefddZeededdfddZdS)CPUAcceleratorzAccelerator for CPU devices.devicereturnNcCs|jdkr td|ddS)zd Raises: ValueError: If the selected device is not CPU. cpuzDevice should be CPU, got z instead.N)type ValueError)selfrr ^/home/ubuntu/sommelier/.venv/lib/python3.10/site-packages/lightning_fabric/accelerators/cpu.py setup_devices zCPUAccelerator.setup_devicecCsdS)Nr )r r r rteardown$szCPUAccelerator.teardowndevicescCst|S)z!Accelerator device parsing logic.)_parse_cpu_coresrr r r parse_devices(szCPUAccelerator.parse_devicescCst|}tdg|S)z*Gets parallel devices for the Accelerator.r )rtorchrrr r rget_parallel_devices.sz#CPUAccelerator.get_parallel_devicescCdS)z!Get the devices when set to auto.r r r r rauto_device_count5z CPUAccelerator.auto_device_countcCr)z&CPU is always available for execution.Tr r r r r is_available;rzCPUAccelerator.is_availableaccelerator_registrycCs|jd||jddS)Nr ) description)register__name__)clsrr r rregister_acceleratorsAs  z$CPUAccelerator.register_accelerators)rN)r __module__ __qualname____doc__rrrrr staticmethodrintstrrlistrrboolr classmethodrr!r r r rrs* $r cpu_coresrcCs<t|tr|rt|}t|tr|dkrtd|S)aParses the cpu_cores given in the format as accepted by the ``devices`` argument in the :class:`~pytorch_lightning.trainer.trainer.Trainer`. Args: cpu_cores: An int > 0 or a string that can be converted to an int > 0. Returns: An int representing the number of processes Raises: MisconfigurationException: If cpu_cores is not an int > 0 rz>`devices` selected with `CPUAccelerator` should be an int > 0.) isinstancer'stripisdigitr& TypeError)r+r r rrKs r) typingrrtyping_extensionsr)lightning_fabric.accelerators.acceleratorr&lightning_fabric.accelerators.registryrrr&r'rr r r rs    4