a d @sZddlmZmZmZddlZddlmZGdddeZeee eefedddZ dS) )DictListUnionN) Acceleratorc@seZdZdZejddddZddddZee e e e e fe d d d Z ee e e e e fe ejd d d Zee dddZeedddZeeddddZdS)CPUAcceleratorzAccelerator for CPU devices.N)devicereturncCs|jdkrtd|ddS)zd Raises: ValueError: If the selected device is not CPU. cpuzDevice should be CPU, got z instead.N)type ValueError)selfrr j/var/www/html/stable-diffusion-webui/venv/lib/python3.9/site-packages/lightning_fabric/accelerators/cpu.py setup_devices zCPUAccelerator.setup_device)rcCsdS)Nr )r r r rteardown!szCPUAccelerator.teardown)devicesrcCs t|}|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_devicescCsdS)z!Get the devices when set to auto.r r r r rauto_device_count0sz CPUAccelerator.auto_device_countcCsdS)z&CPU is always available for execution.Tr r r r r is_available5szCPUAccelerator.is_available)accelerator_registryrcCs|jd||jjddS)Nr ) description)register __class____name__)clsrr r rregister_accelerators:s z$CPUAccelerator.register_accelerators)r __module__ __qualname____doc__rrrr staticmethodrintstrrrrrboolr classmethodrr r r r rrs  &r) cpu_coresrcCs<t|tr|rt|}t|tr0|dkr8td|S)aTParses the cpu_cores given in the format as accepted by the ``devices`` argument in the :class:`~pytorch_lightning.trainer.Trainer`. Args: cpu_cores: 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 rrCs r) typingrrrrZ)lightning_fabric.accelerators.acceleratorrrr%r&rr r r rs .